拓展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("
");
	//for(int i=1;i<=n;i++) printf("%d ",extend[i]);printf("
");
}//对于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
",sm,sc,sum);
	return 0;	
}

拓展kmp(带注释版)

相关推荐