結果
問題 | No.2530 Yellow Cards |
ユーザー |
|
提出日時 | 2023-11-03 23:12:47 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 444 ms / 2,000 ms |
コード長 | 3,184 bytes |
コンパイル時間 | 1,730 ms |
コンパイル使用メモリ | 172,816 KB |
実行使用メモリ | 198,784 KB |
最終ジャッジ日時 | 2024-09-25 21:26:02 |
合計ジャッジ時間 | 6,384 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
#include<bits/stdc++.h>using namespace std;using ll = long long;using pll = pair<ll, ll>;#define drep(i, cc, n) for (ll i = (cc); i <= (n); ++i)#define rep(i, n) drep(i, 0, n - 1)#define all(a) (a).begin(), (a).end()#define pb push_back#define fi first#define se secondmt19937_64 rng(chrono::system_clock::now().time_since_epoch().count());const ll MOD1000000007 = 1000000007;const ll MOD998244353 = 998244353;const ll MOD[3] = {999727999, 1070777777, 1000000007};const ll LINF = 1LL << 60LL;const int IINF = (1 << 30) - 1;template<long long mod>class modint{long long x;public:modint(long long x=0) : x((x%mod+mod)%mod) {}modint operator-() const {return modint(-x);}bool operator==(const modint& a){if(x == a) return true;else return false;}bool operator==(long long a){if(x == a) return true;else return false;}bool operator!=(const modint& a){if(x != a) return true;else return false;}bool operator!=(long long a){if(x != a) return true;else return false;}modint& operator+=(const modint& a) {if ((x += a.x) >= mod) x -= mod;return *this;}modint& operator-=(const modint& a) {if ((x += mod-a.x) >= mod) x -= mod;return *this;}modint& operator*=(const modint& a) {(x *= a.x) %= mod;return *this;}modint operator+(const modint& a) const {modint res(*this);return res+=a;}modint operator-(const modint& a) const {modint res(*this);return res-=a;}modint operator*(const modint& a) const {modint res(*this);return res*=a;}modint pow(long long t) const {if (!t) return 1;modint a = pow(t>>1);a *= a;if (t&1) a *= *this;return a;}// for prime modmodint inv() const {return pow(mod-2);}modint& operator/=(const modint& a) {return (*this) *= a.inv();}modint operator/(const modint& a) const {modint res(*this);return res/=a;}friend std::istream& operator>>(std::istream& is, modint& m) noexcept {is >> m.x;m.x %= mod;if (m.x < 0) m.x += mod;return is;}friend ostream& operator<<(ostream& os, const modint& m){os << m.x;return os;}};using mint = modint<MOD998244353>;void solve(){ll n, k; cin >> n >> k;vector<vector<mint>> dp(k+1, vector<mint>(n+1, 0));dp[0][0] = 1;mint ninv = mint(1)/mint(n);for(ll i=1; i<=k; i++){for(ll j=0; j<=n; j++){//dp[i-1][j]からの寄与if(j>0) dp[i][j-1] += dp[i-1][j]*mint(j)*ninv;if(j<n) dp[i][j+1] += dp[i-1][j]*mint(n-j)*ninv;}}mint ans = 0;for(ll j=0; j<=n; j++){//現時点でj人のイエローカードans += mint(n + (k-j)/2)*dp[k][j];}cout << ans << endl;}int main(){cin.tie(nullptr);ios::sync_with_stdio(false);int T=1;//cin >> T;while(T--) solve();}