結果

問題 No.1275 綺麗な式
ユーザー ktr216
提出日時 2020-10-30 23:00:34
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,352 bytes
コンパイル時間 1,863 ms
コンパイル使用メモリ 176,816 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-22 02:42:41
合計ジャッジ時間 3,438 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 60
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
using namespace std;

typedef long long ll;

ll a, b, n, MOD = 1e9 + 7;
map<ll, ll> mp;

ll mod_pow(ll x, ll n, ll mod) {
    ll res = 1;
    while (n > 0) {
        if (n & 1) res = res * x % mod;
        x = x * x % mod;
        n >>= 1;
    }
    return res;
}

ll solve(ll n) {
    if (mp.find(n) != mp.end()) return mp[n];
    if (n == 0) {
        mp[0] = 2;
        return mp[0];
    }
    if (n == 1) {
        mp[1] = (2 * a) % MOD;
        return mp[1];
    }
    if (n % 2) {
        mp[n] = (solve(n / 2 + 1) * solve(n / 2) - ((mod_pow((a * a - b + MOD) % MOD, n / 2, MOD) * 2 * a) % MOD) + MOD) % MOD;
    } else {
        mp[n] = (solve(n / 2) * solve(n / 2) - ((mod_pow((a * a - b + MOD) % MOD, n / 2, MOD) * 2) % MOD) + MOD) % MOD;
    }
    return mp[n];
}

int main() {
    cin >> a >> b >> n;
    if (a == 0) {
        if (b == 0) {
            cout << "0\n";
            return 0;
        } else {
            if (n % 2) {
                cout << "0\n";
                return 0;
            } else {
                cout << (mod_pow(b, n / 2, MOD) * 2) % MOD << "\n";
                return 0;
            }
        }
    }
    if (b == 0) {
        cout << (mod_pow(a, n, MOD) * 2) % MOD << "\n";
        return 0;
    }
    cout << solve(n) << "\n";
}
0