「BZOJ 1461」字符串的匹配

传送门

---

problem

数据范围：$n\le 5\times 10^5$，$S\le 10000$。

---

solution

这应该是比较板的题吧。

就是在匹配的时候，把比较两个字符的大小换成比较两个字符排名的大小。

而排名可以用树状数组来搞，那就可以解决这道题了。

然后一定要注意关于边界的细节问题。

---

code

#include<bits/stdc++.h>
#define N 500005
#define lowbit(x) (x&(-x))
using namespace std;
int n,m,s,tot;
int A[N],B[N],bit[N],rk1[N],rk2[N],nxt[N],ans[N];
void add(int x,int val)  {for(;x<=s;x+=lowbit(x))bit[x]+=val;}
int Query(int x,int ans=0)  {for(;x;x-=lowbit(x))ans+=bit[x];return ans;}
bool match(int x,int y){
	return (Query(y)==rk1[x])&&(Query(y-1)==rk2[x]);
}
int main(){
	scanf("%d%d%d",&n,&m,&s);
	for(int i=1;i<=n;++i)  scanf("%d",&A[i]);
	for(int i=1;i<=m;++i)  scanf("%d",&B[i]),add(B[i],1),rk1[i]=Query(B[i]),rk2[i]=Query(B[i]-1);
	memset(bit,0,sizeof(bit));
	for(int i=2,j=0;i<=m;++i){
		add(B[i],1);
		while(j&&!match(j+1,B[i])){
			for(int k=i-j;k<i-nxt[j];++k)  add(B[k],-1);
			j=nxt[j];
		}
		if(match(j+1,B[i]))  ++j;
		nxt[i]=j;
	}
	memset(bit,0,sizeof(bit));
	for(int i=1,j=0;i<=n;++i){
		add(A[i],1);
		while(j&&!match(j+1,A[i])){
			for(int k=i-j;k<i-nxt[j];++k)  add(A[k],-1);
			j=nxt[j];
		}
		if(match(j+1,A[i]))  ++j;
		if(j==m){
			ans[++tot]=i-m+1;
			for(int k=i-j+1;k<=i-nxt[j];++k)  add(A[k],-1);
			j=nxt[j];
		}
	}
	printf("%d\n",tot);
	for(int i=1;i<=tot;++i)  printf("%d\n",ans[i]);
	return 0;
}