FZU Problem 2137 奇异字符串

点击打开链接

题意：给定一个长度为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;
}