把 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;
}

Logo

开源鸿蒙跨平台开发社区汇聚开发者与厂商,共建“一次开发,多端部署”的开源生态,致力于降低跨端开发门槛,推动万物智联创新。

更多推荐