字符串中最多数目的子序列 2207

题目#

给你一个下标从 0 开始的字符串 text 和另一个下标从 0 开始且长度为 2 的字符串 pattern ，两者都只包含小写英文字母。

你可以在 text 中任意位置插入 一个 字符，这个插入的字符必须是 pattern[0] 或者 pattern[1] 。注意，这个字符可以插入在 text 开头或者结尾的位置。

请你返回插入一个字符后，text 中最多包含多少个等于 pattern 的 子序列 。

子序列 指的是将一个字符串删除若干个字符后（也可以不删除），剩余字符保持原本顺序得到的字符串。

示例 1：

输入： text = “abdcdbc”, pattern = “ac”
输出： 4
解释：
如果我们在 text[1] 和 text[2] 之间添加 pattern[0] = ‘a’ ，那么我们得到 “ab_adcdbc” 。那么 “ac” 作为子序列出现 4 次。
其他得到 4 个 “ac” 子序列的方案还有 “aabdcdbc” 和 “abdacdbc” 。
但是，“abdcadbc” ，“abdccdbc” 和 “abdcdbcc_” 这些字符串虽然是可行的插入方案，但是只出现了 3 次 “ac” 子序列，所以不是最优解。
可以证明插入一个字符后，无法得到超过 4 个 “ac” 子序列。

示例 2：

输入： text = “aabb”, pattern = “ab” 输出： 6
解释：
可以得到 6 个 “ab” 子序列的部分方案为 “_aaabb” ，“aaabb” 和 “aabb_b” 。

提示：

• 1 <= text.length <= 10^5
• pattern.length == 2
• text 和 pattern 都只包含小写英文字母。

分析与解答#

只需要保留原来的字符串中pattern的部分，然后算两种各有多少个，添加的少的那种在首或尾即可。非常直观，就不必证明了。

如果pattern[0]有a个，pattern[1]有b个，那么插入总是可以增加max(a, b)个子序列。在扫描的时候，如果发现b，则可以增加a个子序列。

特殊情况是pattern中的两个字符相同，这样更加简单，记得做特殊处理。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

class Solution {
public:
    long long maximumSubsequenceCount(string text, string pattern) {
        long long ans = 0;
        int a = 0, b = 0;

        if (pattern[0] == pattern[1]) {
            for (char c : text) {
                if (c == pattern[0]) {
                    a++;
                }
            }
            ans =  (long long)a * (a + 1) / 2;
            return ans;
        }

        for (char c : text) {
            if (c == pattern[0]) {
                a++;
            } else if (c == pattern[1]) {
                b++;
                ans += a;
            }
        }
        return ans + max(a, b);
    }
};