题目链接https://leetcode.cn/problems/maximize-number-of-subsequences-in-a-string/description/?envType=daily-question&envId=2024-09-24
给你一个下标从 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" 。
思路:
把pattern中第一个字符a加到最前面,此时后面b出现的次数最多;把b字符加到最后面,此时前面的a出现的次数最多。
遍历:从后往前,统计a,b出现的次数。当a出现时,加上b出现过的次数。
特殊情况a==b时是等差数列。
代码:
class Solution {
public long maximumSubsequenceCount(String text, String pattern) {
char a = pattern.charAt(0), b = pattern.charAt(1);
long res = 0;
int aCnt = 0, bCnt = 0;
//统计a,b出现次数
for(int i = text.length() - 1; i >= 0; --i){
if(text.charAt(i) == a) {
aCnt++;
res += bCnt;
}else if(text.charAt(i) == b){
bCnt++;
}
}
//特殊情况a、b相同
if(a == b) {
long cnt = aCnt + bCnt + 1;
return ((cnt - 1) + 1) * (cnt - 1) / 2;
}
res += Math.max(aCnt,bCnt);
return res;
}
}