結果

問題 No.2144 MM
ユーザー ぷらぷら
提出日時 2022-12-02 22:36:00
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 109 ms / 2,000 ms
コード長 1,722 bytes
コンパイル時間 2,155 ms
コンパイル使用メモリ 204,948 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-10 00:10:38
合計ジャッジ時間 4,702 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

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