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

3/100天刷题记录

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' ，那么我们得到 "abadcdbc" 。那么 "ac" 作为子序列出现 4 次。
其他得到 4 个 "ac" 子序列的方案还有 "aabdcdbc" 和 "abdacdbc" 。
但是，"abdcadbc" ，"abdccdbc" 和 "abdcdbcc" 这些字符串虽然是可行的插入方案，但是只出现了 3 次 "ac" 子序列，所以不是最优解。
可以证明插入一个字符后，无法得到超过 4 个 "ac" 子序列。

示例 2：

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

提示：

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

思路：

插入的字符增加的子序列相当于pattern两个字符中出现次数多的字符的次数，从而这道题转换为求解text中pattern子序列的问题，直接的两层循环遍历固定超时，我们采用一点动规思想，求解pattern[1]前面pattern[0]的个数，从而得出子序列的个数。通过测试，发现没有考虑到pattern两个字符是一样的情况，我们单独对它讨论，根据简单推导，其子序列个数是其字符的数量counts*(counts - 1)/2，加上插入的字符，即(counts + 1)*count/2。

代码:

class Solution {
public:
    long long maximumSubsequenceCount(string text, string pattern) {
        if(pattern[0] == pattern[1]){
            long long counts = 0;
            for(int i = 0; i < text.size();i++){
                if(text[i] == pattern[0]) counts++;
            }
            return counts*(counts + 1)/2;
        }
        long long ans = 0;
        long long cnt = 0;
        int cnt1 = 0;
        int cnt2 = 0;
        for(int i = 0; i < text.size();i++){
            if(text[i] == pattern[0]) cnt1++;
            if(text[i] == pattern[1]) cnt2++;

        }
        ans = cnt1 > cnt2 ? cnt1 : cnt2;
        int count = 0;
        if(text[0] == pattern[0]) count = 1;
        for(int i = 1; i < text.size();i++){
            if(text[i] == pattern[0]) count++;
                if(text[i] == pattern[1]){
                    cnt+=count;
                }
        }
        return ans + cnt;
    }
};

解答出来发现自己写的有些问题，故此优化：

统计和判定顺序存在一定问题，cnt1和cnt2可以有相等情况。

class Solution {
public:
    long long maximumSubsequenceCount(string text, string pattern) {
        long long ans = 0;
        long long cnt = 0;
        int cnt1 = 0;
        int cnt2 = 0;
        for(int i = 0; i < text.size();i++){
            if(text[i] == pattern[0]) cnt1++;
            if(text[i] == pattern[1]) cnt2++;

        }
        ans = cnt1 >= cnt2 ? cnt1 : cnt2;
        int count = 0;
        if(text[0] == pattern[0]) count = 1;
        for(int i = 1; i < text.size();i++){
                if(text[i] == pattern[1]){
                    cnt+=count;
                }
                if(text[i] == pattern[0]) count++;
        }
        return ans + cnt;
    }
};