結果

問題 No.1275 綺麗な式
コンテスト
ユーザー ktr216
提出日時 2020-10-30 22:49:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 885 bytes
コンパイル時間 1,762 ms
コンパイル使用メモリ 178,148 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-22 02:28:45
合計ジャッジ時間 3,553 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 14 WA * 46
権限があれば一括ダウンロードができます

ソースコード

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;
        return mp[1];
    }
    if (n % 2) {
        mp[n] = (solve(n / 2 + 1) * solve(n / 2) - ((mod_pow(a * a - b, n / 2, MOD) * 2 * a) % MOD) + MOD) % MOD;
    } else {
        mp[n] = (solve(n / 2) * solve(n / 2) - ((mod_pow(a * a - b, n / 2, MOD) * 2) % MOD) + MOD) % MOD;
    }
    return mp[n];
}

int main() {
    cin >> a >> b >> n;
    cout << solve(n) << "\n";
}
0