【KMP】BZOJ2384[Ceoi2011]Match

分析:

首先,可以转化判定条件:如果在一段数中,每一个数左侧,比它小的数和给出的目标序列相同。那么就能认为这段数的大小关系合法。

然后就可以魔改KMP来做了。

所谓魔改,就是之前本来是判定两个字符是否相同,而这里变为判断当前已加入的左侧的数中,比它小的数是否合法。

但是为了高效修改,需要借助树状数组优化修改。

其余部分和KMP较为类似。

#include
#include
#include
#include
#define SF scanf
#define PF printf
#define MAXN 1000010
using namespace std;
int tr[MAXN],a[MAXN],v[MAXN],s[MAXN],h[MAXN],t[MAXN];
int n,m,tot;
int fail[MAXN],ans[MAXN];
int que(int x){
	int res=0;
	while(x){
		res+=tr[x];
		x-=x&(-x);
	}
	return res;
}
void add(int x,int val){
	while(x<=m){
		tr[x]+=val;
		x+=x&(-x);
	}
}
int main(){
	SF("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		SF("%d",&a[i]);
		s[a[i]]=i;
	}
	for(int i=1;i<=n;i++){
		v[i]=que(s[i]);
		add(s[i],1);
	}
	for(int i=1;i<=m;i++){
		SF("%d",&h[i]);
		t[i]=h[i];
	}
	memset(tr,0,sizeof tr);
	int i=2,j=0;
	while(i<=n){
		while(que(s[i])!=v[j+1]){
			for(int k=i-j;k<i-fail[j];k++)
				add(s[k],-1);
			j=fail[j];
		}
		j++;
		fail[i]=j;
		add(s[i],1);
		i++;
	}
	sort(t+1,t+1+m);
	memset(tr,0,sizeof tr);
	i=1,j=0;
	while(i<=m){
		h[i]=lower_bound(t+1,t+1+m,h[i])-t;	
		while(j==n||que(h[i])!=v[j+1]){
			for(int k=i-j;k<i-fail[j];k++)
				add(h[k],-1);
			j=fail[j];
		}
		j++;
		add(h[i],1);
		if(j==n)
			ans[++tot]=i-n+1;
		i++;
	}
	PF("%d\n",tot);
	for(int i=1;i<=tot;i++)
		PF("%d ",ans[i]);
}

你可能感兴趣的:(树状数组,KMP,字符串)