pku3735（矩阵3）

http://162.105.81.212/JudgeOnline/problem?id=3735

 

这个题也是一个关于矩阵的一个经典的应用。

（转自matrix67）

    有n只猫猫排成一排，初始时每只猫猫都没有花生。定义三种操作：给第i只猫猫一颗花生，令第i只猫猫吃掉它所有的花生，交换猫猫i和猫猫j的花生。给出长度不超过k的操作序列，输出循环执行m次操作序列后的结果。数据规模n≤100，k≤100，m≤1 000 000 000。看到这道题我们立马会想到矩阵乘法。如果能为每种操作构造出一个矩阵，m再大二分下来也能承受。我们可以先把序列中的k个操作乘起来，再利用二分自乘m次就可以了。

 

解题思路：

取向量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即为所要求的结果

下面的代码是把矩阵翻转一下写的：

#include<iostream> using namespace std; const int maxn = 110; int M, N, K; __int64 A[maxn][maxn], B[maxn][maxn]; void Mul(__int64 a[][maxn], __int64 b[][maxn]) //矩阵a * 矩阵b； { __int64 tmp[maxn][maxn] = {0}; int i, j, k; for(i=0; i<=N; i++) for(j=0; j<=N; j++) { if(a[i][j]) for(k=0; k<=N; k++) tmp[i][k] += a[i][j] * b[j][k]; //翻转90度的矩阵相乘； } for(i=0; i<=N; i++) for(j=0; j<=N; j++) a[i][j] = tmp[i][j]; //结果保存在矩阵a中； } void MatrixPow(__int64 tmp[][maxn], __int64 a[][maxn], int m) //二分求矩阵A^m; { while(m > 0) //把m转化为二进制； { if(m & 1) Mul(tmp, a); Mul(a, a); m >>= 1; } } void Swap(__int64 &a, __int64 &b) { __int64 c = a; a = b; b = c; } int main() { int i, j, num1, num2; char ch[2]; while(scanf("%d%d%d",&N, &M, &K) && N+M+K) { for(i=0; i<=N; i++) for(j=0; j<=N; j++) A[i][j] = B[i][j] = (i == j);//单位矩阵； while(K--) { scanf("%s",ch); if(ch[0] == 'g') { scanf("%d",&num1); //此解法是把矩阵转90度； A[0][num1]++; } else if(ch[0] == 'e') { scanf("%d",&num1); for(i=0; i<=N; i++) A[i][num1] = 0; } else if(ch[0] == 's') { scanf("%d%d",&num1, &num2); for(i=0; i<=N; i++) Swap(A[i][num1], A[i][num2]); } } MatrixPow(B, A, M); for(i=1; i<N; i++) printf("%I64d ", B[0][i]); printf("%I64d/n",B[0][N]); } return 0; } 

下面再贴个死活过不了的代码；还一直RE的~~还有超时的：矩阵没有翻转的；

#include<iostream> using namespace std; const int MAX = 105; int n, m, k; typedef struct M { __int64 matrix[MAX][MAX]; //不知道为什么这里用__int64，用vc编译回RE啊？经过的大牛帮忙说说~ }Matrix; Matrix A, B; Matrix Mul(Matrix a, Matrix b) //矩阵相乘 { Matrix c; int i, j, t; for(i=1; i<=n; i++) for(j=1; j<=n; j++) { c.matrix[i][j] = 0; if(a.matrix[i][j]) for(t=1; t<=n; t++) c.matrix[i][j] += a.matrix[i][t] * b.matrix[t][j]; } return c; } Matrix Cal(int exp) //矩阵求幂； { Matrix p, q; p = A; q = B; while(exp != 1) { if(exp & 1) { exp--; q = Mul(p, q); } else { exp >>= 1; p = Mul(p, p); } } p = Mul(p, q); return p; } int main() { Matrix ans; __int64 tmp; int num1, num2, i, j; char ch[2]; while(scanf("%d%d%d",&n, &m, &k) && n+m+k) { n++; for(i=1; i<=n; i++) for(j=1; j<=n; j++) A.matrix[i][j] = B.matrix[i][j] = (i == j); while(k--) { scanf("%s",&ch); //getchar(); if(ch[0] == 'g') { scanf("%d",&num1); A.matrix[num1][n]++; } else if(ch[0] == 'e') { scanf("%d",&num1); for(i=1; i<=n; i++) A.matrix[num1][i] = 0; } else if(ch[0] == 's') { scanf("%d%d",&num1, &num2); for(i=1; i<=n; i++) { tmp = A.matrix[num1][i]; A.matrix[num1][i] = A.matrix[num2][i]; A.matrix[num2][i] = tmp; } } } ans = Cal(m); ans = Mul(ans, B); printf("%I64d",ans.matrix[1][n]); for(i=2; i<n; i++) printf(" %I64d",ans.matrix[i][n]); printf("/n"); } return 0; }