結果
問題 | No.2530 Yellow Cards |
ユーザー | okkuukenken |
提出日時 | 2023-11-03 23:04:37 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 387 ms / 2,000 ms |
コード長 | 1,463 bytes |
コンパイル時間 | 8,573 ms |
コンパイル使用メモリ | 350,624 KB |
実行使用メモリ | 238,604 KB |
最終ジャッジ日時 | 2023-11-03 23:04:51 |
合計ジャッジ時間 | 14,048 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,700 KB |
testcase_01 | AC | 2 ms
5,704 KB |
testcase_02 | AC | 41 ms
45,564 KB |
testcase_03 | AC | 18 ms
22,032 KB |
testcase_04 | AC | 2 ms
5,696 KB |
testcase_05 | AC | 40 ms
45,564 KB |
testcase_06 | AC | 2 ms
5,696 KB |
testcase_07 | AC | 346 ms
238,604 KB |
testcase_08 | AC | 373 ms
238,604 KB |
testcase_09 | AC | 349 ms
238,564 KB |
testcase_10 | AC | 346 ms
238,488 KB |
testcase_11 | AC | 345 ms
238,448 KB |
testcase_12 | AC | 351 ms
238,604 KB |
testcase_13 | AC | 387 ms
238,300 KB |
testcase_14 | AC | 310 ms
216,364 KB |
testcase_15 | AC | 229 ms
163,376 KB |
testcase_16 | AC | 184 ms
130,164 KB |
testcase_17 | AC | 308 ms
217,608 KB |
testcase_18 | AC | 77 ms
59,092 KB |
testcase_19 | AC | 18 ms
19,080 KB |
testcase_20 | AC | 86 ms
69,900 KB |
ソースコード
#pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include<bits/stdc++.h> using namespace std; typedef long long ll; typedef unsigned long long ull; constexpr int dx[]={1,0,0,-1},dy[]={0,1,-1,0}; constexpr int Dx[]={1,1,0,-1,-1,-1,0,1},Dy[]={0,1,1,1,0,-1,-1,-1}; constexpr int mod=998244353,Mod=1e9+7,inf=Mod; constexpr ll linf=(ll)inf*inf; typedef pair<ll,int>P; #define m_p make_pair struct fastio{ fastio(){ cin.tie(0); ios::sync_with_stdio(0); cout<<fixed<<setprecision(20); }; }fio; template<class T,class U> bool chmax(T&a,const U&b){ if(a<b){ a=b; return 1; } return 0; } template<class T,class U> bool chmin(T&a,const U&b){ if(a>b){ a=b; return 1; } return 0; } //* #include<atcoder/all> using namespace atcoder; //*/ ll mypow(int x,ll n){ if(n==0) return 1; if(n%2==1) return mypow(x,n-1)*x%mod; ll res=mypow(x,n/2); return res*res%mod; } ll inv(int x){ return mypow(x,mod-2); } ll dp1[5001][5001],dp2[5001][5001]; int main(){ int n,k; cin>>n>>k; int in=inv(n); dp1[0][0]=1; dp2[0][0]=n; for(int i=0;i<k;i++){ for(int j=0;j<=min(n,i);j++){ if(j>0) (dp1[i+1][j-1]+=dp1[i][j]*in%mod*j)%=mod; if(j<n) (dp1[i+1][j+1]+=dp1[i][j]*in%mod*(n-j))%=mod; if(j>0) (dp2[i+1][j-1]+=(dp2[i][j]+dp1[i][j])*in%mod*j)%=mod; if(j<n) (dp2[i+1][j+1]+=dp2[i][j]*in%mod*(n-j))%=mod; } } int ans=0; for(int i=0;i<=n;i++) (ans+=dp2[k][i])%=mod; cout<<ans<<endl; }