#489112 (C++14) No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)

• 問題: No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
• ユーザー: ttkkggww
• 提出日時: 2020-05-29 23:09:12
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 138 ms / 2,000 ms
• コード長: 500 bytes
• コンパイル時間: 1,752 ms
• コンパイル使用メモリ: 166,216 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-24 00:59:56
• 合計ジャッジ時間: 3,381 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	82 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	135 ms; 5,376 KB
testcase_11	AC	63 ms; 5,376 KB
testcase_12	AC	54 ms; 5,376 KB
testcase_13	AC	62 ms; 5,376 KB
testcase_14	AC	65 ms; 5,376 KB
testcase_15	AC	109 ms; 5,376 KB
testcase_16	AC	42 ms; 5,376 KB
testcase_17	AC	45 ms; 5,376 KB
testcase_18	AC	106 ms; 5,376 KB
testcase_19	AC	61 ms; 5,376 KB
testcase_20	AC	3 ms; 5,376 KB
testcase_21	AC	94 ms; 5,376 KB
testcase_22	AC	3 ms; 5,376 KB
testcase_23	AC	2 ms; 5,376 KB
testcase_24	AC	2 ms; 5,376 KB
testcase_25	AC	138 ms; 5,376 KB
testcase_26	AC	135 ms; 5,376 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const ll MOD = 998244353;

ll dp[2][6001];

int main()
{
	int n,q;
	cin >> n >> q;
	vector<ll> A(n);
	for(int i =0;i<n;i++)cin >> A[i];
	dp[0][0] = 1;
	for(int i =0;i<n;i++)
	{
		for(int j = 0;j<6000;j++)
		{
			(dp[1][j] += dp[0][j]*(A[i]-1))%=MOD;
			(dp[1][j+1] += dp[0][j])%=MOD;
		}
		for(int j = 0;j<=6000;j++)
		{
			dp[0][j] = dp[1][j];
			dp[1][j] = 0;
		}
	}
	while(q--)
	{
		int b;
		cin >> b;
		cout<<dp[0][b]<<endl;
	}
}