考研路茫茫——单词情结
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 3322 Accepted Submission(s): 943
Total Submission(s): 3322 Accepted Submission(s): 943
Problem Description
背单词,始终是复习英语的重要环节。在荒废了3年大学生涯后,Lele也终于要开始背单词了。
一天,Lele在某本单词书上看到了一个根据词根来背单词的方法。比如"ab",放在单词前一般表示"相反,变坏,离去"等。
于是Lele想,如果背了N个词根,那这些词根到底会不会在单词里出现呢。更确切的描述是:长度不超过L,只由小写字母组成的,至少包含一个词根的单词,一共可能有多少个呢?这里就不考虑单词是否有实际意义。
比如一共有2个词根 aa 和 ab ,则可能存在104个长度不超过3的单词,分别为
(2个) aa,ab,
(26个)aaa,aab,aac...aaz,
(26个)aba,abb,abc...abz,
(25个)baa,caa,daa...zaa,
(25个)bab,cab,dab...zab。
这个只是很小的情况。而对于其他复杂点的情况,Lele实在是数不出来了,现在就请你帮帮他。
一天,Lele在某本单词书上看到了一个根据词根来背单词的方法。比如"ab",放在单词前一般表示"相反,变坏,离去"等。
于是Lele想,如果背了N个词根,那这些词根到底会不会在单词里出现呢。更确切的描述是:长度不超过L,只由小写字母组成的,至少包含一个词根的单词,一共可能有多少个呢?这里就不考虑单词是否有实际意义。
比如一共有2个词根 aa 和 ab ,则可能存在104个长度不超过3的单词,分别为
(2个) aa,ab,
(26个)aaa,aab,aac...aaz,
(26个)aba,abb,abc...abz,
(25个)baa,caa,daa...zaa,
(25个)bab,cab,dab...zab。
这个只是很小的情况。而对于其他复杂点的情况,Lele实在是数不出来了,现在就请你帮帮他。
Input
本题目包含多组数据,请处理到文件结束。
每组数据占两行。
第一行有两个正整数N和L。(0<N<6,0<L<2^31)
第二行有N个词根,每个词根仅由小写字母组成,长度不超过5。两个词根中间用一个空格分隔开。
每组数据占两行。
第一行有两个正整数N和L。(0<N<6,0<L<2^31)
第二行有N个词根,每个词根仅由小写字母组成,长度不超过5。两个词根中间用一个空格分隔开。
Output
对于每组数据,请在一行里输出一共可能的单词数目。
由于结果可能非常巨大,你只需要输出单词总数模2^64的值。
由于结果可能非常巨大,你只需要输出单词总数模2^64的值。
Sample Input
2 3 aa ab 1 2 a
Sample Output
104 52
Author
linle
分析:和poj2778很相似,poj2778求的是不包含的,这里要求求包含的,所以用总数减去不包含的就是包含的了,这里要求模2的64次方,直接用unsigned __int64就可以了,这个越界了相当于自动取模2的64次方,再根据矩阵乘法的特性,计算就是。。。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
#define LL unsigned __int64
struct node
{
node *fail;
node *next[26];
node *jump[26];
int cnt;
};
node root[33];
queue<node *> q;
int num,n,m;
char s[10];
void insert(char *str) //插入字符
{
int i=0,index;
node *p=root;
while(str[i])
{
index=str[i]-'a';
if(p->next[index]==NULL)
{
p->next[index]=root+num;
memset(root+num,0,sizeof(struct node));
num++;
}
p=p->next[index];
i++;
}
p->cnt=1; //每个字符串的结尾cnt为1
}
void build_ac_autuomation()
{
int i;
q.push(root);
root->fail=NULL;
while(!q.empty())
{
node *temp=q.front();
q.pop();
for(i=0;i<26;i++)
{
node *p=temp->fail; //p是temp沿失败指针返回的位置
while(p!=NULL&&p->next[i]==NULL) //返回的位置后面没有节点了,再沿他的失败指针返回
p=p->fail;
if(temp->next[i]!=NULL)
{
if(p==NULL) //p为空,连到根节点
temp->next[i]->fail=root;
else
{
temp->next[i]->fail=p->next[i]; //不为空,相同的字符连起来
if(p->next[i]->cnt) //fail指针指向的危险,所以原节点危险,fail代表从根节点到这个节点的最长后缀
temp->next[i]->cnt=1;
}
q.push(temp->next[i]);
temp->jump[i]=temp->next[i];
}
else
{
if(p==NULL)
temp->jump[i]=root; //失败指针指向的是空指针,就相当于指向根节点,所以这个节点跳转到根节点
else
temp->jump[i]=p->next[i]; //不为空,跳转到失败指针指向的那个节点
}
}
}
}
//下面是关于矩阵的运算
struct Matrix
{
LL mat[26*2][26*2];
void init()
{
for(int i=0;i<26*2;i++)
for(int j=0;j<26*2;j++)
mat[i][j]=0;
}
};
int len;
Matrix e;
Matrix mul(Matrix a,Matrix b)
{
Matrix res;
int i,j,k;
for(i=0;i<2*len;i++)
for(j=0;j<2*len;j++)
{
res.mat[i][j]=0;
for(k=0;k<2*len;k++)
res.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
}
return res;
}
LL solve(Matrix a,int k)
{
Matrix b,ans=e;
int i,j;
b.init();
for(i=0;i<len;i++)
for(j=0;j<len;j++)
b.mat[i][j]=a.mat[i][j];
for(i=0;i<len;i++)
for(j=len;j<2*len;j++)
if(i+len==j)
b.mat[i][j]=1;
for(i=len;i<2*len;i++)
for(j=len;j<2*len;j++)
if(i==j)
b.mat[i][j]=1;
k++;
while(k)
{
if(k&1)
ans=mul(ans,b);
b=mul(b,b);
k/=2;
}
LL res=0;
for(i=0;i<len;i++)
res+=ans.mat[i][len];
return res-1;
}
LL sum(LL a,int k)
{
Matrix b,ans=e;
b.init();
b.mat[0][0]=26;
b.mat[0][1]=1;
b.mat[1][1]=1;
b.mat[1][0]=0;
k++;
while(k)
{
if(k&1)
ans=mul(ans,b);
b=mul(b,b);
k/=2;
}
return ans.mat[0][1]-1;
}
int main()
{
int i,j,k;
for(i=0;i<26*2;i++)
for(j=0;j<26*2;j++)
if(i==j)
e.mat[i][j]=1;
while(scanf("%d%d",&n,&m)==2)
{
num=1;
memset(root,0,sizeof(struct node));
for(i=0;i<n;i++)
{
scanf("%*c%s",s);
insert(s);
}
build_ac_autuomation();
int id=0,id1;
Matrix res;
for(i=0;i<num;i++)
{
if(root[i].cnt==0) //节点本身不危险才进行判断
{
id1=0;
for(j=0;j<num;j++)
{
if(root[j].cnt==0) //这个节点也不危险
{
LL count=0;
for(k=0;k<26;k++) //查询j节点中有多少个节点能跳转到i;
{
if(root[j].jump[k]==root+i)
count++;
}
res.mat[id][id1]=count; //id同i节点,id1同j节点
id1++;
}
}
id++;
}
}
len=id;
LL all=sum(26,m); //计算m长度总共有多少种序列,26^1+26^2+.....+26^m;
LL left=solve(res,m); //计算不包含的序列总数
all-=left;
printf("%I64u\n",all);
}
return 0;
}