結果

問題 No.2876 Infection
ユーザー kotatsugamekotatsugame
提出日時 2024-09-06 23:11:31
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 94 ms / 2,000 ms
コード長 1,864 bytes
コンパイル時間 2,382 ms
コンパイル使用メモリ 115,244 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-06 23:11:37
合計ジャッジ時間 4,391 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 94 ms
6,944 KB
testcase_06 AC 93 ms
6,940 KB
testcase_07 AC 93 ms
6,944 KB
testcase_08 AC 93 ms
6,944 KB
testcase_09 AC 94 ms
6,940 KB
testcase_10 AC 18 ms
6,944 KB
testcase_11 AC 16 ms
6,944 KB
testcase_12 AC 14 ms
6,940 KB
testcase_13 AC 23 ms
6,944 KB
testcase_14 AC 85 ms
6,944 KB
testcase_15 AC 65 ms
6,944 KB
testcase_16 AC 5 ms
6,940 KB
testcase_17 AC 2 ms
6,944 KB
testcase_18 AC 20 ms
6,940 KB
testcase_19 AC 3 ms
6,944 KB
testcase_20 AC 54 ms
6,944 KB
testcase_21 AC 3 ms
6,940 KB
testcase_22 AC 59 ms
6,944 KB
testcase_23 AC 23 ms
6,944 KB
testcase_24 AC 19 ms
6,944 KB
testcase_25 AC 58 ms
6,940 KB
testcase_26 AC 3 ms
6,944 KB
testcase_27 AC 21 ms
6,940 KB
testcase_28 AC 3 ms
6,940 KB
testcase_29 AC 78 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<cassert>
#include<atcoder/modint>
#include<atcoder/convolution>
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 N;
mint p,q;
mint pp[2525],qp[2525];
mint dp[2][2525];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N;
	{
		int x;cin>>x;
		p=mint(x)/100;
		q=1-p;
		pp[0]=qp[0]=1;
		for(int i=1;i<2525;i++)
		{
			pp[i]=pp[i-1]*p;
			qp[i]=qp[i-1]*q;
		}
		for(int i=0;i<2525;i++)
		{
			pp[i]*=C.invfac(i);
			qp[i]*=C.invfac(i);
		}
	}
	int now=0;
	dp[now][1]=1;
	mint ans=0;
	for(int i=1;i<=N;i++)
	{
		int nxt=1-now;
		for(int j=i;j<=N;j++)
		{
			dp[nxt][j]=0;
			dp[now][j]*=C.factorial(N-j);
		}
		vector<mint>L(N+1),R(N+1);
		for(int j=i;j<=N;j++)L[j]=dp[now][j];
		for(int k=0;k<=N;k++)R[k]=pp[k];
		L=atcoder::convolution(L,R);
		for(int jk=i;jk<=N;jk++)dp[nxt][jk]=L[jk]*qp[N-jk];
		/*
		for(int j=i;j<=N;j++)
		{
			for(int k=0;k<=N-j;k++)dp[nxt][j+k]+=dp[now][j]*pp[k]*qp[N-j-k];
		}
		*/
		now=nxt;
		ans+=dp[now][i]*i;
	}
	cout<<ans.val()<<endl;
}
0