⭐️ 本文已收录到 AndroidFamily，技术和职场问题，请重视大众号 [彭旭锐] 和 BaguTree Pro 知识星球发问。

学习数据结构与算法的关键在于掌握问题背面的算法思想框架，你的思考越抽象，它能覆盖的问题域就越广，理解难度也更复杂。在这个专栏里，小彭与你共享每场 LeetCode 周赛的解题陈述，一同体会上分之旅。

本文是 LeetCode 上分之旅系列的第 43 篇文章，往期回忆请移步到文章末尾~

LeetCode 双周赛 112

T1. 判别经过操作能否让字符串持平 I（Easy）

  • 标签：模仿

T2. 判别经过操作能否让字符串持平 II（Medium）

  • 标签：模仿、计数、排序

T3. 简直唯一子数组的最大和（Medium）

  • 标签：滑动窗口、计数

T4. 计算一个字符串的 k 子序列美丽值最大的数目（Hard）

  • 标签：枚举、贪心、排序、乘法原理、组合数

LeetCode 周赛上分之旅 #43 计算机科学本质上是数学吗？

T1. 判别经过操作能否让字符串持平 I（Easy）

https://leetcode.cn/problems/check-if-strings-can-be-made-equal-with-operations-i/

题解（模仿）

因为只能沟通距离偶数倍的位置，因而相当于比较两个字符串相同奇偶性下标上的元素是否持平。

  • 写法 1：根据散列表
class Solution {
    fun canBeEqual(s1: String, s2: String): Boolean {
        return setOf(s1[0], s1[2]) == setOf(s2[0], s2[2]) && setOf(s1[1], s1[3]) == setOf(s2[1], s2[3])
    }
}
  • 写法 2：根据字符串
class Solution:
    def checkStrings(self, s1: str, s2: str) -> bool:
        return sorted(s1[0::2]) == sorted(s2[0::2]) and sorted(s1[1::2]) == sorted(s2[1::2])

复杂度剖析：

  • 时刻复杂度：O(1)O(1)
  • 空间复杂度：O(1)O(1)

T2. 判别经过操作能否让字符串持平 II（Medium）

https://leetcode.cn/problems/check-if-strings-can-be-made-equal-with-operations-ii/

题解（模仿）

同上，别离计算奇偶下标上的元素个数是否持平。

写法 1：根据计数；

class Solution {
    fun checkStrings(s1: String, s2: String): Boolean {
        val U = 26
        val cnts = Array(2) { IntArray(U) }
        for ((i, e) in s1.withIndex()) {
            cnts[i % 2][e - 'a']++
        }
        for ((i, e) in s2.withIndex()) {
            cnts[i % 2][e - 'a']--
        }
        return cnts[0].all {it == 0} && cnts[1].all {it == 0}
    }
}

复杂度剖析：

  • 时刻复杂度：O(n+U)O(n + U) 线性遍历时刻与计数时刻；
  • 空间复杂度：O(U)O(U) 计数数组空间。

写法 2：根据字符串：

class Solution:
    def checkStrings(self, s: str, t: str) -> bool:
        return all(sorted(s[p::2]) == sorted(t[p::2]) for p in range(2))

复杂度剖析：

  • 时刻复杂度：O(nlgn)O(nlgn) 排序时刻；
  • 空间复杂度：O(n)O(n) 结构字符串空间。

T3. 简直唯一子数组的最大和（Medium）

https://leetcode.cn/problems/maximum-sum-of-almost-unique-subarray/

题解（滑动窗口 + 计数）

滑动窗口模板题，维护窗口中不同元素的品种数和总和：

class Solution {
    fun maxSum(nums: List<Int>, m: Int, k: Int): Long {
        var cnts = HashMap<Int, Int>()
        var type = 0
        var sum = 0L
        var ret = 0L
        for (j in nums.indices) {
            // 滑入
            cnts[nums[j]] = cnts.getOrDefault(nums[j], 0) + 1
            if (1 == cnts[nums[j]]!!) type++
            sum += nums[j]
            // 滑出
            if (j >= k) {
                val i = j - k
                cnts[nums[i]] = cnts[nums[i]]!! - 1
                if (0 == cnts[nums[i]]) type --
                sum -= nums[i]
            }
            // 记载
            if (j >= k - 1 && type >= m) {
                ret = max(ret, sum)
            }
        }
        return ret
    }
}

复杂度剖析：

  • 时刻复杂度：O(n)O(n) 线性遍历时刻；
  • 空间复杂度：O(n)O(n) 散列表空间。

T4. 计算一个字符串的 k 子序列美丽值最大的数目（Hard）

https://leetcode.cn/problems/count-k-subsequences-of-a-string-with-maximum-beauty/

