結果

問題 No.1092 modular arithmetic
ユーザー QCFium
提出日時 2020-06-24 14:12:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 63 ms / 2,000 ms
コード長 873 bytes
コンパイル時間 1,496 ms
コンパイル使用メモリ 168,100 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-03 20:09:02
合計ジャッジ時間 3,552 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

int ri() {
    int n;
    scanf("%d", &n);
    return n;
}

int mpow(int a, int b, int mod) {
    int res = 1;
    for (; b; b >>= 1) {
        if (b & 1) res = (int64_t) res * a % mod;
        a = (int64_t) a * a % mod;
    }
    return res;
}

int main() {
    int p = ri();
    int n = ri();
    int a[n];
    for (auto &i : a) i = ri();
    std::string s;
    std::cin >> s;
    int cur = a[0];
    for (int i = 0; i + 1 < n; i++) {
        if (s[i] == '+') {
            cur += a[i + 1];
            if (cur >= p) cur -= p;
        } else if (s[i] == '-') {
            cur -= a[i + 1];
            if (cur < 0) cur += p;
        } else if (s[i] == '*') {
            cur = (int64_t) cur * a[i + 1] % p;
        } else {
            cur = (int64_t) cur * mpow(a[i + 1], p - 2, p) % p;
        }
    }
    printf("%d\n", cur);
    return 0;
}
0