#671104 (C++14) No.1559 Next Rational

• 問題: No.1559 Next Rational
• ユーザー: tnakao0123
• 提出日時: 2021-06-28 14:02:57
• 言語: C++14 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 1,927 bytes
• コンパイル時間: 876 ms
• コンパイル使用メモリ: 95,628 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-06-25 12:09:53
• 合計ジャッジ時間: 1,607 ms
• ジャッジサーバーID （参考情報）: judge1 / judge5

ファイルパターン	結果
sample	AC * 3
other	AC * 15

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/* -*- coding: utf-8 -*-
 *
 * 1559.cc:  No.1559 Next Rational - yukicoder
 */
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;
/* constant */
const int MOD = 1000000007;
/* typedef */
typedef long long ll;
typedef int vec[2];
typedef vec mat[2];
/* global variables */
mat ma, mb;
/* subroutines */
inline void initvec(vec a) { memset(a, 0, sizeof(vec)); }
inline void initmat(mat a) { memset(a, 0, sizeof(mat)); }
inline void unitmat(mat a) {
  a[0][0] = a[1][1] = 1; a[0][1] = a[1][0] = 0;
}
inline void copymat(const mat a, mat b) {
  memcpy(b, a, sizeof(mat));
}
inline void mulmat(const mat a, const mat b, mat c) {
  for (int i = 0; i < 2; i++)
    for (int j = 0; j < 2; j++) {
      c[i][j] = 0;
      for (int k = 0; k < 2; k++)
        c[i][j] = (c[i][j] + (ll)a[i][k] * b[k][j] % MOD) % MOD;
    }
}
inline void powmat(const mat a, ll b, mat c) {
  mat s, t;
  copymat(a, s);
  unitmat(c);
  while (b > 0) {
    if (b & 1) {
      mulmat(c, s, t);
      copymat(t, c);
    }
    mulmat(s, s, t);
    copymat(t, s);
    b >>= 1;
  }
}
int powmod(int a, int n) {  // a^n % MOD
  int pm = 1;
  while (n > 0) {
    if (n & 1) pm = (ll)pm * a % MOD;
    a = (ll)a * a % MOD;
    n >>= 1;
  }
  return pm;
}
/* main */
int main() {
  ll n;
  int a, b, k;
  scanf("%lld%d%d%d", &n, &a, &b, &k);
  int x = (((ll)a * a % MOD + (ll)b * b % MOD) % MOD + k) % MOD;
  int y = (ll)a * b % MOD;
  
  ma[0][0] = 0, ma[0][1] = 1;
  ma[1][0] = (MOD - 1), ma[1][1] = (ll)x * powmod(y, MOD - 2) % MOD;
  
  powmat(ma, n - 1, mb);
  int v = ((ll)mb[0][0] * a % MOD + (ll)mb[0][1] * b % MOD) % MOD;
  printf("%d\n", v);
  return 0;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX