poj 3735 Training little cats 构造矩阵

g i 让第i只猫得到一个豆

e i  让第i只猫吃掉所有的豆

s i j 让第i，j只猫交换豆

给出n只猫，k条指令，所有指令重复m次

因为是反复重复，所以可以用矩阵乘法，构造矩阵，利用矩阵的快速幂

因为有加法（得到一个豆），所以得构造（n+1）*（n+1）的矩阵，最后一阶为1

注意：矩阵乘法时，矩阵内有很多0，所有可以判断一下，在决定是否计算，否者会超时

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=110;
int n=maxn,m,k;
int nn,mm,kk;
typedef struct
{
    long long m[maxn][maxn];
} Matrax;
Matrax a,per;
Matrax multi(Matrax a,Matrax b) //矩阵乘法，(a*b)%m
{
    Matrax c;
    for(int i=0;i<n;i++) for(int j=0;j<n;j++) c.m[i][j]=0;
    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++)
                    if (b.m[k][j])
                        c.m[i][j]+=a.m[i][k]*b.m[k][j];
        }
    }
    return c;
}
Matrax power(Matrax a,int k) //矩阵快速幂，(a^k)%m
{
    Matrax p,ans;
    p=a;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++) ans.m[i][j]=(i==j);//初始化ans为单位矩阵
    while(k)
    {
        if(k&1)
        {
            ans=multi(ans,p);
            k--;
        }
        k/=2;
        p=multi(p,p);
    }
    return ans;
}
void sett(Matrax &a)
{
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            a.m[i][j]=0;
    for(int i=0;i<n;i++) a.m[i][i]=1;
}
int main()
{
    std::ios::sync_with_stdio(false);
    while(cin>>nn>>mm>>kk)
    {
        if(nn==0&&mm==0&&kk==0) break;
        n=nn+2;
        Matrax ans,a;
        sett(ans);
        for(long long i=0;i<kk;i++)
        {
            sett(a);
            char s;
            cin>>s;
            long long e,f;
            if(s=='g')
            {
                cin>>e;
                a.m[nn][e-1]=1;
                ans=multi(ans,a);
                continue;
            }
            if(s=='e')
            {
                cin>>e;
                a.m[e-1][e-1]=0;
                ans=multi(ans,a);
                continue;
            }
            else
            {
                cin>>e>>f;
                a.m[e-1][e-1]=0;
                a.m[f-1][e-1]=1;
                a.m[f-1][f-1]=0;
                a.m[e-1][f-1]=1;
                ans=multi(ans,a);
                continue;
            }
        }
        Matrax out;
        for(int i=0;i<nn;i++) for(int j=0;j<nn;j++) out.m[i][j]=0;
        out.m[0][nn]=1;
        ans=power(ans, mm);
        out=multi(out,ans);
        for(int i=0;i<nn;i++)
        {
            cout<<out.m[0][i]<<" ";
        }
        cout<<endl;
    }
    return 0;
}