結果
問題 | No.2529 Treasure Hunter |
ユーザー |
|
提出日時 | 2023-11-03 21:48:10 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 12 ms / 2,000 ms |
コード長 | 1,624 bytes |
コンパイル時間 | 793 ms |
コンパイル使用メモリ | 75,208 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-09-25 19:57:09 |
合計ジャッジ時間 | 1,531 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
#include<iostream>#include<cassert>#include<atcoder/modint>using namespace std;#include<vector>template<typename T>struct combination{vector<T>fac,ifac;combination(size_t N=0):fac(1,1),ifac(1,1){make_table(N);}void make_table(size_t N){if(fac.size()>N)return;size_t now=fac.size();N=max(N,now*2);fac.resize(N+1);ifac.resize(N+1);for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i;ifac[N]=1/fac[N];for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1);}T factorial(size_t n){make_table(n);return fac[n];}T invfac(size_t n){make_table(n);return ifac[n];}T P(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k];}T C(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k]*ifac[k];}T H(size_t n,size_t k){if(n==0)return k==0?1:0;return C(n-1+k,k);}};using mint=atcoder::modint998244353;combination<mint>C;int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int T;cin>>T;for(;T--;){int N,M;cin>>M>>N;mint dp[2][3];int now=0;dp[now][0]=1;dp[now][1]=0;dp[now][2]=0;for(int i=0;i<N;i++){int nxt=1-now;for(int j=0;j<3;j++)dp[nxt][j]=0;dp[nxt][0]+=dp[now][0];dp[nxt][1]+=dp[now][0]*M;if(M>=3)dp[nxt][2]+=dp[now][0]*(C.C(M,2)-M);dp[nxt][0]+=dp[now][1];dp[nxt][1]+=dp[now][1]*(M-1);if(M>=3)dp[nxt][2]+=dp[now][1]*(C.C(M-1,2)-M+2);dp[nxt][0]+=dp[now][2];dp[nxt][1]+=dp[now][2]*(M-2);if(M>=4)dp[nxt][2]+=dp[now][2]*(C.C(M,2)-3*M+7);now=nxt;}cout<<(dp[now][0]+dp[now][1]+dp[now][2]).val()<<"\n";}}