描述
对于一个字符串 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;
}