結果

問題 No.1815 K色問題
ユーザー okkuukenken
提出日時 2020-01-09 19:16:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,631 bytes
コンパイル時間 1,005 ms
コンパイル使用メモリ 80,732 KB
最終ジャッジ日時 2025-01-08 17:00:29
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 3 WA * 5 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#define int long long
using namespace std;
constexpr int mod=1e9+7;
int pow(int x,int n){
    int res=1;
    while(n){
        if(n&1)
            (res*=x)%=mod;
        (x*=x)%=mod;
        n>>=1;
    }
    return res;
}
int inv(int n){
    return pow(n,mod-2);
}
int fact[100001];
int mod_fact(int n){
    if(fact[n]!=0)
        return fact[n];
    if(n==0)
        return fact[n]=1;
    return fact[n]=n*mod_fact(n-1)%mod;
}
int comb(int n,int k){
    return mod_fact(n)*inv(mod_fact(k)*mod_fact(n-k)%mod)%mod;
}
typedef vector<int>vec;
typedef vector<vec>mat;
mat mul(mat a,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])%=mod;
    return c;
}
mat pow_mat(mat a,int 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;
}
signed main(){
    int n,m;
    cin>>n>>m;
    int ans=m*(m-1)%mod;
    if(m&1)
        ans=mod-ans;
    for(int i=3;i<=m;i+=1){
        mat a(1,vec(2)),b(2,vec(2)),c(2,vec(1));
        a[0][0]=1,a[0][1]=1;
        b[0][0]=(i*i*i-i*i*6+i*14-13)%mod,b[0][1]=(i*i*i-i*i*6+i*13-10)%mod,b[1][0]=(i*i-i*4+5)%mod,b[1][1]=(i*i-i*3+3)%mod;
        c[0][0]=(i*i*i-i*i*3+i*2)%mod,c[1][0]=(i*i-i)%mod;
        int tmp=mul(mul(a,pow_mat(b,n-1)),c)[0][0];
        (tmp*=comb(m,i))%=mod;
        if((m-i)&1)
            tmp=mod-tmp;
        (ans+=tmp)%=mod;
    }
    cout<<ans<<endl;
}
0