結果

問題 No.3515 Anti EIKO
コンテスト
ユーザー ZeriToki
提出日時 2026-04-24 22:39:39
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 342 ms / 2,000 ms
コード長 4,767 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,642 ms
コンパイル使用メモリ 349,792 KB
実行使用メモリ 7,976 KB
最終ジャッジ日時 2026-04-24 22:40:00
合計ジャッジ時間 8,619 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge4_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rrep(i, s, n) for (int i = (s); i < (int)(n); i++)
template<typename T> bool chmin(T& a, T b){if(a > b){a = b; return true;} return false;}
template<typename T> bool chmax(T& a, T b){if(a < b){a = b; return true;} return false;}
const long long mod=998244353;
const long long mod2=469762049;
const long long mod100=1000000007;

class MATH{
    public:
    vector<ll>factorial;//階乗データ
    vector<ll>factorial_inv;
    void factorial_init(ll N,ll m){//階乗を入れる
        factorial.assign(N+1,0);
        factorial_inv.assign(N+1,0);
        ll p=1;
        factorial[0]=1;
        for(int i=1;i<=N;i++){
            p=(p*i)%m;
            factorial[i]=p;
        }
        factorial_inv[N]=modinv(factorial[N],m);
        for(int i=N-1;i>=0;i--){
            factorial_inv[i]=(factorial_inv[i+1]*(i+1))%m;
        }
        return;
    }
    ll power(ll a,ll b,ll m){//a^b
        a%=m;
        ll p=a,Answer=1;
        for(ll i=0;i<60;i++){
            ll wari=(1LL<<i);
            if((b/wari)%2==1){
                Answer=(Answer*p)%m;
            }
            p=(p*p)%m;
        }
        return Answer;
    }
    ll Division(ll a,ll b,ll m){//a/b
        return (a*power(b,m-2,m))%m;
    }
    ll modinv(ll x,ll m){//逆元
        return power(x,m-2,m);
    }
    ll comb(ll n,ll r,ll m){//nCrを求める(factorial必要)
        if(n<0 || r<0 || n-r<0) return 0;
        ll res=factorial[n];
        res=(res*factorial_inv[r])%m;
        res=(res*factorial_inv[n-r])%m;
        return res;
    }
    ll GCD(ll A,ll B){//GCD(A,B)
        while(A>=1 && B>=1){
            if(A>=B) A=(A%B);
            else B=(B%A);
        }
        if(A!=0) return A;
        return B;
    }
    ll modlog(ll a,ll b,ll m){//a^x=b(mod m)を満たす最小のx(ないなら-1)
        a%=m;b%=m;
        ll lo=-1,hi=m;
        while(hi-lo>1){
            ll mid=(lo+hi)/2;
            if(mid*mid>=m) hi=mid;
            else lo=mid;
        }
        ll sqrtM=hi;
        map<ll,ll>baby;
        ll amari=a;
        for(ll r=1;r<sqrtM;r++){
            if(!baby.count(amari)) baby[amari]=r;
            amari=(amari*a)%m;
        }
        ll A=power(modinv(a,m),sqrtM,m);
        amari=b;
        for(ll q=0;q<sqrtM;q++){
            if(amari==1 && q>=1) return q*sqrtM;
            else if(baby.count(amari)) return q*sqrtM+baby[amari];
            amari=(amari*A)%m;
        }
        return -1;
    }
};
MATH math;

template<typename T>struct Matrix{
	int n,m;
	vector<vector<T>>table;
	Matrix(int N=0,int M=0):n(N),m(M){
		table.assign(n+1,{});
		for(int i=1;i<=n;i++) table[i].assign(m+1,0);
	}
	Matrix operator+(const Matrix<T>&a){
		assert(a.n==n);
		assert(a.m==m);
		Matrix<T>res(a.n,a.m);
		for(int i=1;i<=a.n;i++){
			for(int j=1;j<=a.m;j++){
				res.table[i][j]=(a.table[i][j]+table[i][j]);
			}
		}
		return res;
	}
	Matrix operator-(const Matrix<T>&a){
		assert(a.n==n);
		assert(a.m==m);
		Matrix<T>res(a.n,a.m);
		for(int i=1;i<=a.n;i++){
			for(int j=1;j<=a.m;j++){
				res.table[i][j]=(a.table[i][j]-table[i][j]);
			}
		}
		return res;
	}
	Matrix operator*(const Matrix<T>&a){
		assert(m==a.n);
    	Matrix<T>res(n,a.m);
    	for(int i=1;i<=n;i++){
    	    for(int j=1;j<=a.m;j++){
    	        for(int k=1;k<=m;k++){
					res.table[i][j]+=table[i][k]*a.table[k][j]%mod;
                    res.table[i][j]%=mod;
				}
   		    }
    	}
    	return res;
	}
};
template<typename T>Matrix<T>power(Matrix<T>a,long long k){
	assert(a.n==a.m);
    Matrix<T>res(a.n,a.m);
    Matrix<T>now=a;
    for(int i=1;i<=a.n;i++) res.table[i][i]=1;
    for(int i=0;i<=60;i++){
        if((k>>i)&1LL) res=res*now;
        now=now*now;
    }
    return res;
}
int main(){
    ios::sync_with_stdio(false);
	cin.tie(nullptr);
    int N;
    ll K;
    cin>>N>>K;
    math.factorial_init(100,mod);
    Matrix<ll>dat(N*4+1,N*4+1);
    for(int i=1;i<=N*4+1;i++){
        if(i==N*4+1) continue;
        if(i%4==1){
            dat.table[i][1]=math.modinv(5,mod)*4%mod;
            dat.table[i][i+1]=math.modinv(5,mod);
            continue;
        }
        dat.table[i][i+1]+=math.modinv(5,mod);
        dat.table[i][1]+=math.modinv(5,mod)*3%mod;
        dat.table[i][2]+=math.modinv(5,mod);
    }
    Matrix<ll>ans(1,N*4+1);
    ans.table[1][1]=1;
    /*for(int i=1;i<=N*4+1;i++){
        for(int j=1;j<=N*4+1;j++){
            cout<<dat.table[i][j]<<" ";
        }
        cout<<endl;
    }
    cout<<endl;
    for(int i=1;i<=K;i++){
        ans=ans*dat;
        for(int j=1;j<=N*4+1;j++){
            cout<<ans.table[1][j]<<" ";
        }
        cout<<endl;
    }*/
    ans=ans*power(dat,K);
    cout<<ans.table[1][4*N+1]<<endl;
}
0