首页 > 代码库 > HDU 1711 Number Sequence---KMP原始

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;
}

 

HDU 1711 Number Sequence---KMP原始