#858790 (C++17) No.1939 Numbered Colorful Balls

• 問題: No.1939 Numbered Colorful Balls
• ユーザー: wasa855
• 提出日時: 2023-04-13 13:54:44
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 37 ms / 2,000 ms
• コード長: 1,612 bytes
• コンパイル時間: 2,401 ms
• コンパイル使用メモリ: 198,004 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-17 17:45:35
• 合計ジャッジ時間: 3,383 ms
• ジャッジサーバーID （参考情報）: judge1 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	33 ms; 5,248 KB
testcase_01	AC	4 ms; 5,376 KB
testcase_02	AC	5 ms; 5,376 KB
testcase_03	AC	37 ms; 5,376 KB
testcase_04	AC	4 ms; 5,376 KB
testcase_05	AC	36 ms; 5,376 KB
testcase_06	AC	6 ms; 5,376 KB
testcase_07	AC	8 ms; 5,376 KB
testcase_08	AC	5 ms; 5,376 KB
testcase_09	AC	14 ms; 5,376 KB
testcase_10	AC	5 ms; 5,376 KB
testcase_11	AC	7 ms; 5,376 KB
testcase_12	AC	6 ms; 5,376 KB
testcase_13	AC	5 ms; 5,376 KB
testcase_14	AC	10 ms; 5,376 KB
testcase_15	AC	7 ms; 5,376 KB
testcase_16	AC	17 ms; 5,376 KB
testcase_17	AC	12 ms; 5,376 KB
testcase_18	AC	10 ms; 5,376 KB
testcase_19	AC	16 ms; 5,376 KB
testcase_20	AC	7 ms; 5,376 KB
testcase_21	AC	11 ms; 5,376 KB
testcase_22	AC	5 ms; 5,376 KB
testcase_23	AC	5 ms; 5,376 KB
testcase_24	AC	10 ms; 5,376 KB
testcase_25	AC	6 ms; 5,376 KB
testcase_26	AC	31 ms; 5,376 KB
testcase_27	AC	4 ms; 5,376 KB
testcase_28	AC	4 ms; 5,376 KB
testcase_29	AC	6 ms; 5,376 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define eb emplace_back
//mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
#define mod 998244353
#define ll long long
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3f
inline int read()
{
	char ch=getchar(); int nega=1; while(!isdigit(ch)) {if(ch=='-') nega=-1; ch=getchar();}
	int ans=0; while(isdigit(ch)) {ans=ans*10+ch-48;ch=getchar();}
	if(nega==-1) return -ans;
	return ans;
}
void print(vector<int> x){for(int i=0;i<(int)x.size();i++) printf("%d%c",x[i]," \n"[i==(int)x.size()-1]);}
inline int add(int x,int y) {return x+y>=mod?x+y-mod:x+y;}
inline int add(int x,int y,int z) {return add(add(x,y),z);}
inline int sub(int x,int y) {return x-y<0?x-y+mod:x-y;}
inline int mul(int x,int y) {return 1LL*x*y%mod;}
inline int mul(int x,int y,int z) {return mul(mul(x,y),z);}
#define inc(x,y) x=add(x,y)
#define dec(x,y) x=sub(x,y)
#define MN 100005
int f[MN],g[MN],N,M,m=100;
int fac[MN],inv[MN],ifac[MN];
void init()
{
	fac[0]=1; for(int i=1;i<MN;i++) fac[i]=mul(fac[i-1],i);
	inv[0]=inv[1]=1; for(int i=2;i<MN;i++) inv[i]=mul(mod-mod/i,inv[mod%i]);
	ifac[0]=1; for(int i=1;i<MN;i++) ifac[i]=mul(ifac[i-1],inv[i]);
}
signed main()
{
	init();
	cin>>N>>M;
	f[0]=0; for(int i=1;i<=M;i++) f[i]=read();
	sort(f+1,f+M+1);
	g[0]=1;
	for(int n=0;n<=N;n++)
	{
		int res=0;
		for(int i=1;i<=M&&f[i]-1<=n;i++) inc(res,mul(f[i],g[n+1-f[i]]));
		res=mul(res,N+1);
		for(int i=1;i<=M&&f[i]<=n;i++) dec(res,mul(n-f[i]+1,g[n-f[i]+1]));
		g[n+1]=mul(res,inv[n+1]);
	}
	// cout<<g[N]<<endl;
	cout<<mul(inv[N+1],g[N])<<endl;
	return 0;
}