結果

問題 No.2383 Naphthol
ユーザー i_am_noob
提出日時 2023-07-14 22:00:09
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 8 ms / 2,000 ms
コード長 1,892 bytes
コンパイル時間 1,763 ms
コンパイル使用メモリ 193,256 KB
最終ジャッジ日時 2025-02-15 14:02:46
ジャッジサーバーID
（参考情報） 		judge3 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
using pii=pair<int,int>;

#define all(a) a.begin(),a.end()
#define pb push_back
#define sz(a) ((int)a.size())

const int N=210005,mod=998244353;
int add(int x, int y){x+=y; if(x>=mod) x-=mod; return x;}
int sub(int x, int y){x-=y; if(x<0) x+=mod; return x;}
int mul(int x, int y){return ((ll)x)*y%mod;}
int Pow(int x, ll y=mod-2){int res=1; for(; y; x=mul(x,x),y>>=1) if(y&1) res=mul(res,x); return res;}
int fac[N],inv[N],ifac[N];
inline int C(int n, int m){if(m<0||m>n) return 0; return mul(fac[n],mul(ifac[m],ifac[n-m]));}
void init_comb(){
    fac[0]=inv[1]=ifac[0]=1;
    for(int i=1; i<N; ++i) fac[i]=mul(fac[i-1],i);
    for(int i=2; i<N; ++i) inv[i]=mul(inv[mod%i],mod-mod/i);
    for(int i=1; i<N; ++i) ifac[i]=mul(ifac[i-1],inv[i]);
}

int n,k,cnt[5];

signed main(){
    ios_base::sync_with_stdio(0),cin.tie(0);
    init_comb();
    cin >> n >> k;
    if(2*n+4-k<k) k=2*n+4-k;
    if(n==1){
        if(k<=1) cout << "1\n";
        else if(k==2) cout << "3\n";
        else cout << "3\n";
        return 0;
    }
    cnt[0]=C(2*n+4,k);
    cnt[1]=k&1?0:C(n+2,k/2);
    if(n&1){
        for(int i=0; i<=2; ++i) if((k+i)%2==0) cnt[2]=add(cnt[2],mul(C(2,i),C(n+1,(k-i)/2)));
    }
    else{
        if(k%2==0) cnt[2]=C(n+2,k/2);
    }
    cnt[3]=cnt[1];
    if(n&1){
        if(k%4==0) cnt[4]=C((n+1)/2,k/4);
        if(k%4==2) cnt[4]=C((n+1)/2,k/4);
    }
    else{
        if(k%4==0) cnt[4]=C((n+2)/2,k/4);
    }
    for(int i=1; i<=3; ++i) cnt[i]=sub(cnt[i],cnt[4]);
    for(int i=1; i<=4; ++i) cnt[0]=sub(cnt[0],cnt[i]);
    //cout << cnt[0] << ' ' << cnt[1] << ' ' << cnt[2] << ' ' << cnt[3] << "\n";
    int res=0;
    for(int i=0; i<5; ++i){
        if(i==0) res=add(res,mul(cnt[0],Pow(4)));
        else if(i==4) res=add(res,cnt[i]);
        else res=add(res,mul(cnt[i],Pow(2)));
    }
    cout << res << "\n";
}
0