子串分值

子串分值

---

from lanqiao 11 - H

from acwing 2868

Time limit：1s

Memory limit：256MB

---

---

暴力方法大概是O(n³)，差不多只能过20%到30%的数据。

这里讲一个巧妙的方法，并且利用并查集（或者说是简单的索引）思想来（优化）实现。

该题是说要找子串，我们不用找子串，找子串就是暴力了。

比如举例一个字符串ababa
我们只需要找到每个字母对最终答案的贡献值就可以了
1 2 3 4 5
a b a b a
我们看到第3个字符a
它的前一个相同字符是在第1个位置，后一个相同字符是在第5个位置
那么该字符能出现的子串只能是在第2个位置到第4个位置之间，因为如果超过了第2个和第4个位置，就会出现至少两个字符a
那么第3个字符就不会对函数f有贡献了
那么第3个字符的贡献值就是（3 - 1） * （5 - 3） = 4
然后我们累加每个字母的贡献，就可以得到最终的结果了

ac代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
string s;
int pre[100005],rear[100005];   //pre存储该位置的前一个相同字母的位置，rear存储后一个
int a[30];                      //存储该字母的索引
int sum;                        //所求结果
signed main(){
    cin>>s;
    for(int i = 0;i < 26;++i)   //初始化每个字母的前序索引为s开始的前一个位置
        a[i] = -1;
    for(int i = 0,j = s.length();i < j;++i){    //找到每个位置的前一个相同字母的位置
        int x = s[i] - 'a';
        pre[i] = a[x];
        a[x] = i;
    }
    for(int i = 0,j = s.length();i < 26;++i)    //初始化每个字母的后续索引为s结束的后一个位置
        a[i] = j;
    for(int i = s.length() - 1;i >= 0;--i){     //找到每个位置的后一个相同字母的位置
        int x = s[i] - 'a';
        rear[i] = a[x];
        a[x] = i;
    }
    for(int i = 0,j = s.length();i < j;++i)     //累加每个字母的贡献值
        sum += (i - pre[i]) * (rear[i] - i);
    cout<<sum;
    return 0;
}