leetcode 6050. 字符串的总引力—(每日一难day8)

6050. 字符串的总引力

字符串的 引力 定义为：字符串中 不同 字符的数量。

例如，“abbca” 的引力为 3 ，因为其中有 3 个不同字符 ‘a’、‘b’ 和 ‘c’ 。
给你一个字符串 s ，返回 其所有子字符串的总引力 。

子字符串 定义为：字符串中的一个连续字符序列。

示例 1：

输入：s = “abbca”
输出：28
解释：“abbca” 的子字符串有：

• 长度为 1 的子字符串：“a”、“b”、“b”、“c”、“a” 的引力分别为 1、1、1、1、1，总和为 5 。
• 长度为 2 的子字符串：“ab”、“bb”、“bc”、“ca” 的引力分别为 2、1、2、2 ，总和为 7 。
• 长度为 3 的子字符串：“abb”、“bbc”、“bca” 的引力分别为 2、2、3 ，总和为 7 。
• 长度为 4 的子字符串：“abbc”、“bbca” 的引力分别为 3、3 ，总和为 6 。
• 长度为 5 的子字符串：“abbca” 的引力为 3 ，总和为 3 。 引力总和为 5 + 7 + 7 + 6 + 3 = 28 。

示例 2：

输入：s = “code”
输出：20
解释：“code” 的子字符串有：

• 长度为 1 的子字符串：“c”、“o”、“d”、“e” 的引力分别为 1、1、1、1 ，总和为 4 。
• 长度为 2 的子字符串：“co”、“od”、“de” 的引力分别为 2、2、2 ，总和为 6 。
• 长度为 3 的子字符串：“cod”、“ode” 的引力分别为 3、3 ，总和为 6 。
• 长度为 4 的子字符串：“code” 的引力为 4 ，总和为 4 。 引力总和为 4 + 6 + 6 + 4 = 20 。

提示：

1 <= s.length <= 1e5
s 由小写英文字母组成

解析
枚举以每个点为结尾的引力值，所有点的引力值之和就是总引力值。
例如：abacb
计算最后一个b为结尾的引力值：b bc bca bcab bcaba
计算b的前一个字母c为结尾的引力值：c ca cab caba
可以看出两者的关系：
b为结尾引力值 = c为结尾引力值+b对c中各个子串的贡献+b本身的贡献。

b对c贡献1，对ca贡献1，对cab没有贡献（cab中本身包含b），对caba没有贡献，b自身贡献1.

更详细解释参考链接
解析

code

class Solution {
public:
    // 总共26个字母
    long long appealSum(string s) {
        int n=s.size();
        int f[n+5];
        memset(f,0,sizeof(f));
        unordered_map<char,int> ma; 
        f[0]=1;
        ma[s[0]]=0;
        long long res=1;
        for(int i=1;i<n;i++){
            int b=-1;
            if(ma.find(s[i])!=ma.end())
                b=ma[s[i]];
            f[i]+=f[i-1];
            f[i]+=i-b;
            res+=f[i];
            ma[s[i]]=i;
        }
        return res; 
    }
};