问题剖析

  • 问题目标： 求一切长为 kk 的子序列中美丽值是最大值的子序列数目；
  • 问题要件： 先计算长为 kk 的子序列的最大美丽值，再计算满足美丽值是最大值的子序列计划数；
  • 关键信息 1： 子序列要挑选不重复的字母；
  • 关键信息 2： 同一个字符在原字符串中的不同位置能够结构不同子序列；
  • 中心变量： f(c)f(c) 是 字符 cc 的呈现次数，美丽值是子序列中字符的 f(c)f(c) 之和；
  • 鸿沟状况： 已然子序列要挑选不重复的字母，那么存在鸿沟状况，当 kk > 字符串的字符品种数：那么一定不能结构 kk 子序列，回来 00。

题解一（暴力枚举 + 乘法原理）

最简略的做法，咱们能够枚举一切或许的 kk 子序列，并记载呈现最大美丽值的计划数，怎么完成呢？

  • 办法 1 – 考虑到子序列需求保留原字符串的顺序，直接的想法是枚举字符串中的每个下标 s[i]s[i] 选和不选，但是时刻复杂度是 O(2n)O(2^n) 明显不成立；
  • 办法 2 – 事实上咱们不需求从原字符串的角度枚举，而是能够从字符集的角度枚举，那样时刻复杂度就能够用乘法原理来优化。比如说 a 的呈现次数是 22，而 b 的呈现次数是 33，那么一切 a 与 b 能够结构的子序列计划数就是 2 * 3 = 6。

那么，办法会不会超时呢，咱们来简略剖析下：

因为字符集的巨细 UU 最多只要 2626 个，那么子序列的计划数最多有 C26kC_{26}^k 个，而因为 kk 大于 UU 的计划是不存在的，因而合法的计划数最多只要 CUU2=C2613=10400600C_{U}^{\frac{U}{2}} = C_{26}^{13} = 10400600 约等于 10710^7。只要咱们保证求解每个子问题的时刻复杂度是 O(1)O(1) 的话是能够经过的。

枚举完成：

class Solution {
    fun countKSubsequencesWithMaxBeauty(s: String, k: Int): Int {
        val MOD = 1000000007
        // 计数
        val cnts = HashMap<Char, Int>()
        for (e in s) {
            cnts[e] = cnts.getOrDefault(e, 0) + 1
        }
        val m = cnts.size
        if (m < k) return 0 // 特判
        // 枚举子序列
        val keys = cnts.toList()
        var maxCount = 0L
        var maxF = 0
        // 回溯
        fun count(index: Int, size: Int, curF: Int, curCount: Long) {
            // 停止条件
            if (size == k) {                
                if (curF > maxF) {
                    maxF = curF
                    maxCount = curCount // 更新最大美丽值计划数
                } else if (curF == maxF) {
                    maxCount = (maxCount + curCount) % MOD // 添加计划数
                }
                return
            }
            if (size + m - index < k) return // 剪枝（长度不够）
            for (i in index until m) {
                val (c, cnt) = keys[i]
                count(i + 1, size + 1, curF + cnt, curCount * cnt % MOD /* 乘法原理 */) 
            }
        }
        count(0, 0, 0, 1)
        return maxCount.toInt()
    }
}

复杂度剖析：

  • 时刻复杂度：O(Cmk)O(C_m^k) 其间 mm 为字符品种 ；
  • 空间复杂度：O(m)O(m) 散列表空间与递归栈空间。

题解二（排序 + 贪心 + 乘法原理）

考虑 k=1k = 1 的鸿沟状况：

明显需求挑选 f(c)f(c) 值最大的 11 个字母，如果存在 mm 个字母的 f(c)f(c) 等于最大值，那么存在 Cm1=mC_m^1 = m 种计划。这说明咱们没必要枚举一切字母的子序列： 因为子序列中的字符是不重复的，因而 kk 子序列必然要挑选 f(c)f(c) 值最大的 kk 个字母，咱们能够将字母依照 f(c)f(c) 倒序排序，优先取 f(c)f(c) 更大的字母。

详细完成上：

咱们将字母依照 f(c)f(c) 分桶排序，如果桶内字母数量 KK 小于等于 kk，那么桶内元素都需求挑选，否则还要计算桶内元素挑选 kk 个的计划数：

  • 挑选桶内一切元素，计划数为 cntKcnt^K
  • 挑选桶内部分元素，计划数为 CKk⋅cntkC_K^k cnt^k

其间涉及到幂运算，实质是倍增思想：

// 快速幂 x^n
private fun powM(a: Int, b: Int, mod: Int) : Long {
    var x = a.toLong()
    var n = b.toLong()
    var ret = 1L
    while (n > 0L) {
        if (n % 2 == 1L) ret = ret * x % mod
        x = x * x % mod
        n /= 2
    }
    return ret
}

其间涉及到 组合数：

  • 计算式：
