LightOJ 1268 Unlucky Strings 【KMP+矩阵】

http://www.lightoj.com/volume_showproblem.php?problem=1268

题目大意：

给出一个字符集和一个字符串和正整数n，问由给定字符集组成的所有长度为n的串中不以给定字符串为连续子串的有多少个？

字符集大小最大50，字符串最大50，n最大10^9。最后结果模2^32

解题思路：

设给定字符串为str。

首先必须明确的是解题的主要思路是矩阵快速幂，因为n太大了，囧、

然后要考虑的是矩阵表示的什么。设矩阵mat

我们设mat[i][j]表示从原来str的第i个字符匹配到第j个字符有多少种方案。

按照KMP求出的next数组可以构造出来初始的矩阵，即只“经过”一个字符时的结果。

然后矩阵快速幂求出来mat^n。这个矩阵的响应项表示就是经过了n个字符后的方案数。

我们要求的是从开始（字符0）匹配到的结果，那么答案就是矩阵的第一行的所有值之和。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int next[51];
char str[51];
int app[255];
void get_next(){
	int len=strlen(str),k;
	memset(next,0,sizeof(next));
	for(int i=1;i<len;i++){
		k=next[i-1];
		while(k&&str[k]!=str[i])k=next[k-1];
		if(str[k]==str[i])k++;
		next[i]=k;
	}
}
char s[51];
struct matrix{
	unsigned int a[50][50];
	void init(){
		memset(a,0,sizeof(a));
		for(int i=0;i<50;i++)a[i][i]=1;
	}
	void set_zero(){
		memset(a,0,sizeof(a));
	}
};
matrix mul (matrix &a,matrix &b){
	matrix res;
	res.init();
	for(int i=0;i<50;i++)for(int j=0;j<50;j++){
		unsigned int x=0;
		for(int k=0;k<50;k++){
			x+=a.a[i][k]*b.a[k][j];
		}
		res.a[i][j]=x;
	}
	return res;
}
matrix pow(matrix &a,int p){
	matrix res,b=a;
	res.init();
	while(p){
		if(p&1){
			res=mul(res,b);
		}
		p/=2;
		b=mul(b,b);
	}
	return res;
}
int main(){
	int T,ca=1;
	int count;
	scanf("%d",&T);
	int len;
	while(T--){
		count=0;
		memset(app,0,sizeof(app));
		printf("Case %d: ",ca++);
		scanf("%d",&len);
		scanf("%s",s);
		scanf("%s",str);
		get_next();
		int n=strlen(str);
		for(int i=0;s[i];i++){
			if(app[s[i]]==0)count++;
			app[s[i]]++;
		}
		matrix a,res;
		a.set_zero();
		for(int i=0;i<255;i++){
			if(app[i]){
				if(i==str[0])a.a[0][1]++;
				else a.a[0][0]++;
			}
		}
		for(int i=1;i<n;i++){
			for(int j=0;j<255;j++) if(app[j]){ 
				int k=i;
				while(k&&str[k]!=j)k=next[k-1];
				if(j==str[k]&&k==n-1);
				else if(j==str[k])a.a[i][k+1]++;
				else a.a[i][k]++;
			}
		}
		res=pow(a,len-1);
		res=mul(a,res);
		unsigned int ans=0;
		for(int i=0;i<n;i++)ans+=res.a[0][i];
		printf("%u\n",ans);
	}
}