結果

問題 No.1939 Numbered Colorful Balls
ユーザー wasa855wasa855
提出日時 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,242 ms
コンパイル使用メモリ 205,344 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-09 11:52:05
合計ジャッジ時間 3,550 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
6,820 KB
testcase_01 AC 4 ms
6,816 KB
testcase_02 AC 6 ms
6,816 KB
testcase_03 AC 37 ms
6,816 KB
testcase_04 AC 5 ms
6,820 KB
testcase_05 AC 37 ms
6,820 KB
testcase_06 AC 7 ms
6,820 KB
testcase_07 AC 7 ms
6,820 KB
testcase_08 AC 5 ms
6,816 KB
testcase_09 AC 15 ms
6,816 KB
testcase_10 AC 5 ms
6,816 KB
testcase_11 AC 8 ms
6,816 KB
testcase_12 AC 5 ms
6,820 KB
testcase_13 AC 5 ms
6,816 KB
testcase_14 AC 10 ms
6,816 KB
testcase_15 AC 7 ms
6,816 KB
testcase_16 AC 17 ms
6,820 KB
testcase_17 AC 12 ms
6,820 KB
testcase_18 AC 11 ms
6,820 KB
testcase_19 AC 16 ms
6,816 KB
testcase_20 AC 7 ms
6,816 KB
testcase_21 AC 10 ms
6,820 KB
testcase_22 AC 4 ms
6,816 KB
testcase_23 AC 5 ms
6,816 KB
testcase_24 AC 11 ms
6,816 KB
testcase_25 AC 6 ms
6,816 KB
testcase_26 AC 31 ms
6,820 KB
testcase_27 AC 4 ms
6,820 KB
testcase_28 AC 4 ms
6,820 KB
testcase_29 AC 6 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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