#775547 (C++14) No.1939 Numbered Colorful Balls

• 問題: No.1939 Numbered Colorful Balls
• ユーザー: leukocyte
• 提出日時: 2022-07-02 10:06:20
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 64 ms / 2,000 ms
• コード長: 670 bytes
• コンパイル時間: 1,585 ms
• コンパイル使用メモリ: 169,160 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-05 07:54:39
• 合計ジャッジ時間: 3,160 ms
• ジャッジサーバーID （参考情報）: judge1 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	56 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	16 ms; 5,376 KB
testcase_03	AC	64 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	63 ms; 5,376 KB
testcase_06	AC	18 ms; 5,376 KB
testcase_07	AC	10 ms; 5,376 KB
testcase_08	AC	5 ms; 5,376 KB
testcase_09	AC	27 ms; 5,376 KB
testcase_10	AC	3 ms; 5,376 KB
testcase_11	AC	13 ms; 5,376 KB
testcase_12	AC	5 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	13 ms; 5,376 KB
testcase_15	AC	7 ms; 5,376 KB
testcase_16	AC	31 ms; 5,376 KB
testcase_17	AC	19 ms; 5,376 KB
testcase_18	AC	15 ms; 5,376 KB
testcase_19	AC	28 ms; 5,376 KB
testcase_20	AC	8 ms; 5,376 KB
testcase_21	AC	20 ms; 5,376 KB
testcase_22	AC	3 ms; 5,376 KB
testcase_23	AC	9 ms; 5,376 KB
testcase_24	AC	19 ms; 5,376 KB
testcase_25	AC	9 ms; 5,376 KB
testcase_26	AC	52 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	2 ms; 5,376 KB
testcase_29	AC	11 ms; 5,376 KB

ソースコード

#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;
}