Leetcode2207. 字符串中最多数目的子字符串

Every day a Leetcode

题目来源：2207. 字符串中最多数目的子字符串

解法1：贪心 + 一次遍历

设 pattern 的第一个字符为 x，第二个字符为 y。

根据题意，x 插入的位置越靠左，答案的个数越多；y 插入的位置越靠右，答案的个数越多。

那么 x 应置于 text 最左侧，yyy 应置于 text 最右侧。

我们可以遍历 text 来统计答案：每遇到一个 y，就累加左边 x 的个数。

注意特判 x=y 的情况。当两个字符相同时，设 x 的出现次数为 cnt，子字符串的出现次数 = cnt * (cnt + 1) / 2。

代码实现时，放 x 还是放 y 可以一次遍历求出来：

• 放 x 在最左侧：每次遇到 y 都会多算一个 x，因此答案会多 cnt_y 个（cnt_y 是 y 在 text 中的出现次数）。
• 放 y 在最右侧：这个 y 会与 text 的所有 x 组成子序列，因此答案会多 cnt_x 个（cnt_x 是 x 在 text 中的出现次数）。

代码：

/*
 * @lc app=leetcode.cn id=2207 lang=cpp
 *
 * [2207] 字符串中最多数目的子字符串
 */

// @lc code=start
class Solution
{
public:
    long long maximumSubsequenceCount(string text, string pattern)
    {
        char x = pattern[0], y = pattern[1];
        if (x == y)
        {
            long long cnt = count(text.begin(), text.end(), x);
            return cnt * (cnt + 1) / 2;
        }

        long long ans = 0;
        int cnt_x = 0, cnt_y = 0;
        for (char &c : text)
        {
            if (c == x)
                cnt_x++;
            else if (c == y)
            {
                cnt_y++;
                // 每遇到一个 y，就累加左边 x 的个数
                ans += cnt_x;
            }
        }
        return ans + max(cnt_x, cnt_y);
    }
};
// @lc code=end

结果：

复杂度分析：

时间复杂度：O(n)，其中 n 是字符串 text 的长度。

空间复杂度：O(1)。