P3390 【模板】矩阵快速幂

把 f（ll k） 写成 f（int k）

结果又是tle又是wa的。找了半天错误。

（为了节省一点时间，因为函数不能return数组，还把数组改写到结构体里）

我在想考试怎么办，考试不能看过没过，

过了样例，实则零蛋，

粗心那么一点点，就可能完全没分，实在可惜。

ac代码1：

#include<iostream>
#define ll long long
using namespace std;
const ll M=1e9+7;
const int N=110;
ll n,k;
ll a[N][N],ans[N][N];
void mul(ll x[N][N],ll y[N][N],int flag){
    ll tmp[N][N];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++){
            tmp[i][j]=0;
            for(int k=1;k<=n;k++)
                tmp[i][j]=(tmp[i][j]+(x[i][k]*y[k][j])%M)%M;
        }
            
    if(flag){
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                ans[i][j]=tmp[i][j];
    }
    else{
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                a[i][j]=tmp[i][j];       
    }
}
void f(ll k){
    while(k){
        if(k&1) mul(ans,a,1);
        mul(a,a,0);
        k>>=1;
    }
}
int main(){
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++){
        ans[i][i]=1;
        for(int j=1;j<=n;j++)
            scanf("%d",&a[i][j]);
    }
    f(k);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            printf("%d ",ans[i][j]);
        printf("\n");
    }
    return 0;
}

ac代码2（写在结构体里）

#include<iostream>
#define ll long long
using namespace std;
const ll M=1e9+7;
const int N=110;
ll n,k;
struct matrix{
    ll a[N][N];
}s,ans;
matrix mul(matrix x,matrix y){
    matrix tmp;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++){
            tmp.a[i][j]=0;
            for(int k=1;k<=n;k++)
                tmp.a[i][j]=(tmp.a[i][j]+(x.a[i][k]*y.a[k][j])%M)%M;
        }
    return tmp;
}
void f(ll k){
    while(k){
        if(k&1) ans=mul(ans,s);
        s=mul(s,s);
        k>>=1;
    }
}
int main(){
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++){
        ans.a[i][i]=1;
        for(int j=1;j<=n;j++)
            scanf("%d",&s.a[i][j]);
    }
    f(k);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            printf("%d ",ans.a[i][j]);
        printf("\n");
    }
    return 0;
}

：