結果
問題 | No.1275 綺麗な式 |
ユーザー |
|
提出日時 | 2021-03-25 17:12:08 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 2,125 bytes |
コンパイル時間 | 4,660 ms |
コンパイル使用メモリ | 201,212 KB |
最終ジャッジ日時 | 2025-01-19 21:38:54 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 60 |
ソースコード
#include <bits/stdc++.h>using namespace std;const long long MOD = 1e9 + 7;typedef vector<long long> vec;typedef vector<vec> matrix;matrix eye(int n) {matrix mat(n, vec(n));for (int i = 0; i < n; i++) {mat[i][i] = 1;}return mat;}matrix zeros(int r, int c) {return matrix(r, vec(c));}matrix zeros(int n) {return zeros(n, n);}matrix add(const matrix &a, const matrix &b) {matrix c = zeros(a.size(), a[0].size());for (int i = 0; i < c.size(); i++) {for (int j = 0; j < c[0].size(); j++) {c[i][j] = a[i][j] + b[i][j];}}return c;}matrix mul(const matrix &a, const matrix &b) {matrix c = zeros(a.size(), b[0].size());for (int i = 0; i < a.size(); i++) {for (int k = 0; k < b.size(); k++) {for (int j = 0; j < b[0].size(); j++) {c[i][j] += a[i][k] * b[k][j];}}}return c;}matrix modmul(const matrix &a, const matrix &b, long long mod) {matrix c = zeros(a.size(), b[0].size());for (int i = 0; i < a.size(); i++) {for (int k = 0; k < b.size(); k++) {for (int j = 0; j < b[0].size(); j++) {c[i][j] = (c[i][j] + a[i][k] * b[k][j]) % mod;}}}return c;}matrix pow(const matrix &a, long long p) {matrix ret = eye(a.size());matrix x = a;while (p) {if (p & 1)ret = mul(ret, x);x = mul(x, x);p >>= 1;}return ret;}matrix modpow(const matrix &a, long long p, long long mod) {matrix ret = eye(a.size());matrix x = a;while (p) {if (p & 1)ret = modmul(ret, x, mod);x = modmul(x, x, mod);p >>= 1;}return ret;}long long solve(long long a, long long b, long long n) {if (n == 0) return 2;if (n == 1) return 2 * a % MOD;matrix A = eye(2);A[0][0] = 2 * a % MOD; A[0][1] = ((b - a * a) % MOD + MOD) % MOD;A[1][0] = 1; A[1][1] = 0;A = modpow(A, n-1, MOD);return (2 * a * A[0][0] % MOD + 2 * A[0][1] % MOD) % MOD;}int main() {ios_base::sync_with_stdio(0);cin.tie(0);long long a, b, n;cin >> a >> b >> n;cout << solve(a, b, n) << endl;return 0;}