( VIJOS )VOJ 1049 送给圣诞夜的礼品 矩阵快速幂

https://vijos.org/p/1049

 

非常普通的矩阵快速幂...

但是我

第一次写忘了矩阵不能交换律...

第一二次提交RE直到看到题解才发现这道题不能用递归快速幂...

第三次提交成了c编译错误...

第四次提交WA发现写循环快速幂的时候少清零了一个f...

所以提交了五次才终于对了,什么垃圾的代码能力...通过率杀手...

希望下次能捡起我的脑子....

代码很简单 如下

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=110;
const double eps=1e-8;
int n,m;
struct mat{
    int e[maxn][maxn];
};
mat pro(mat x,mat y){
    mat z;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            z.e[i][j]=0;
            for(int k=1;k<=n;k++){
                z.e[i][j]+=x.e[i][k]*y.e[k][j];
            }
        }
    }return z;
}
mat pow(mat x,int k){
    mat z;
    bool f=1;
    while(k>0){
        if(k%2!=0){
            if(f){
                z=x;
                f=0;
            }else{
                z=pro(x,z);
            }
        }k/=2;
        x=pro(x,x);
    }
    return z;
}
int main(){
    int k;
    scanf("%d%d%d",&n,&m,&k);
    int x,z,w;
    z=k/m; w=k%m; mat c;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            c.e[i][j]=0;
        }
    }
    mat a=c; mat d=c;
    for(int i=1;i<=m;i++){
        mat b=c;
        if(i==1){
            for(int j=1;j<=n;j++){
                scanf("%d",&x);
                a.e[j][x]=1;
            }
            if(w>=1){
                d=a;    
            }
            continue;
        }
        for(int j=1;j<=n;j++){
            scanf("%d",&x);
            b.e[j][x]=1;
        }
        a=pro(b,a);
        if(i<=w){
            d=a;
        }
    }
    if(z==0){
        a=d;
    }else if(w==0){
        c=a;
        a=pow(c,z);
    }else{
        c=a;
        a=pro(d,pow(c,z));
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(a.e[i][j]!=0){
                printf("%d ",j);
                break;
            }
        }
    }
    cout<<endl;
    return 0;
}