HDU 1711 Number Sequence---KMP原始
#include<stdio.h> #include<string.h> #include<math.h> #include<stdlib.h> #include<vector> #include<queue> #include<stack> #include<iostream> #include<string> #include<algorithm> using namespace std; #define MaxSize 1000005 #define inf 0x3f3f3f3f #define LL long long int int n,m; int ori[MaxSize],tar[MaxSize],NEXT[MaxSize];//有点醉,本来写的next和algorithm库里面的东西重名CE了。所以用不着的头文件别乱写。 void get_NEXT() { int i=0;//i是指在目标串中走到了i位 int j=NEXT[0]=-1;//把这里设置成-1,而不是0,是非常巧妙的。j一直都是NEXT[i] while(i<m)//得到目标串的NEXT,注意范围 { if(j==-1 || tar[j]==tar[i])//j==-1有两种情况,第一个是刚开始从第0位开始的时候,第二个是后面找不到相同的前缀后一位的时候 NEXT[++i] = ++j; else j=NEXT[j]; /*如果一直这样代换下去,走到j=0的时候,然后进行下一次循环:j==-1不满足, 于是就比较tar[0]和tar[i],就是a.....|b的第一位a和现在在i位上的b比较。 如果j==0的时候tar[0]==tar[i]还不满足,说明前面所有的前缀的之后一位都没有和当前i位一样的。 那就应该有NEXT[++i]=0。在代码中,如果j==0的时候tar[0]==tar[i]还不满足,j就会走到-1, (因为我们设置的NEXT[0]=-1)。于是下一步循环的时候,++j就就恰是0,于是NEXT[++i]=0,就是巧妙在这里*/ } } int match() { int i=0;//i是指在原始串中走到了i位置 int j=0;//j是指在目标串中走到了j位置 while(i<n)//在原始串中从头到尾匹配,如果走到最后,过程中都没有return,说明没有找到匹配。最后while结束后就返回-1 { if(j==-1 || ori[i]==tar[j])//一位一位匹配下去 { i ++; j ++; } else//走到某一位的时候发现不能匹配,那么就移位。移位之后,j就跑到前缀的后一位,然后继续往后匹配。当然此时的i是不应该变的。 j=NEXT[j]; /*如果j代换到了j==0,(原始串:....a.....,目标串b.....)那么就是用当前原始串的第i位a比较目标串第一位b 如果还不能匹配,那么j就代换成-1,,原始串的i就到下一位,目标串的j本来是-1,加1之后就变成0,也就是又从第一位开始比较 巧妙的=NEXT[0]=-1。*/ if (j == m) return i-m+1;//匹配完了,j就会走到目标串的最后一位(即m-1位)之后,j++就变成了m,这里就应该return了。 } return -1; } int main() { int T; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); for(int i=0; i<n; i++) scanf("%d",&ori[i]); for(int i=0; i<m; i++) scanf("%d",&tar[i]); get_NEXT(); /*for(int i=0;i<m;i++) printf("%d ",NEXT[i]);printf("\n");*/ printf("%d\n",match()); } return 0; }