hdu 2429 Word Game

先把可以首尾相连的单词间连边，这样问题就转化为在n步以内，从起点到终点走奇数步，一共多少种走法

由矩阵乘法的意义，矩阵A中的aij可以表示从i到j走一步可以有多少种走法，A^2中的aij可以表示从i到j走一步可以有多少种走法，依此类推，A^n中的aij表示走n步有多少种走法

 

那么，问题就转化为S=A+A^3+A^5+...+A^k中，S_se 的值，s,e分别为起点和终点，k为比n小的数中最大的奇数

还是可以二分

f(k)=A+A^3+A^5+...+A^k

k%4=1时，f(k)=f(k-2)+A^k

k%4=3时，f(k)=f(k/2)+f(k/2)*A^(k/2)

 

一开始把数组开大了，导致了栈溢出，还有就是_pow函数里矩阵x忘记初始化导致错误，这些都要注意

 

代码：

#include<iostream> #include<memory.h> #include<string> #include<cstdio> #include<algorithm> #include<math.h> #include<stack> #include<queue> using namespace std; #define M 10001 const int MAX=31; int len[MAX],n,m; char ss[MAX][20]; struct Mat { int a[MAX][MAX]; Mat operator * (const Mat &x) { Mat tmp; int i,j,k; for(i=0;i<m;i++) { for(j=0;j<m;j++) { tmp.a[i][j]=0; for(k=0;k<m;k++) { tmp.a[i][j]+=a[i][k]*x.a[k][j]; tmp.a[i][j]%=M; } } } return tmp; } Mat operator +(const Mat &x) { Mat tmp; int i,j; for(i=0;i<m;i++) { for(j=0;j<m;j++) { tmp.a[i][j]=(a[i][j]+x.a[i][j])%M; } } return tmp; } }p; int find(char str[]) { for(int i=0;i<m;i++) { if(strcmp(str,ss[i])==0) { return i; } } } Mat _pow(int k) { if(k==1) return p; else { Mat x,y; for(int i=0;i<m;i++) { for(int j=0;j<m;j++) { x.a[i][j]=0; y.a[i][j]=p.a[i][j]; } x.a[i][i]=1; } while(k) { if(k&1) { x=x*y; } y=y*y; k>>=1; } return x; /*if(k&1) { return _pow(k-1)*p; } else { Mat tmp=_pow(k/2); return tmp*tmp; }*/ } } inline Mat cal(int k) { //cout<<"k="<<k<<endl; if(k==1) return p; else { if(k%4==1) { //return cal(k-2)+_pow(k); Mat tmp=cal((k-2)/2); return tmp+tmp*_pow(k/2)+_pow(k); } else { Mat tmp=cal(k/2); return tmp+tmp*_pow(k/2+1); } } } int main() { int i,j,T,s,t; scanf("%d",&T); char word[50]; while(T--) { scanf("%d",&m); memset(p.a,0,sizeof(p.a)); for(i=0;i<m;i++) { scanf("%s",&ss[i]); len[i]=strlen(ss[i]); } scanf("%s",&word); s=find(word); scanf("%s",&word); t=find(word); scanf("%d",&n); for(i=0;i<m;i++) { for(j=0;j<m;j++) { if(ss[i][len[i]-1]==ss[j][0]) { p.a[i][j]++; } } } if(n%2==0) n--; Mat res=cal(n); printf("%d/n",res.a[s][t]); } return 0; }