結果
問題 | No.2876 Infection |
ユーザー | GOTKAKO |
提出日時 | 2024-09-06 22:53:32 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,330 bytes |
コンパイル時間 | 2,323 ms |
コンパイル使用メモリ | 214,000 KB |
実行使用メモリ | 18,944 KB |
最終ジャッジ日時 | 2024-09-06 22:54:11 |
合計ジャッジ時間 | 15,705 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | AC | 39 ms
18,816 KB |
testcase_06 | AC | 40 ms
18,944 KB |
testcase_07 | TLE | - |
testcase_08 | TLE | - |
testcase_09 | TLE | - |
testcase_10 | AC | 110 ms
6,944 KB |
testcase_11 | AC | 92 ms
6,944 KB |
testcase_12 | AC | 54 ms
6,944 KB |
testcase_13 | AC | 245 ms
7,040 KB |
testcase_14 | AC | 1,652 ms
16,128 KB |
testcase_15 | AC | 21 ms
11,264 KB |
testcase_16 | AC | 9 ms
6,940 KB |
testcase_17 | AC | 2 ms
6,940 KB |
testcase_18 | AC | 9 ms
6,940 KB |
testcase_19 | AC | 2 ms
6,940 KB |
testcase_20 | AC | 413 ms
8,704 KB |
testcase_21 | AC | 4 ms
6,944 KB |
testcase_22 | AC | 560 ms
9,728 KB |
testcase_23 | AC | 257 ms
7,168 KB |
testcase_24 | AC | 126 ms
6,944 KB |
testcase_25 | AC | 505 ms
9,344 KB |
testcase_26 | AC | 3 ms
6,940 KB |
testcase_27 | AC | 163 ms
6,940 KB |
testcase_28 | AC | 6 ms
6,940 KB |
testcase_29 | AC | 1,337 ms
14,464 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=0; i<=N; i++) for(int k=N-1; k>0; k--) if(dp.at(i).at(k) != 0){ int left = N-i; dp.at(i).at(k-1) += dp.at(i).at(k)*infe.at(left).at(0); for(int l=1; l<=left; l++) dp.at(i+l).at(k-1+l) += dp.at(i).at(k)*infe.at(left).at(l); } mint answer = 0; for(int i=1; i<=N; i++) answer += dp.at(i).at(0)*i; cout << answer.v << endl; }