結果

問題 No.2144 MM
ユーザー tree_splaytree_splay
提出日時 2022-12-02 22:41:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 1,496 bytes
コンパイル時間 2,208 ms
コンパイル使用メモリ 202,136 KB
実行使用メモリ 6,016 KB
最終ジャッジ日時 2024-10-10 00:17:00
合計ジャッジ時間 3,965 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 22 ms
5,248 KB
testcase_08 AC 34 ms
5,888 KB
testcase_09 AC 22 ms
5,248 KB
testcase_10 AC 34 ms
5,888 KB
testcase_11 AC 34 ms
5,888 KB
testcase_12 AC 34 ms
5,760 KB
testcase_13 AC 33 ms
5,760 KB
testcase_14 AC 33 ms
6,016 KB
testcase_15 AC 28 ms
5,632 KB
testcase_16 AC 9 ms
5,248 KB
testcase_17 AC 26 ms
5,376 KB
testcase_18 AC 7 ms
5,248 KB
testcase_19 AC 27 ms
5,248 KB
testcase_20 AC 24 ms
5,248 KB
testcase_21 AC 17 ms
5,248 KB
testcase_22 AC 5 ms
5,248 KB
testcase_23 AC 24 ms
5,248 KB
testcase_24 AC 9 ms
5,248 KB
testcase_25 AC 2 ms
5,248 KB
testcase_26 AC 2 ms
5,248 KB
testcase_27 AC 2 ms
5,248 KB
testcase_28 AC 2 ms
5,248 KB
testcase_29 AC 2 ms
5,248 KB
testcase_30 AC 2 ms
5,248 KB
testcase_31 AC 2 ms
5,248 KB
testcase_32 AC 2 ms
5,248 KB
testcase_33 AC 2 ms
5,248 KB
testcase_34 AC 2 ms
5,248 KB
testcase_35 AC 2 ms
5,248 KB
testcase_36 AC 2 ms
5,248 KB
testcase_37 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

#define INF 1234567890
#define ll long long
#define MOD 998244353

int N, M;
int A[200201];
ll dp[200201];

int main()
{
	ios::sync_with_stdio(0); cin.tie(0);
	cin.exceptions(ios::badbit | ios::failbit);
	cin >> N >> M;
	// dp[0][0] = 1;
	// for(int i=1; i<=N; i++)
	// {
	// 	for(int j=0; j<M; j++)
	// 	{
	// 		for(int k=0; k<=M-2; k++)
	// 			dp[i][j] += dp[i-1][(-(j-k)%M+M)%M];
	// 		cout << "dp[" << i << "][" << j << "] = " << dp[i][j] << "\n";
	// 	}
	// 	cout << "\n";
	// }

	ll sum = 0;
	for(int i=1; i<=N; i++)
	{
		cin >> A[i];
		if (i%2 == 1) sum += A[i];
		else sum -= A[i];
	}
	sum %= M; sum += M; sum %= M;
	if (sum)
	{
		cout << "-1\n";
		return 0;
	}

	dp[0] = 0;
	for(int i=1; i<=N; i++)
	{
		if (i%2 == 0) dp[i] = (dp[i-1] * (M-1) - 1) % MOD;
		else dp[i] = (dp[i-1] * (M-1) + 1) % MOD;
	}

	sum = 0;
	ll res = 0;
	for(int i=1; i<=N-1; i++)
	{
		// 2 3 4 case
		// 0 x x
		// 1 x x

		// 2 0 x
		// 2 1 x
		// 2 2 x

		// 2 3 0
		// 2 3 1
		// 2 3 2
		// 2 3 3
		// 2 3 4

		// sum -> -sum+k
		sum = -sum, sum %= M, sum += M, sum %= M;

		if (A[i] > 0)
		{
			res += dp[N-i] * A[i], res %= MOD;
			// [sum, sum+A[i]-1]
			ll L = sum, R = sum+A[i]-1;
			if ((N-i)%2 == 1 && L <= M-1 && M-1 <= R) res--;
			if ((N-i)%2 == 0 && ((L <= 0 && 0 <= R) || (L <= M && M <= R))) res++; //
		}

		sum += A[i], sum %= M;
	}
	// i = N case
	assert(sum <= A[N]);
	res++;

	res %= MOD; res += MOD; res %= MOD;
	cout << res << "\n";
	return 0;
}
0