[poj 3261]Milk Patterns

后缀数组搞一下就可以了喵~

其实这道题的第一个想法是 SAM ，建完后缀自动机后拓扑排序跑一遍统计下每个子串的出现次数就 O(N) 就妥妥过掉了

后缀树也是 O(N) 的，统计一下每个节点对应的子树中有多少个叶子节点即可

可是看了一下数据范围，欸呦你妹啊，字符数 1000000+1 是什么心态啊！居然把我的后缀树和后缀 zdj 都卡掉了啊！

 

然后很痛苦的想了想后缀数组的做法:

搞出 sa, rank, height 三个数组后，跑一遍二分，用一个 tot 表示找到了几个子串

二分答案 L，O(N) 扫个遍如果 height[i]>=L 说明又找到一个子串，tot++，否则重新计算

不过注意这里 tot 的初始值是 1 而不是 0 

不过这么个写法怎么写都是 O(NlogN) 的了，如果有更好的写法，请教我做人喵~

 

#include <cstdio>
#include <cstring>
const int sizeOfN=20002;
const int sizeOfType=1200021;

namespace IOspace
{
	inline int getint()
	{
		register int num=0;
		register char ch;
		do ch=getchar(); while (ch<'0' || ch>'9');
		do num=num*10+ch-'0', ch=getchar(); while (ch>='0' && ch<='9');
		return num;
	}
	inline void putint(int num)
	{
		char stack[15];
		register int top=0;
		if (num==0) stack[top=1]='0';
		for ( ;num;num/=10) stack[++top]=num%10+'0';
		for ( ;top;top--) putchar(stack[top]);
		putchar('\n');
	}
}

int N, K;
int A[sizeOfN];

namespace suffixArray
{
	int sa[sizeOfN], rank[sizeOfN], height[sizeOfN];

	inline void swap(int *& x, int *& y) {int * t=x; x=y; y=t;}
	inline void buildArray()
	{
		static int t[2][sizeOfN];
		static int c[sizeOfType];
		int * x=t[0], * y=t[1];
		int M=sizeOfType;

		for (int i=0;i<M;i++) c[i]=0;
		for (int i=0;i<N;i++) c[x[i]=A[i]]++;
		for (int i=1;i<M;i++) c[i]+=c[i-1];
		for (int i=N-1;i>=0;i--) sa[--c[x[i]]]=i;

		for (int j=0, k=1;j<N;M=j, k<<=1)
		{
			j=0;
			for (int i=N-k;i<N;i++) y[j++]=i;
			for (int i=0;i<N;i++) if (sa[i]>=k) y[j++]=sa[i]-k;

			for (int i=0;i<M;i++) c[i]=0;
			for (int i=0;i<N;i++) c[x[y[i]]]++;
			for (int i=1;i<M;i++) c[i]+=c[i-1];
			for (int i=N-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];

			swap(x, y);
			j=1; x[sa[0]]=0;
			for (int i=1;i<N;i++)
				x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k]?j-1:j++;
		}
	}
	inline void calcHeight()
	{
		int l=0;
		for (int i=0;i<N;i++) rank[sa[i]]=i;
		for (int i=0;i<N;i++)
		{
			int j=sa[rank[i]-1];
			for (l=l?l-1:0;A[i+l]==A[j+l];l++);
			height[rank[i]]=l;
		}
	}
}

inline int search();

int main()
{
	N=IOspace::getint(), K=IOspace::getint();
	for (int i=0;i<N;i++) A[i]=IOspace::getint()+1;
	A[N++]=0;

	suffixArray::buildArray();
	suffixArray::calcHeight();

	IOspace::putint(search());

	return 0;
}
inline bool check(int l)
{
	int tot=1;
	for (int i=1;i<N;i++)
		if (suffixArray::height[i]>=l)
		{
			if (++tot>=K)
				return true;
		}
		else
			tot=1;

	return false;
}
inline int search()
{
	int L=0, R=N, M;

	while (L+1<R)
	{
		M=(L+R)>>1;
		if (check(M)) L=M;
		else R=M;
	}

	return L;
}

 

机房里把 插入代码(推荐) 给卡了真是 sang xin bing kuang！我要申诉！