結果

問題 No.1939 Numbered Colorful Balls
コンテスト
ユーザー leukocyte
提出日時 2022-07-02 10:06:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 69 ms / 2,000 ms
コード長 670 bytes
コンパイル時間 1,771 ms
コンパイル使用メモリ 169,032 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-27 02:47:32
合計ジャッジ時間 3,620 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int N=1e5;
const int mod=998244353;
int add(int x,int y){ return (x+=y)>=mod?x-mod:x; }
int ksm(ll x,int tp,int s=1){
	for(;tp;x=x*x%mod,tp>>=1) if(tp&1) s=x*s%mod;
	return s;
}
int n,m;
int l[N+5];
int f[N+5];
int main(){
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;i++)
		scanf("%d",&l[i]);
	sort(l+1,l+1+m);
	f[0]=1;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m&&l[j]<=i;j++)
			f[i]=(f[i]+1ll*l[j]*f[i-l[j]])%mod;
		f[i]=1ll*f[i]*(n+1)%mod;
		for(int j=1;j<=m&&l[j]<=i;j++)
			f[i]=(f[i]-1ll*(i-l[j])*f[i-l[j]])%mod;
		f[i]=ksm(i,mod-2,f[i]+mod);
	}
	printf("%d\n",ksm(n+1,mod-2,f[n]));
	return 0;
}
0