JZOJ 3798.【NOIP2014模拟8.22】临洮巨人

$Description$

给定一个长度为$n$的字符串，要求求出包含$A,B,C$数目相同的子串个数

数据范围：$n\le 10^6$

---

$Solution$

枚举两个端点，这样就唯一确定了一个区间，然后$O(n)$判断，这样做是$O(n^3)$的，可以拿到30$pts$

将上述算法的判断改成前缀和，即

设$su{m}_{i,0/1/2}$表示$1\sim i$中含$A/B/C$的个数，则一个串合法，当且仅当

$su{m}_{i,0}-su{m}_{j,0}=su{m}_{i,1}-su{m}_{j,1}=su{m}_{i,2}-su{m}_{j,2}(j<i)$

此$[j+1,i]$区间合法

这样就可以$O(n^2)$搞，可以拿到70分的高分

上述算法的判断显然已经最优，唯一的瓶颈就是枚举区间，我们考虑只枚举右端点，然后计算合法的左端点数量

考虑将连等条件转换为两个同时成立的条件，即将上述判断式转换为

$su{m}_{i,0}-su{m}_{j,0}=su{m}_{i,1}-su{m}_{j,1}$
且
$su{m}_{i,1}-su{m}_{j,1}=su{m}_{i,2}-su{m}_{j,2}$

注意到这里$i$与$j$相关比较难搞，考虑把它们分开，即移项：

$su{m}_{i,0}-su{m}_{i,1}=su{m}_{j,0}-su{m}_{j,1}$
且
$su{m}_{i,1}-su{m}_{i,2}=su{m}_{j,1}-su{m}_{j,2}$

设
$p_i=su{m}_{i,0}-su{m}_{i,1}$
$q_i=su{m}_{i,1}-su{m}_{i,2}$

则条件转换成为
$p_i=p_j$且$q_i=q_j(j<i)$的数对个数

也就是说对于每一个确定的$i$，需要找前面所有等于它的$j$，似乎有些麻烦（当然也可以做）

考虑如何将两个条件同时判断
一个显然的想法是$hash$，即令$h_i=hash(p_i,q_i)$，这样我们就转换成为

对于每一个确定的$h_i$，求$h_j=h_i(j<i)$的数对个数，显然我们直接一个$stable\_sort$这样在一段连续的长度为$k=(i-j+1)$相同序列中，这段对答案的贡献就是$\frac{k(k-1)}{2}$

于是问题得到解决，时间复杂度$O(nlogn)$

补充：

1. $hash$函数的构造方法：考虑到$0\le p_i,q_i\le n$，直接令$h_i=n{p}_i+q_i$即可，这样一定不会有哈希冲突
2. 下列代码中$i$对应的是左端点，$j$是右端点，各位不要理解错了
3. 注意判断式中的$j$是包括0的，因为左端点可以是1，所以我们新添加进去一个0即可，即$h[++n]=0$

$Code$

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define N 1000010
using namespace std;char s[N];
int n;
LL sum[N][3],h[N],ans;
signed main()
{
	scanf("%s",s+1);
	n=strlen(s+1);
	for(register int i=1;i<=n;i++)
	{
		sum[i][0]=sum[i-1][0]+(s[i]=='A');
		sum[i][1]=sum[i-1][1]+(s[i]=='B');
		sum[i][2]=sum[i-1][2]+(s[i]=='C');
		h[i]=1ll*n*(sum[i][1]-sum[i][0])+(sum[i][2]-sum[i][1]);
	}
	h[++n]=0;
	stable_sort(h+1,h+1+n);
	for(register int i=1,j;i<=n;i=j+1)
	{
		for(j=i;j<n&&h[i]==h[j+1];j++);
		ans+=(j-i+1)*(j-i)/2;
	}
	printf("%lld",ans);
}

昨天看了$zyc$的D1T2题解后，看到这道题直接就一眼正解了，看来题目的狩猎还是非常重要的