POJ 3735 Training little cats（矩阵乘法）

【题目链接】 http://poj.org/problem?id=3735

【题目大意】

　　有一排小猫，给出一系列操作，包括给一只猫一颗花生，
　　让某只猫吃完所有的花生以及交换两只猫的花生，
　　求完成m次操作集合之后每只猫的花生数量

【题解】

　　创建一个1*(n+1)的初始矩阵，
　　对于给第i只猫一个花生就相当于乘上修改了A[0][i]的单位矩阵
　　如给第一只猫一个花生就修改A[0][1]为1
　　得到:
　　　　1 1 0 0
　　　　0 1 0 0
　　　　0 0 1 0
　　　　0 0 0 1
　　然后我们用初始矩阵去乘这个矩阵，就完成了加的操作
　　对于让第i只猫吃完所有的花生，我们将A[i][i]修改为0
　　如让第三只猫吃完花生，那么就相当于将A[3][3]改为0
　　得到
　　　　1 0 0 0
　　　　0 1 0 0
　　　　0 0 1 0
　　　　0 0 0 0
　　然后我们用初始矩阵去乘这个矩阵，就完成了清零的操作
　　对于交换花生的操作，我们交换单位矩阵的第x和y行，然后相乘即可
　　比如交换第一只猫和第三只猫的花生，我们就得到
　　　　1 0 0 0
　　　　0 0 0 1
　　　　0 0 1 0
　　　　0 1 0 0
　　乘到初始矩阵上即可
　　我们发现经过这些操作，我们最后会得到一个操作集合的矩阵表达
　　而这些操作的变换其实不用整个乘上矩阵，只要直接修改单位矩阵的部分值即可
　　第一种操作的影响可以转化为A[0][x]++
　　第二种操作的影响可以转化为A[0~n][x]=0
　　第三种操作的影响可以转化为i_0^n swap(A[i][x],A[i][y])
　　那么我们直接计算转置矩阵即可。

【代码】

#include <cstdio>
#include <vector>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
typedef long long LL; 
typedef vector<vector<LL> > mat;
LL n,m,k,x,y;
char op[10];
mat mul(mat &A,mat &B){
    mat C(A.size(),vector<LL>(B[0].size()));
    rep(i,A.size())rep(k,B.size())if(A[i][k]){
        rep(j,B[0].size())C[i][j]+=A[i][k]*B[k][j];
    }return C;
}
mat pow(mat A,LL n){
    mat B(A.size(),vector<LL>(A.size()));
    rep(i,A.size())B[i][i]=1;
    for(;n;n>>=1){if(n&1)B=mul(B,A);A=mul(A,A);}
    return B;
}
int main(){
    while(scanf("%lld%lld%lld",&n,&m,&k),n+m+k){  
        mat A(n+1,vector<LL>(n+1));
        rep(i,n+1)A[i][i]=1;
        while(k--){
            scanf("%s",&op);
            if(op[0]=='g'){
                scanf("%lld",&x);
                A[0][x]++;
            }else if(op[0]=='e'){
                scanf("%lld",&x);
                rep(i,n+1)A[i][x]=0;
            }else{
                scanf("%lld%lld",&x,&y);
                rep(i,n+1)swap(A[i][x],A[i][y]);
            }
        }if(m){
            A=pow(A,m);
            printf("%lld",A[0][1]);
            for(int i=2;i<=n;i++)printf(" %lld",A[0][i]);
            puts("");
        }else{
            printf("0");
            rep(i,n-1)printf(" 0");
            puts("");
        }
    }return 0;
}