#981515 (C++17) No.2754 Cumulate and Drop

• 問題: No.2754 Cumulate and Drop
• ユーザー: ttkkggww
• 提出日時: 2024-05-14 15:48:28
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 98 ms / 2,000 ms
• コード長: 870 bytes
• コンパイル時間: 4,271 ms
• コンパイル使用メモリ: 263,716 KB
• 実行使用メモリ: 8,132 KB
• 最終ジャッジ日時: 2024-05-14 15:48:36
• 合計ジャッジ時間: 8,031 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	70 ms; 7,364 KB
testcase_01	AC	71 ms; 7,364 KB
testcase_02	AC	70 ms; 7,360 KB
testcase_03	AC	70 ms; 7,492 KB
testcase_04	AC	70 ms; 7,488 KB
testcase_05	AC	71 ms; 7,488 KB
testcase_06	AC	70 ms; 7,488 KB
testcase_07	AC	70 ms; 7,296 KB
testcase_08	AC	77 ms; 7,620 KB
testcase_09	AC	89 ms; 7,680 KB
testcase_10	AC	94 ms; 7,872 KB
testcase_11	AC	87 ms; 7,624 KB
testcase_12	AC	97 ms; 8,008 KB
testcase_13	AC	94 ms; 7,876 KB
testcase_14	AC	97 ms; 7,880 KB
testcase_15	AC	85 ms; 7,620 KB
testcase_16	AC	75 ms; 7,496 KB
testcase_17	AC	97 ms; 8,008 KB
testcase_18	AC	98 ms; 8,004 KB
testcase_19	AC	98 ms; 8,132 KB

ソースコード

#include<bits/stdc++.h>
#include <iostream>
using namespace std;
#include<atcoder/all>
using namespace atcoder;
using ll = long long;
using mint = modint998244353;
const int MAX = 505050;
mint fac[MAX],facinv[MAX];
void init(){
	fac[0] = fac[1] = 1;
	facinv[0] = facinv[1] = 1;
	for(int i = 2;i<MAX;i++){
		fac[i] = fac[i-1] * i;
		facinv[i] = fac[i].inv();
	}
}
mint biom(int n,int k){
	if(n<k)return 0;
	if(n<0||k<0)return 0;
	return fac[n]*facinv[k]*facinv[n-k];
}

mint routesum(int x,int y){
	return biom(x+y,x);
}

int N;
vector<int> a;

void solve(){
	mint ans = 0;
	for(int i = 0;i<N;i++){
		mint coef = biom(N+i-1,i) - biom(N+i-1,i-1);
		ans += mint(a[N-i-1])*(coef);
	}
	cout<< ans.val() << endl;
}

signed main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	init();
	cin >> N;
	a = vector<int>(N);
	for(int i = 0;i<N;i++){
		cin >> a[i];
	}
	solve();
}