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

• 問題: No.1939 Numbered Colorful Balls
• ユーザー: wasa855
• 提出日時: 2023-04-13 13:53:35
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,594 bytes
• コンパイル時間: 1,854 ms
• コンパイル使用メモリ: 197,636 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-04-17 17:44:13
• 合計ジャッジ時間: 2,868 ms
• ジャッジサーバーID （参考情報）: judge4 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	4 ms; 5,376 KB
testcase_02	AC	4 ms; 5,376 KB
testcase_03	AC	6 ms; 5,376 KB
testcase_04	AC	3 ms; 5,376 KB
testcase_05	AC	7 ms; 5,376 KB
testcase_06	AC	5 ms; 5,376 KB
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	AC	4 ms; 5,376 KB
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	AC	3 ms; 5,376 KB
testcase_23	AC	4 ms; 5,376 KB
testcase_24	WA	-
testcase_25	AC	4 ms; 5,376 KB
testcase_26	WA	-
testcase_27	AC	4 ms; 5,376 KB
testcase_28	AC	3 ms; 5,376 KB
testcase_29	AC	5 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();
	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;
}