【POJ】【矩阵快速幂】Training little cats

POJ 3735 Training little cats

题目传送门

题目大意

有 $n$ 只猫，初始时各有花生 $0$ 颗，有 $k$ 个命令，每个命令为一下三种之一：

• 给编号为 $i$ 的猫一颗花生；
• 令编号为 $i$ 的猫吃掉所有花生；
• 令编号为 $i,j$ 的猫交换花生。

问将这 $k$ 个命令循环执行 $m$ 轮后每只猫的花生数目。

分析

显然由于这个极大的 $m$ 我们不可能直接模拟。

注意到所有的操作是循环的，这意味着我们可以利用矩阵来进行加速。

由基本的线性代数知识可以知道，第二、三个操作实际上是很好解决的。对于第二个操作，我们仅需将对应的一行全部清空，而第三个操作就交换两行即可解决。

对于第一个操作，我们则可以加一行一列用于辅助，每次仅需对该行上加一即可，如下所示：

假设有三只猫，目前给第三只猫一颗花生
则矩阵为：
1 0 0 0 – 辅助行
0 1 0 0 – 第一只猫
0 0 1 0 – 第二只猫
1 0 0 1 – 第三只猫

同时我们也可分析发现所有操作其实是可以叠加在一起的。这样就可以完成所有操作了。最后仅需将矩阵做 $m$ 次幂即可，答案简单推导一下可以知道在第一列。

参考代码

注意这个题有点卡常，但观察可以发现

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;
const int MaxN = 100;

struct Matrix {
    int n;
    ll a[MaxN + 5][MaxN + 5];
    inline Matrix() {n = 0, memset(a, 0, sizeof a);}
    inline Matrix(int x) {
        n = x;
        memset(a, 0, sizeof a);
        for(int i = 0; i <= n; i++)
            a[i][i] = 1;
    }
    Matrix operator * (const Matrix &rhs) const {
        Matrix ret;
        ret.n = n;
        for(int i = 0; i <= n; i++)
            for(int k = 0; k <= n; k++)
                if(a[i][k])
                    for(int j = 0; j <= n; j++)
                        ret.a[i][j] += a[i][k] * rhs.a[k][j];
        return ret;
    }
};

int N;

Matrix QuickPow(Matrix a, ll m) {
    Matrix ret(N);
    while(m) {
        if(m & 1) ret = ret * a;
        a = a * a;
        m >>= 1;
    }
    return ret;
}

int main() {
    while(true) {
        ll m;
        int k;
        scanf("%d %lld %d", &N, &m, &k);
        if(N == 0 && m == 0 && k == 0)
            break;
        Matrix A(N);
        for(int i = 1; i <= k; i++) {
            char opt;
            scanf("\n%c", &opt);
            if(opt == 'g') {
                int x;
                scanf("%d", &x);
                A.a[x][0]++;
            } else if(opt == 'e') {
                int x;
                scanf("%d", &x);
                for(int j = 0; j <= N; j++)
                    A.a[x][j] = 0;
            } else if(opt == 's') {
                int x, y;
                scanf("%d %d", &x, &y);
                for(int j = 0; j <= N; j++)
                    swap(A.a[x][j], A.a[y][j]);
            }
        }
        Matrix t = QuickPow(A, m);
        for(int i = 1; i <= N; i++)
            printf("%lld ", t.a[i][0]);
        puts("");
    }
	return 0;
}