AcWing:2868. 子串分值

描述

对于一个字符串 S，我们定义 S的分值 f(S) 为 S 中恰好出现一次的字符个数。

例如 f(“aba”)=1，f(“abc”)=3, f(“aaa”)=0

现在给定一个字符串 S[0…n−1]（长度为 n），请你计算对于所有 S 的非空子串 S[i…j](0≤i≤j<n)， f(S[i…j]) 的和是多少。

输入格式

输入一行包含一个由小写字母组成的字符串 S。

输出格式

输出一个整数表示答案。

数据范围

对于 20% 的评测用例，1≤n≤10；
对于 40% 的评测用例，1≤n≤100；
对于 50% 的评测用例，1≤n≤1000；
对于 60% 的评测用例，1≤n≤10000；
对于所有评测用例，1≤n≤100000。

输入样例：

ababc

输出样例：

21

样例说明

所有子串 f 值如下：

a     1
ab    2
aba   1
abab  0
ababc 1
 b    1
 ba   2
 bab  1
 babc 2
  a   1
  ab  2
  abc 3
   b  1
   bc 2
    c 1

用贡献法

举个例子
***a**a***a****

分析第二个a.
对答案的贡献就是:

*a    **a     即与前一个a位置差再减一
a*   a**   a***    即与后一个a位置差再减一
*a*   *a**   *a***    **a*   **a**    **a***   即上述两结果的乘积
三部分相加即是每个位置对答案的贡献

（别忘了还要加上自身，可以在最后加上长度即可）

 AC代码:

#include <bits/stdc++.h>

/* 定义 */
using namespace std;
const int N = 1e5 + 10;
typedef long long LL;

/* 记录每个字母的位置 */
vector<int> pos[26];
char s[N];

int main()
{
    scanf("%s", s); // 读入
    int n = strlen(s); // 求长度

    /* 求每个字母位置 */
    for(int i = 0 ; i < n ;i ++)
    {
        int id = s[i] - 'a'; // 字母索引
        /* 如果是第一次出现 就先加入一个 -1 作为首位 */
        if(!pos[id].size()) pos[id].push_back(-1);
        pos[id].push_back(i); // 加入位置
    }

    /* 添加末尾 */
    for(int i = 0; i < 26; i ++)
        /* 如果出现过该字母 就加 n 作为末尾*/
        if(pos[i].size()) pos[i].push_back(n); 
    

    LL ans = 0; // 初始化ans

    /* 核心 */
    for(int i = 0;i < 26; i ++)
        if(pos[i].size())
            for(int j = 1; j < pos[i].size() - 1 ; j ++)
            {
                /* 计算贡献值 三部分相加 */
                ans += (pos[i][j] - pos[i][j - 1] - 1) + (pos[i][j + 1] - pos[i][j] -1) + 1ll * (pos[i][j] - pos[i][j - 1] - 1) * (pos[i][j + 1] - pos[i][j] - 1);
            }

    /* 加上单独的 */
    ans += n;

    cout << ans;
    return 0;
}