题意:给定一个长度为n的字符串,要求这个字符串的所有子串的价值总和
思路:题意的奇异串是AxA,就是x旁边两个串是要一样的,不是相反的。注意x不能在A中出现,根据这个,A的范围只可能在x与上一个字母x之间,可以直接枚举。那么我们可以枚举这个字符串的每一个字母为x,然后往两边扩展去判断。判断的过程利用hash,注意hash函数的使用。使用unsigned long long,这样爆了unsigned long long后就相等于取模
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef unsigned long long int64;
const int MAXN = 100010;
char str[MAXN];
int64 hash[MAXN];
int64 base[MAXN];
void initHash(){
int len = strlen(str+1);
int seed = 131;
hash[0] = 0;
for(int i = 1 ; i <= len ; i++)
hash[i] = hash[i-1]*seed+str[i];
base[0] = 1;
for(int i = 1 ; i <= len ; i++)
base[i] = base[i-1]*seed;
}
int64 getHash(int l , int r){
return hash[r]-hash[l-1]*base[r-l+1];
}
int64 solve(){
int len = strlen(str+1);
initHash();
int64 ans = 0;
for(int i = 2 ; i < len ; i++){
int left = i-1;
int right = i+1;
while(left >= 1 && right <= len){
if(str[left] == str[i] || str[right] == str[i])
break;
if(getHash(left,i-1) == getHash(i+1,right))
ans += (int64)((i-left)*2+1)*((i-left)*2+1);
left--;
right++;
}
}
return ans;
}
int main(){
int cas;
scanf("%d" , &cas);
while(cas--){
scanf("%s" , str+1);
cout<<solve()<<endl;
}
return 0;
}