poj 3735 Training little cats(关键是想到构建中间矩阵)

poj 3735 Training little cats

初始化一个n+1阶的单位矩阵

原始矩阵(三只猫,从上到下分)：

1

an=0

bn=0

cn=0

第一个1是用来响应g操作+1的

g x : a[x][0]++

e x : 将a第x行全部清空

s x y : 将a第x行和y行交换

先遍历一边操作构造出一个中间矩阵a，再计算a^m，和原始矩阵相乘便是答案，由于原始矩阵只有第一个元素为1，则第x只猫直接输出a[x][0]就行了

#include<stdio.h>
#include<string.h>
#define ll long long
#define MAXN 105
struct matrix
{
    ll m[MAXN][MAXN];
};
int n;
void debug(matrix a)
{
    printf("\n");
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=n;j++)
            printf("%lld ",a.m[i][j]);
        printf("\n");
    }
}
matrix multi(const matrix &a,const matrix &b)
{
    matrix ans;
    memset(&ans,0,sizeof(ans));
    for(int i=0;i<=n;i++)
        for(int k=0;k<=n;k++)
            if(a.m[i][k])
            {
                for(int j=0;j<=n;j++)
                    ans.m[i][j]+=a.m[i][k]*b.m[k][j];
            }
    return ans;
}
matrix pow(matrix a,int k)
{
    matrix ans;
    memset(&ans,0,sizeof(ans));
    for(int i=0;i<=n;i++) ans.m[i][i]=1;
    //debug(ans);
    while(k)
    {
        if(k&1) ans=multi(ans,a);
        //debug(ans);
        a=multi(a,a);
        //debug(a);
        k>>=1;
    }
    return ans;
}
void swap(ll &a,ll &b)
{
    ll t=a;
    a=b;
    b=t;
}

int main()
{
    matrix a;
    char op[3];
    int x,y,k,m;
    while(scanf("%d%d%d",&n,&k,&m)!=EOF&&(n+k+m))
    {
        memset(&a,0,sizeof(a));
        for(int i=0;i<=n;i++) a.m[i][i]=1;
        while(m--)
        {
            scanf("%s",op);
            if(op[0]=='g')
            {
                scanf("%d",&x);
                a.m[x][0]++;
            }
            else if(op[0]=='e')
            {
                scanf("%d",&x);
                for(int i=0;i<=n;i++)
                    a.m[x][i]=0;
            }
            else
            {
                scanf("%d%d",&x,&y);
                for(int i=0;i<=n;i++)
                    swap(a.m[x][i],a.m[y][i]);
            }
        }
       // debug(a);
        a=pow(a,k);
        //debug(a);
        printf("%lld",a.m[1][0]);
        for(int i=2;i<=n;i++)
            printf(" %lld",a.m[i][0]);
        printf("\n");
    }
    return 0;
}