LC-828. 统计子串中的唯一字符（贡献法 思考变化量）

828. 统计子串中的唯一字符

困难

我们定义了一个函数 countUniqueChars(s) 来统计字符串 s 中的唯一字符，并返回唯一字符的个数。

例如：s = "LEETCODE" ，则其中 "L", "T","C","O","D" 都是唯一字符，因为它们只出现一次，所以 countUniqueChars(s) = 5 。

本题将会给你一个字符串 s ，我们需要返回 countUniqueChars(t) 的总和，其中 t 是 s 的子字符串。输入用例保证返回值为 32 位整数。

注意，某些子字符串可能是重复的，但你统计时也必须算上这些重复的子字符串（也就是说，你必须统计 s 的所有子字符串中的唯一字符）。

示例 1：

输入: s = "ABC"
输出: 10
解释: 所有可能的子串为："A","B","C","AB","BC" 和 "ABC"。
     其中，每一个子串都由独特字符构成。
     所以其长度总和为：1 + 1 + 1 + 2 + 2 + 3 = 10

示例 2：

输入: s = "ABA"
输出: 8
解释: 除了 countUniqueChars("ABA") = 1 之外，其余与示例 1 相同。

示例 3：

输入：s = "LEETCODE"
输出：92

提示：

• 1 <= s.length <= 105
• s 只包含大写英文字符

贡献法

https://leetcode.cn/problems/count-unique-characters-of-all-substrings-of-a-given-string/solutions/1804144/by-endlesscheng-ko4z/?envType=daily-question&envId=2023-11-26

问题：有一个字符串 s，求满足条件的 s 的子串的个数「和」

思考字串统计类问题的通用技巧之一：

贡献法：将所有子串按照其末尾字符的下标分组，考虑两组相邻的字串：以 s[i-1] 结尾的字串、以 s[i] 结尾的字串。

• 以 s[i] 结尾的字串，可以看成以 s[i-1] 结尾的字串，在末尾添加上 s[i] 组成。

那么思考：从左往右遍历s，考虑将 s[i] 添加到以 s[i-1] 结尾的子串的末尾，添加后，以s[i-1]结尾的字串值会如何变化？

class Solution {
    /**
    我们在从左往右遍历s的同时，对每个字母s[i]
        记录其上一次出现的下标last[s[i]]和上上一次出现的下标last1[s[i]]
    我们可以算出从「以s[i-1]结尾的子串」和「以s[i]结尾的子串」
    countUniqueChars的值，增加/减少了多少
        增加了 i - last0[s[i]] (s[i] 单独作为子串贡献了1)
        减少了 last0[s[i]] - last1[s[i]]
    两者总变化量为 i - 2*last0[s[i]] + last1[s[i]]

    在计算时，如果last0[s[i]]和last1[s[i]]不存在，可以看作-1，保证式子的正确性
       注：i + 2 - 1 ==> i从未出现过，则countUniqueChars的值就会增加i和i自身贡献1
     */
    public int uniqueLetterString(String s) {
        int ans = 0, total = 0; // 这里total就是countUniqueChars
        int[] last0 = new int[26], last1 = new int[26];
        Arrays.fill(last0, -1);
        Arrays.fill(last1, -1);
        for(int i = 0; i < s.length(); i++){
            int c = s.charAt(i) - 'A';
            total += i - 2 * last0[c] + last1[c];
            ans += total;
            last1[c] = last0[c];
            last0[c] = i;
        }
        return ans;
    }
}