結果

問題 No.2144 MM
ユーザー ぷらぷら
提出日時 2022-12-02 22:36:00
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 107 ms / 2,000 ms
コード長 1,722 bytes
コンパイル時間 2,619 ms
コンパイル使用メモリ 198,884 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-18 07:19:39
合計ジャッジ時間 5,306 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 67 ms
5,376 KB
testcase_08 AC 107 ms
5,376 KB
testcase_09 AC 66 ms
5,376 KB
testcase_10 AC 106 ms
5,376 KB
testcase_11 AC 106 ms
5,376 KB
testcase_12 AC 107 ms
5,376 KB
testcase_13 AC 103 ms
5,376 KB
testcase_14 AC 106 ms
5,376 KB
testcase_15 AC 88 ms
5,376 KB
testcase_16 AC 25 ms
5,376 KB
testcase_17 AC 76 ms
5,376 KB
testcase_18 AC 21 ms
5,376 KB
testcase_19 AC 80 ms
5,376 KB
testcase_20 AC 72 ms
5,376 KB
testcase_21 AC 53 ms
5,376 KB
testcase_22 AC 13 ms
5,376 KB
testcase_23 AC 75 ms
5,376 KB
testcase_24 AC 22 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 2 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 2 ms
5,376 KB
testcase_36 AC 2 ms
5,376 KB
testcase_37 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

constexpr int mod = 998244353;

long long modpow(long long a,long long b) {
    long long ans = 1;
    while(b) {
        if(b & 1) {
            (ans *= a) %= mod;
        }
        (a *= a) %= mod;
        b /= 2;
    }
    return ans;
}

int main() {
    int N,M;
    cin >> N >> M;
    vector<int>A(N);
    int sum = 0;
    for(int i = 0; i < N; i++) {
        cin >> A[i];
        if(i%2 == 0) {
            sum += A[i];
            sum %= M;
        }
        else {
            sum += M-A[i];
            sum %= M;
        }
    }
    if(sum != 0) {
        cout << -1 << endl;
        return 0;
    }
    vector<int>f(N+1);
    f[0] = 1;
    for(int i = 0; i < N; i++) {
        f[i+1] = f[i];
        if(i%2 == 0) {
            f[i+1] += mod-modpow(M-1,i+1);
            f[i+1] %= mod;
        }
        else {
            f[i+1] += modpow(M-1,i+1);
            f[i+1] %= mod;
        }
    }
    int res = 0,ans = 0;
    for(int i = 0; i+1 < N; i++) {
        int tmp = A[i]*modpow(M-1,N-i-2)%mod;
        if(i+2 < N) {
            int a = f[N-i-3];
            if((N-i-3)%2 == 0) {
                a = mod-a;   
            }
            tmp += 1ll*A[i]*a%mod;
            tmp %= mod;
        }
        if((N-i)%2 == 1 && (M-res)%M < A[i]) {
            tmp += 1;
            tmp %= mod;
        }
        if((N-i)%2 == 0 && (res+M-1)%M < A[i]) {
            tmp += mod-1;
            tmp %= mod;
        }
        ans += tmp;
        ans %= mod;
        if((N-i)%2 == 1) {
            res += A[i];
            res %= M;
        }
        else {
            res += M-A[i];
            res %= M;
        }
    }
    ans++;
    ans %= mod;
    cout << ans << endl;
}
0