// 组合数计算公式 O(k)
private fun comb(n: Int, k: Int, mod: Int) : Int {
    var ret = 1L
    for (i in 1 .. k) {
        ret = ret * (n - i + 1) / i % mod
    }
    return ret.toInt()
}
  • 递推式（杨辉三角）：
// 递归 O(n^2)
private fun comb(n: Int, k: Int, mod: Int) : Int {
    if (n == k) {
        return 1
    } else if (k == 1) {
        return n
    } else {
        return (comb(n - 1, k - 1, mod) + comb(n - 1, k, mod)) % mod
    }
}
// 迭代 O(n^2)
private fun comb(n: Int, k: Int, mod: Int) : Int {
    val c = Array(n + 1) { IntArray(n + 1) }
    for (i in 1 .. n) {
        c[i][0] = 1
        c[i][i] = 1
        for (j in 1 until i) {
            c[i][j] = (c[i-1][j] + c[i-1][j-1]) % mod
        }
    }
    return c[n][k]
}
  • 卢卡斯定理： 当问题规划很大，且模不太大时运用 Lucas 定理。

LeetCode 周赛上分之旅 #43 计算机科学本质上是数学吗？

// 组合数计算公式
private fun comb(n: Long, k: Long, mod: Int) : Int {
    var n = n
    var ret = 1L
    for (i in 1 .. k) {
        ret = ret * n-- / i % mod
    }
    return ret.toInt()
}
// 卢卡斯定理
fun Lucas(n: Long, k: Long, mod: Int) : Long {
    if (k == 0L) return 1L;
    return (comb(n % mod, k % mod, mod) * Lucas(n / mod, k / mod, mod)) % mod;
}

完整代码：

class Solution {
    fun countKSubsequencesWithMaxBeauty(s: String, k: Int): Int {
        val MOD = 1000000007
        // 计数
        val cnts = HashMap<Char, Int>()
        var maxCnt = 0
        for (e in s) {
            cnts[e] = cnts.getOrDefault(e, 0) + 1
            maxCnt = max(maxCnt, cnts[e]!!)
        }
        val m = cnts.size
        if (m < k) return 0 // 特判
        // 有序调集
        val map = TreeMap<Int, Int>() { c1, c2 -> 
            c2 - c1
        }
        // 二次频率
        for ((_, c) in cnts) {
            map[c] = map.getOrDefault(c, 0) + 1
        }
        val cntCnts = map.toList()
        // println(cntCnts.joinToString())
        // 结构计划
        var ret = 1L
        var leftK = k
        for ((cnt, K) in cntCnts) {
            if (K > leftK) {
                ret = ret * powM(cnt, leftK, MOD) * comb(K, leftK, MOD) % MOD
            } else {
                ret = ret * powM(cnt, K, MOD) % MOD
            }
            leftK -= K
            if (leftK <= 0) break
        }
        return ret.toInt()
    }
    // 组合数计算公式 C_n^k
    private fun comb(n: Int, k: Int, mod: Int) : Int {
        if (n == k) {
            return 1
        } else if (k == 1) {
            return n
        } else {
            return (comb(n - 1, k - 1, mod) + comb(n - 1, k, mod)) % mod
        }
    }
    // 快速幂 x^n
    private fun powM(x_: Int, n_: Int, mod: Int) : Long {
        var x = x_.toLong()
        var n = n_.toLong()
        var ret = 1L
        while (n > 0L) {
            if (n % 2 == 1L) ret = ret * x % mod
            x = x * x % mod
            n /= 2
        }
        return ret
    }
}

Python 中组合数和幂运算能够很方便地运用库函数：

class Solution:
    def countKSubsequencesWithMaxBeauty(self, s: str, k: int) -> int:
        MOD = 10 ** 9 + 7
        ans = 1
        cnt = Counter(Counter(s).values())
        for c, num in sorted(cnt.items(), reverse=True): # 二次计数
            if num >= k:
                return ans * pow(c, k, MOD) * comb(num, k) % MOD
            ans *= pow(c, num, MOD)
            k -= num
        return 0

复杂度剖析：

  • 时刻复杂度：O(n+m)O(n + m) 首要时刻在枚举字符串的环节；
  • 空间复杂度：O(m)O(m) 散列表空间。

推荐阅读

LeetCode 上分之旅系列往期回忆：

  • LeetCode 单周赛第 360 场 当 LeetCode 考树上倍增，出题的趋势在改变吗
  • LeetCode 单周赛第 359 场 结合离散化的线性 DP 问题
  • LeetCode 双周赛第 111 场 按部就班地解决动态规划问题
  • LeetCode 双周赛第 110 场 结合排序不等式的动态规划

⭐️ 永远信任美好的事情即将发生，欢迎加入小彭的 Android 沟通社群~

LeetCode 周赛上分之旅 #43 计算机科学本质上是数学吗？