結果
問題 | No.2383 Naphthol |
ユーザー |
|
提出日時 | 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 |
ソースコード
#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";}