LOJ2507 CEOI2011 Matching

题目链接

参考了 神仙yyb的博客

现在发现kmp不仅能匹配字符串，还可以用于处理任意模式匹配中的状态，如这题中已经匹配的序列中的数的大小关系就是一种状态，使用kmp找到模式序列的每一个前缀的border，即一个最长的前缀和后缀使得它们的数字大小关系相同，即匹配状态相同，然后在失配时跳转到一个前缀的状态继续匹配，就可以解决这个问题啦！

代码：

 #include<bits/stdc++.h>
 using namespace std;
 #define N 1000007
 int sum[N],n,m;
 void modify(int x,int d)
 {
     for(;x<=m;x+=x&-x)
         sum[x]+=d;
 }
 int query(int x)
 {
     int ans=;
     for(;x;x-=x&-x)
         ans+=sum[x];
     return ans;
 }
 int read()
 {
     char c;
     int x;
     while((c=getchar())<||c>);
     x=c-;
     while((c=getchar())>=&&c<=)
         x=x*+c-;
     return x;
 }
 int rk[N],nxt[N],a[N],b[N],p[N];
 vector<int> ans;
 int main()
 {
     int i;
     scanf("%d%d",&n,&m);
     for(i=;i<=n;i++)
         p[read()]=i;
     for(i=;i<=n;i++)
     {
         modify(p[i],);
         rk[i]=query(p[i]);
     }
     memset(sum,,sizeof(sum));
     nxt[]=;
     int last=;
     for(i=;i<=n;i++)
     {
         int len=nxt[i-];
         while(len&&query(p[i])+!=rk[len+])
         {
             len=nxt[len];
             while(last<i-len)modify(p[last++],-);
         }
         if(query(p[i])+==rk[len+])
         {
             nxt[i]=len+;
             modify(p[i],);
         }
         else nxt[i]=;
     }
     for(i=;i<=m;i++)
         a[i]=b[i]=read();
     sort(b+,b+m+);
     for(i=;i<=m;i++)a[i]=lower_bound(b+,b+m+,a[i])-b;
     memset(sum,,sizeof(sum));
     int len=;
     last=;
     for(i=;i<=m;i++)
     {
         while(len&&query(a[i])+!=rk[len+])
         {
             len=nxt[len];
             while(last<i-len)modify(a[last++],-);
         }
         if(query(a[i])+==rk[len+])
         {
             len++;
             modify(a[i],);
             if(len==n)ans.push_back(i-len+);
         }
         else len=;
     }
     int s=ans.size();
     printf("%d\n",s);
     for(i=;i<s;i++)
         printf("%d ",ans[i]);
     printf("\n");
     return ;
 }