#741951 (C++17) No.1857 Gacha Addiction

提出ソース

結果

問題	No.1857 Gacha Addiction
ユーザー	沙耶花
提出日時	2022-02-25 22:18:06
言語	C++17 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	698 ms / 6,000 ms
コード長	1,572 bytes
コンパイル時間	5,211 ms
コンパイル使用メモリ	279,576 KB
実行使用メモリ	28,344 KB
最終ジャッジ日時	2023-09-16 17:20:54
合計ジャッジ時間	28,276 ms
ジャッジサーバーID （参考情報）	judge14 / judge13

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	21 ms 11,520 KB
testcase_01	AC	21 ms 11,756 KB
testcase_02	AC	21 ms 11,460 KB
testcase_03	AC	20 ms 11,432 KB
testcase_04	AC	28 ms 11,780 KB
testcase_05	AC	28 ms 11,848 KB
testcase_06	AC	28 ms 11,772 KB
testcase_07	AC	28 ms 11,856 KB
testcase_08	AC	28 ms 11,792 KB
testcase_09	AC	215 ms 17,080 KB
testcase_10	AC	215 ms 17,056 KB
testcase_11	AC	214 ms 17,112 KB
testcase_12	AC	213 ms 17,128 KB
testcase_13	AC	216 ms 17,032 KB
testcase_14	AC	694 ms 28,328 KB
testcase_15	AC	696 ms 28,224 KB
testcase_16	AC	692 ms 28,160 KB
testcase_17	AC	698 ms 28,156 KB
testcase_18	AC	691 ms 28,248 KB
testcase_19	AC	693 ms 28,344 KB
testcase_20	AC	693 ms 28,152 KB
testcase_21	AC	697 ms 28,268 KB
testcase_22	AC	696 ms 28,152 KB
testcase_23	AC	691 ms 28,152 KB
testcase_24	AC	690 ms 28,212 KB
testcase_25	AC	694 ms 28,244 KB
testcase_26	AC	691 ms 28,224 KB
testcase_27	AC	692 ms 28,328 KB
testcase_28	AC	693 ms 28,252 KB
testcase_29	AC	693 ms 28,328 KB
testcase_30	AC	692 ms 28,252 KB
testcase_31	AC	693 ms 28,260 KB
testcase_32	AC	693 ms 28,268 KB
testcase_33	AC	694 ms 28,324 KB
testcase_34	AC	692 ms 28,176 KB
testcase_35	AC	695 ms 28,312 KB
testcase_36	AC	698 ms 28,200 KB
testcase_37	AC	693 ms 28,228 KB
testcase_38	AC	695 ms 28,340 KB
testcase_39	AC	233 ms 17,900 KB
testcase_40	AC	247 ms 18,196 KB
testcase_41	AC	312 ms 19,632 KB
testcase_42	AC	77 ms 13,684 KB
testcase_43	AC	621 ms 26,448 KB
testcase_44	AC	683 ms 28,204 KB
testcase_45	AC	21 ms 11,548 KB
testcase_46	AC	21 ms 11,440 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for(int i=0;i<(n);i++)
#define Inf 1000000001
using A = array<vector<mint>,2>;
struct combi{
	deque<mint> kaijou;
	deque<mint> kaijou_;
	
	combi(int n){
		kaijou.push_back(1);
		for(int i=1;i<=n;i++){
			kaijou.push_back(kaijou[i-1]*i);
		}
		
		mint b=kaijou[n].inv();
		
		kaijou_.push_front(b);
		for(int i=1;i<=n;i++){
			int k=n+1-i;
			kaijou_.push_front(kaijou_[0]*k);
		}
	}
	
	mint combination(int n,int r){
		if(r>n)return 0;
		mint a = kaijou[n]*kaijou_[r];
		a *= kaijou_[n-r];
		return a;
	}
	
	mint junretsu(int a,int b){
		mint x = kaijou_[a]*kaijou_[b];
		x *= kaijou[a+b];
		return x;
	}
	
	mint catalan(int n){
		return combination(2*n,n)/(n+1);
	}
	
};

int main(){
	
	
	
	combi C(1000000);
	
	int n,s;
	cin>>n>>s;
	queue<A> Q;
	
	rep(i,n){
		int p;
		cin>>p;
		mint t = p;
		t /= s;
		A temp;
		rep(j,2)temp[j] = vector<mint>(2,1);
		temp[0][1] = t;
		temp[1][1] = t*t;
		temp[1][0] = 0;
		Q.push(temp);
	}
	while(Q.size()>1){
		auto x = Q.front();
		Q.pop();
		auto y = Q.front();
		Q.pop();
		A z;
		z[0] = convolution(x[0],y[0]);
		auto z0 = convolution(x[0],y[1]);
		auto z1 = convolution(x[1],y[0]);
		if(z0.size()>z1.size())swap(z0,z1);
		rep(j,z0.size())z1[j] += z0[j];
		z[1] = z1;
		Q.push(z);
	}
	
	mint ans = 0;
	auto z = Q.front();
	rep(i,z[1].size()){
		if(i==0)continue;
		mint t = z[1][i];
		t *= C.kaijou[i];
		t *= i+1;
		ans += t;
	}
	
	cout<<ans.val()<<endl;
	
	return 0;
}

yukicoder

結果

テストケース

ソースコード