結果
問題 | No.2529 Treasure Hunter |
ユーザー |
![]() |
提出日時 | 2023-11-03 22:04:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 29 ms / 2,000 ms |
コード長 | 2,482 bytes |
コンパイル時間 | 1,759 ms |
コンパイル使用メモリ | 120,004 KB |
最終ジャッジ日時 | 2025-02-17 18:19:39 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
#include <iostream>#include <algorithm>#include <iomanip>#include <vector>#include <queue>#include <deque>#include <set>#include <map>#include <tuple>#include <cmath>#include <numeric>#include <functional>#include <cassert>#include <atcoder/modint>#define debug_value(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << #x << "=" << x << endl;#define debug(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << x << endl;template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }using namespace std;typedef long long ll;template<typename T>vector<vector<T>> vec2d(int n, int m, T v){return vector<vector<T>>(n, vector<T>(m, v));}template<typename T>vector<vector<vector<T>>> vec3d(int n, int m, int k, T v){return vector<vector<vector<T>>>(n, vector<vector<T>>(m, vector<T>(k, v)));}template<typename T>void print_vector(vector<T> v, char delimiter=' '){if(v.empty()) {cout << endl;return;}for(int i = 0; i+1 < v.size(); i++) cout << v[i] << delimiter;cout << v.back() << endl;}using mint = atcoder::modint998244353;ostream& operator<<(ostream& os, const mint& m){os << m.val();return os;}#define N_MAX 200002mint inv[N_MAX],fac[N_MAX],finv[N_MAX];void init(){const ll MOD = mint::mod();fac[0]=1;fac[1]=1;finv[0]=1;finv[1]=1;inv[1]=1;for(int i=2;i<N_MAX;i++){inv[i]=mint(MOD)-inv[MOD%i]*(MOD/i);fac[i]=fac[i-1]*(ll) i;finv[i]=finv[i-1]*inv[i];}}mint comb(ll n, ll r){if(n < r){return mint(0);}else{return fac[n]*finv[r]*finv[n-r];}}void solve(){int n, m; cin >> n >> m;auto dp = vec2d(m+1, 3, mint(0));dp[0][0] = 1;mint not_adj = comb(n, 2) - n;for(int i = 0; i < m; i++){for(int j = 0; j < 3; j++){dp[i+1][0] += dp[i][j];dp[i+1][1] += dp[i][j]*(n-j);}if(n > 2){dp[i+1][2] += dp[i][0]*not_adj;dp[i+1][2] += dp[i][1]*(comb(n, 2) - n - (n-1) + 2);dp[i+1][2] += dp[i][2]*(comb(n, 2) - n - (2*n-3) + 4);}}cout << dp[m][0]+dp[m][1]+dp[m][2] << endl;}int main(){ios::sync_with_stdio(false);cin.tie(0);cout << setprecision(10) << fixed;init();int t; cin >> t;while(t--) solve();}