LeeCode 828. Unique Letter String

题意

一个字符串$S$的$UNIQ(S)$值为其中只出现一次的字符的数量。
现给定字符串$S$，求所有的$UNIQ(sub(S))$的和，其中$sub(S)$为$S$的连续子串。

题解

这题$S$的长度是10000，直接暴力是肯定不行的。
那么我们一个个的考虑，先考虑字符 A A <script type="math/tex" id="MathJax-Element-255">A</script>。假设是：
* * * * A * * A * A * * * * * * A
因为只有在出现一次时，这个字符才是对最后的结果是有用的，那么就分别考虑只包一个A的子串有多少个，那么对最终的答案的贡献就是多少。

ans += (pos['A'][j + 1] - pos['A'][j]) * (pos['A'][j] - pos['A'][j - 1]);
pos['A'][j] 表示第j个'A'的位置

代码

class Solution {
public:
    int uniqueLetterString(string S) {
        vector<int> pos[26];
        for (int i = 0; i < 26; i++) {
            pos[i].push_back(-1);//把边界加入，便于计算
            pos[i].push_back(S.size());//把边界加入，便于计算
        }
        for (int i = 0; i < S.size(); i++) {
            pos[S[i] - 'A'].push_back(i);
        }
        int ans = 0;
        for (int i = 0; i < 26; i++) {
            sort(pos[i].begin(), pos[i].end());
            for (int j = 1; j < pos[i].size() - 1; j++) {
                ans += (pos[i][j + 1] - pos[i][j]) * (pos[i][j] - pos[i][j - 1]);
            }
        }
        return ans;
    }
};