HDU 2243 考研路茫茫——单词情结(AC自动机)

输出用%I64d会WA。。。。。
矩阵套矩阵实现矩阵等比数列求和

#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
#define maxn 50
#define Maxc 26
#define LL long long
using namespace std;

int tr[maxn][Maxc],fail[maxn],tag[maxn],tot;
inline void Insert(char *s,int u=0)
{	for(int i=0,len=strlen(s),v;i<len;i++) v=s[i]-'a',u=tr[u][v]?tr[u][v]:tr[u][v]=++tot;
	tag[u]=1;}
void Build()
{	queue<int>q;
	for(int i=0;i<Maxc;i++) if(tr[0][i]) q.push(tr[0][i]),fail[tr[0][i]]=0;
	for(int u;!q.empty();)
	{	u=q.front(),q.pop();
		for(int i=0;i<Maxc;i++) 
			if(tr[u][i]) fail[tr[u][i]]=tr[fail[u]][i],tag[tr[u][i]]|=tag[fail[tr[u][i]]],q.push(tr[u][i]);
			else tr[u][i]=tr[fail[u]][i];}}
char s[20];
int cnt=0,id[maxn];
struct Mat
{
	LL a[100][100];
	Mat(){memset(a,0,sizeof a);}
	Mat operator *(const Mat &B)const
	{
		Mat ret;
		for(int i=0;i<cnt*2;i++)
			for(int j=0;j<cnt*2;j++) if(a[i][j])
				for(int k=0;k<cnt*2;k++) if(B.a[j][k])
					ret.a[i][k]+=a[i][j]*B.a[j][k];
		return ret;
	}
}M,R;
int main()
{
	int n,m;
	while(~scanf("%d%d",&n,&m))
	{
		for(int i=0;i<n;i++) scanf("%s",s),Insert(s);
		Build();
		cnt = 0;
		for(int i=0;i<=tot;i++) if(!tag[i]) id[i]=cnt++;
		memset(M.a,0,sizeof M.a);
		for(int i=0;i<=tot;i++) if(!tag[i])
			for(int j=0;j<Maxc;j++) if(!tag[tr[i][j]])
				M.a[id[i]][id[tr[i][j]]]++;
		for(int i=0;i<cnt;i++) M.a[i][i+cnt]=M.a[i+cnt][i+cnt]=1;
		R=M;
		int tmp = m;
		for(;m;m>>=1,M=M*M) if(m&1) R=R*M;
		LL ans = 0;
		for(int i=0;i<cnt;i++)
			ans -= R.a[0][i+cnt];
		cnt=1;
		M.a[0][0]=26,M.a[0][1]=M.a[1][1]=1,M.a[1][0]=0;
		R=M;
		for(;tmp;tmp>>=1,M=M*M) 
			if(tmp&1) R=R*M;
		ans += R.a[0][1];
		printf("%I64d\n",ans);
		for(int i=0;i<=tot;i++)
			memset(tr,0,sizeof tr),fail[i]=tag[i]=0;
		tot = 0;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值