結果
問題 | No.2876 Infection |
ユーザー | GOTKAKO |
提出日時 | 2024-09-06 23:03:32 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 3,203 bytes |
コンパイル時間 | 2,703 ms |
コンパイル使用メモリ | 212,420 KB |
実行使用メモリ | 18,944 KB |
最終ジャッジ日時 | 2024-09-06 23:04:47 |
合計ジャッジ時間 | 23,933 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,816 KB |
testcase_02 | AC | 2 ms
6,940 KB |
testcase_03 | AC | 2 ms
6,944 KB |
testcase_04 | AC | 2 ms
6,944 KB |
testcase_05 | TLE | - |
testcase_06 | TLE | - |
testcase_07 | TLE | - |
testcase_08 | TLE | - |
testcase_09 | TLE | - |
testcase_10 | AC | 128 ms
6,944 KB |
testcase_11 | AC | 105 ms
6,944 KB |
testcase_12 | AC | 63 ms
6,944 KB |
testcase_13 | AC | 282 ms
7,040 KB |
testcase_14 | AC | 1,901 ms
16,128 KB |
testcase_15 | AC | 861 ms
11,136 KB |
testcase_16 | AC | 11 ms
6,944 KB |
testcase_17 | AC | 2 ms
6,940 KB |
testcase_18 | AC | 197 ms
6,940 KB |
testcase_19 | AC | 2 ms
6,940 KB |
testcase_20 | AC | 483 ms
8,704 KB |
testcase_21 | AC | 4 ms
6,940 KB |
testcase_22 | AC | 646 ms
9,728 KB |
testcase_23 | AC | 297 ms
7,168 KB |
testcase_24 | AC | 147 ms
6,944 KB |
testcase_25 | AC | 580 ms
9,344 KB |
testcase_26 | AC | 3 ms
6,944 KB |
testcase_27 | AC | 185 ms
6,944 KB |
testcase_28 | AC | 7 ms
6,940 KB |
testcase_29 | AC | 1,521 ms
14,336 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; //悪あがき. #pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") long long mod = 998244353; //入力が必ず-mod<a<modの時. struct mint{ long long v = 0; mint(){} mint(int a){v = a<0?a+mod:a;} mint(long long a){v = a<0?a+mod:a;} mint(unsigned long long a){v = a;} long long val(){return v;} void modu(){v %= mod;} mint repeat2mint(const long long a,long long b){ mint ret = 1,p = a; while(b){ if(b&1) ret *= p; p *= p; b >>= 1; } return ret; }; mint &operator=(const mint &b) = default; mint operator-() const {return mint(0)-(*this);} mint operator+(const mint b){return mint(v)+=b;} mint operator-(const mint b){return mint(v)-=b;} mint operator*(const mint b){return mint(v)*=b;} mint operator/(const mint b){return mint(v)/=b;} mint operator+=(const mint b){ v += b.v; if(v >= mod) v -= mod; return *this; } mint operator-=(const mint b){ v -= b.v; if(v < 0) v += mod; return *this; } mint operator*=(const mint b){v = v*b.v%mod; return *this;} mint operator/=(mint b){ if(b == 0) assert(false); int left = mod-2; while(left){if(left&1) *this *= b; b *= b; left >>= 1;} return *this; } mint operator++(int){*this += 1; return *this;} mint operator--(int){*this -= 1; return *this;} bool operator==(const mint b){return v == b.v;} bool operator!=(const mint b){return v != b.v;} bool operator>(const mint b){return v > b.v;} bool operator>=(const mint b){return v >= b.v;} bool operator<(const mint b){return v < b.v;} bool operator<=(const mint b){return v <= b.v;} mint pow(const long long x){return repeat2mint(v,x);} mint inv(){return mint(1)/v;} }; int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int N; mint x; cin >> N >> x.v; x /= 100; mint y = mint(1)-x; vector<mint> px(N+1,1),py(N+1,1); for(int i=1; i<=N; i++) px.at(i) = px.at(i-1)*x; for(int i=1; i<=N; i++) py.at(i) = py.at(i-1)*y; int Limit = N; vector<mint> fac(Limit+1,1),inv(Limit+1,1),facinv(Limit+1,1); for(int i=1; i<=Limit; i++) fac.at(i).v = fac.at(i-1).v*i %mod; for(int i=2; i<=Limit; i++) inv.at(i).v = mod-(mod/i*inv.at(mod%i).v %mod); for(int i=2; i<=Limit; i++) facinv.at(i).v = facinv.at(i-1).v*inv.at(i).v %mod; auto nCr = [&](int n, int r) -> mint { if(n < r || r < 0 || n < 0) return mint(0); return fac.at(n)*facinv.at(r)*facinv.at(n-r); }; vector<vector<mint>> infe(N+1,vector<mint>(N+1)); //i人残っている状態でk人感染する確率. for(int i=0; i<=N; i++) for(int k=0; k<=i; k++) infe.at(i).at(k) = px.at(k)*py.at(i-k)*nCr(i,k); vector<vector<mint>> dp(N+1,vector<mint>(N+1)); dp.at(1).at(1) = 1; for(int i=1; i<=N; i++) for(int k=i; k>0; k--) for(int l=0; l<=N-i; l++) dp.at(i+l).at(k-1+l) += dp.at(i).at(k)*infe.at(N-i).at(l); mint answer = 0; for(int i=1; i<=N; i++) answer += dp.at(i).at(0)*i; cout << answer.v << endl; }