-
Notifications
You must be signed in to change notification settings - Fork 10
/
QTLOVE2.cpp
79 lines (64 loc) 路 1.44 KB
/
QTLOVE2.cpp
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
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
typedef unsigned long long ull;
#define X first
#define Y second
#define pb push_back
#define mp make_pair
#define ep emplace_back
#define EL printf("\n")
#define sz(A) (int) A.size()
#define FOR(i,l,r) for (int i=l;i<=r;i++)
#define FOD(i,r,l) for (int i=r;i>=l;i--)
#define fillchar(a,x) memset(a, x, sizeof (a))
#define faster ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
const int mod = 1e9+7;
struct mat {
int x, y;
ll a[3][3];
};
mat operator * (mat a, mat b) {
mat c {a.x, b.y};
FOR(i,1,c.x) FOR(j,1,c.y) c.a[i][j] = 0ll;
FOR(i,1,c.x)
FOR(j,1,c.y)
FOR(k,1,a.y)
c.a[i][j] = (c.a[i][j] + a.a[i][k] * b.a[k][j] % mod) % mod;
return c;
}
mat power(mat a, ll x) {
if (x == 1) return a;
mat b = power(a, x/2);
return x%2 ? b*b*a : b*b;
}
ll n, m;
mat a, x;
int main() {
// freopen("INP.TXT", "r", stdin);
// freopen("OUT.TXT", "w", stdout);
cin >> n >> m;
if (n <= 3) {
ll ans = m;
if (n == 2) ans = m * (m-1) % mod;
if (n == 3) ans = m * (m-1) % mod * (m-2) % mod;
printf("%lld\n", ans);
return 0;
}
if (m == 1) {
puts("0");
return 0;
}
a.x = 2;
a.y = 1;
a.a[1][1] = m * (m-1) % mod * (m-2) % mod;
a.a[2][1] = m * (m-1) % mod * (m-1) % mod * (m-1) % mod;
x.x = x.y = 2;
x.a[1][1] = -1;
x.a[1][2] = 1;
x.a[2][2] = m-1;
a = power(x, n-3) * a;
printf("%lld\n", (a.a[1][1] + (ll)mod) % mod);
return 0;
}