拓展kmp(带注释版)
做了一个多月的网络流耗掉了一堆智商。然而又入新坑字符串= =。
据ljm大神说exkmp能解决kmp能做的和不能做的题(所以他至今还不会kmp),于是懒惰的我直接学exkmpXD。理解了很久啊看了N篇博客,跟着模版打了注释Orz于是还是先记下来忘了再来看好了。
(其实有没有发现代码跟马拉车是差不多的诶ovo
#include<cstdio> #include<cstdlib> #include<cstring> #include<iostream> #include<algorithm> using namespace std; #define Maxn 100100 #define LL long long //20160216 T3 int n; int m[2*Maxn],c[Maxn];//诶设用c来配m啦 int next[Maxn],extend[Maxn]; //设suffix_m(i)表示m[i..len],就是从i开始的后缀啦(suffix_c同理~) //extend[i]表示suffix_m(i)能和c匹配的最大长度 //next[i]表示suffix_c(i)能和c匹配的最大长度 int mymax(int x,int y) {return x>y?x:y;} void exkmp() { int id=1,mx=0;//id为匹配到目前最远位置的i,mx为匹配到的最远位置 memset(next,0,sizeof(next)); next[1]=n,next[2]=0; for(int i=2;i<=n;i++) { if (mx>i+next[i-id+1]-1) next[i]=next[i-id+1];//第一种情况 else //第二种情况:需要继续匹配才能确定next[i]的值 { next[i]=mymax(mx-i+1,0); //next[i]==next[i-id]; while(i+next[i]<=n&&c[i+next[i]]==c[next[i]+1]) next[i]++; if (i+next[i]-1>mx) mx=i+next[i]-1,id=i;//更新 } }//话说如果没有把那个串复制一遍到后面的话,设j=mx-i+1, //要if (j<0) j=0;一下,就是再从头开始匹配 //不过每次mymax(mx-i+1,0)就是下面那样就不用管这些啊↑ id=0,mx=0; for(int i=1;i<=n;i++) { if (mx>i+next[i-id+1]-1) extend[i]=next[i-id+1];//第一种情况,搞清一下next[i-id+1]的含义 else//第二种情况:需要继续匹配才能确定extend[i]的值 { extend[i]=mymax(mx-i+1,0); //extend[i]=next[i-id]; while (i+extend[i]<=2*n&&m[i+extend[i]]==c[extend[i]+1]) extend[i]++; if (i+extend[i]-1>mx) mx=i+extend[i]-1,id=i;//更新 } } //for(int i=1;i<=n;i++) printf("%d ",next[i]);printf("\n"); //for(int i=1;i<=n;i++) printf("%d ",extend[i]);printf("\n"); }//对于exkmp,嗯如果还不懂认真看http://blog.csdn.net/dyx404514/article/details/41831947吧 int main() { LL sm=0,sc=0,sum=0; scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&m[i]); for(int i=1;i<=n;i++) scanf("%d",&c[i]); for(int i=n+1;i<=2*n;i++) m[i]=m[i-n]; //我的话这里直接先复制一遍了 如果两个串要互相做一次的话还是不要这样子了 exkmp(); for(int i=1;i<=n;i++) { if(extend[i]>=n) sum+=n; else { sum+=extend[i]+1; if(m[i+extend[i]]>c[extend[i]+1]) sm++; else sc++; } } printf("%I64d %I64d %I64d\n",sm,sc,sum); return 0; }