2207. 字符串中最多数目的子序列 - 力扣(LeetCode)
本题的示例解释有明显的误导倾向,刚开始遇到很容易想歪。其实那个任意位置插入毫无用处,本题真正有用的插入位置只有pattern[0]插入字符串开头,或者pattern[1]插入字符串末尾。这样计算子序列的时候前面(或后面)所有能够成为子序列的元素都可以组成子序列了。
明白这点后整理一下思路:
- 先遍历原字符串进行统计,看看原字符串中有多少个子串;
- 在遍历的过程中顺便统计一下pattern的两个元素各有多少个,看看是插入在开头的pattern[0]有更多的子串,还是插入在末尾的pattern[1]有更多的子串,这将决定了最后插入哪个元素;
- 计算结果,将原字符串的子串数目加上插入最佳元素后的子串数目即可。
上代码:
class Solution {
public:
long long maximumSubsequenceCount(string text, string pattern) {
long long ans = 0;
int ptn0 = 0, ptn1 = 0;//统计pattern两个元素的个数,也就是其对应的子串的数目
for(char c : text){ //遍历字符串
if(c == pattern[1]){
ptn1++;
ans += ptn0; //统计原字符串中子串的数量
}
if(c == pattern[0]){
ptn0++;
}
}
return ans + max(ptn0, ptn1); //插入元素的步骤:选择插入后子串增加的更多的元素插入
}
};
为什么要先判断pattern[1]呢?如果先判断pattern[0]会怎么样?
如果先判断pattern[0],在特殊情况下就会出现将子串计算多了的情况,比如这样:
可以数数看。