pku 3735 Training little cats (矩阵幂乘)

题目来源：http://acm.pku.edu.cn/JudgeOnline/problem?id=3735

解题思路：

取向量b表示最后的结果，我这里对于n个数，构造n+1维的向量，目的是有利于后面的‘g’操作

 

对于每一次的操作用矩阵表示，这里的矩阵a是（n+1，n+1），初始化为一个单位矩阵

‘g’ num 操作，表示使第num个数增加1，使矩阵a的第num行最后一个元素增加1，即a[num][last]+=1

‘e’ num 操作，表示使第num个数变为0，使矩阵a的第num行全部变为0

即for(i=1:i<=last;i++) a[num][i]=0;

‘s’ num1，num2操作，表示使第num1和num2元素交换位置，使矩阵a 的第num1和num2行互换就可以了for(i=1:i<=last;i++) swap(a[num1][i],a[num2][i]);

 

然后计算a的m次方

最后再与初始化的b向量进行一次乘法，得到的向量的1到last-1即为所要求的结果

 

该题注意的地方：

1：矩阵乘法中做乘法前判断两个乘数是否均为0

2：构造矩阵a的过程，开始我是每进行一次操作就用一次矩阵乘法，导致超时，其实只要每次修改下矩阵就可以了

3：注意最后结果可能超过int的范围，要用__int64

看status里还有几个是0ms过的，不知道他们是用什么方法，还请知道的大牛指教！

 

#include <string.h> #include <stdio.h> const int s=102; int size; //n m k __int64 temp1[s][s]; __int64 temp2[s][s]; //a*b=c a,b均是矩阵 void multply1(__int64 a[][s],__int64 b[][s],__int64 c[][s]) { for(int i=1;i<=size;i++) for(int j=1;j<=size;j++) { c[i][j]=0; for(int k=1;k<=size;k++) if(a[i][k]&&b[k][j]) c[i][j]+=a[i][k]*b[k][j]; } } //a*b=c a是矩阵，b是向量 void multply2(__int64 a[][s],__int64 b[],__int64 c[]) { for (int i=1;i<=size;i++) { c[i]=0; for(int k=1;k<=size;k++) if(a[i][k]&&b[k]) c[i]+=a[i][k]*b[k]; } } //b->a void copy(__int64 a[][s],__int64 b[][s]) { for(int i=1;i<=size;i++) for (int j=1;j<=size;j++) { a[i][j]=b[i][j]; } } //a^m a是矩阵 void chang(__int64 a[][s],__int64 m,__int64 b[][s]) { if(m>0) { //int temp[s][s]; chang(a,m/2,b); //display(b); copy(temp1,b); copy(temp2,b); multply1(temp1,temp2,b); if(m%2!=0) { copy(temp1,b); multply1(temp1,a,b); } } else { //memset(b,0,sizeof(b)); 为什么这里用这个不可以！ 用了b没有被初始化，不知道为什么 for(int ii=1;ii<=size;ii++) for(int iii=1;iii<=size;iii++) b[ii][iii]=0; for (int i=1;i<=size;i++) { b[i][i]=1; } } } void swap(__int64 &x,__int64 &y) { __int64 temp=x; x=y; y=temp; } int main() { int n,k; __int64 m; char ch; while (scanf("%d%I64d%d",&n,&m,&k) && n+m+k) { size=n+1; //初始化ans __int64 ans[s][s]; memset(ans,0,sizeof(ans)); for(int i=1;i<=size;i++) ans[i][i]=1; while(k--) { getchar(); scanf("%c",&ch); int mult[s][s]; //init(ch,mult); int num; if (ch=='g') { scanf("%d",&num); ans[num][size]+=1; //display(mult); } else if(ch=='s') { int num1,num2; scanf("%d%d",&num1,&num2); for (int j3=1;j3<=size;j3++) swap(ans[num1][j3],ans[num2][j3]); //display(mult); } else if(ch=='e') { scanf("%d",&num); for(int jjj=1;jjj<=size;jjj++) ans[num][jjj]=0; //display(mult); } } __int64 an[s][s]; __int64 one[s]; __int64 two[s]; chang(ans,m,an); //其实下面的多余，an矩阵的最后一列就是答案，可以改进 int jj; for (jj=1;jj<size;jj++) one[jj]=0; one[jj]=1; multply2(an,one,two); printf("%I64d ",two[1]); for(int j=2;j<size;j++) printf("%I64d ",two[j]); printf("/n"); } }