东大OJ ：49 DG之童年的游戏

最新推荐文章于 2022-09-18 15:32:53 发布

蝶翼的罪

最新推荐文章于 2022-09-18 15:32:53 发布

阅读量863

点赞数

分类专栏：数学基础文章标签： ACM 算法矩阵快速幂 DG之童年的游戏

本文链接：https://blog.csdn.net/dieyidezui/article/details/26362115

版权

数学基础专栏收录该内容

3 篇文章 0 订阅

订阅专栏

原题链接：http://acm.seu.edu.cn/oj/problem.php?id=38

分析：矩阵快速幂。注意细节即可。

代码：

#include<cstdio>
#include<vector>
using namespace std;
typedef long long LL;
typedef vector<LL> vec;
typedef vector<vec> mat;
mat mul(const mat &A,const mat &B){
    mat C(A.size(),vec(B[0].size()));
    for(int i = 0; i < A.size(); i++){
        for(int k = 0; k < B.size();k++){
            for(int j = 0; j < B[0].size(); j++)
                C[i][j] += A[i][k] * B[k][j];
        }
    }
    return C;
}
mat pow(mat A, LL n){
    mat B(A.size(),vec(A.size()));
    for(int i = 0; i < A.size(); i++) B[i][i] = 1;
    while(n){
       if(n&1) B = mul(B, A);
       A = mul(A, A);
       n >>= 1;
    }
    return B;
}
int main(){
    int n,m,k,a,b;
    char s[2];
    while(scanf("%d%d%d", &n, &m, &k), n){
        mat A(n+1, vec(n+1));
        mat B(n+1, vec(n+1));
        for(int i = 0; i <= n; i++) A[i][i] = B[i][i] = 1;
        while(k--){
            scanf("%s", s);
            switch(*s){
                case 'a':
                    scanf("%d", &a);
                    A[a-1][n] = 1;
                    B = mul(A, B);
                    A[a-1][n] = 0;
                    break;
                case 'b':
                    scanf("%d", &a);
                    A[a-1][a-1] = 0;
                    B = mul(A, B);
                    A[a-1][a-1] = 1;
                    break;
                case 'c':
                    scanf("%d%d", &a, &b);
                    A[a-1][a-1] = A[b-1][b-1] = 0;
                    A[b-1][a-1] = A[a-1][b-1] = 1;
                    B = mul(A, B);
                    A[a-1][a-1] = A[b-1][b-1] = 1;
                    A[b-1][a-1] = A[a-1][b-1] = 0;
                    break;
            }
        }
        B = pow(B, m);
     //   for(int i = 0; i <= n; i++) printf("%d ",B[i][n]);printf("\n");
        printf("%d",B[0][n]); for(int i = 1; i < n; i++) printf(" %d", B[i][n]); printf("\n");
    }
    return 0;
}