题目地址
先拓展一下,对于斐波那契数列也就是 $f_0 = 1, f_1 = 1, f_i = f_{i - 1} + f_{i - 2}, i \ge 2$。
那么有 $\sum_{i = 0} ^ n f_i = f_{n + 2} - 1$ 如果说是 $\sum_{i = 1} ^ n f_i = f_{n + 2} - 2$。
具体来说就是考虑用 $f_{n + 2}$ 进行展开即可。
我们回到题目,考虑题目要求出的式子 $\sum_{i = 1} ^ n f_i^k$。
我们考虑对于 $f_i$ 有通向公式 $f_i = \frac{\sqrt 5}{5} (\frac{1 + \sqrt 5}{2} - \frac{1 - \sqrt 5}{2})^ i$
为了方便我们设 $a = \frac{1 + \sqrt 5}{2}, b = \frac{1 - \sqrt 5}{2}, c = \frac{\sqrt 5}{5}$
那么我们的式子就是:
$$
\begin{aligned}
Ans &= \sum_{i = 1} ^ n f_i ^ k \\
&= \sum_{i = 1} ^ n (a - b) ^ {nk} c^k\\
&= \sum_{i = 1}^n c^k \sum_{j = 0} ^ k \binom{k}{j} a^{ij} b^ {i(k-j)} \\
&= c^k \sum_{j = 0} ^ k \binom{k}{j}\sum_{i = 1} ^ n (a^jb^{k - j}) ^ i \\
&= c^k \sum_{j = 0} ^ k \binom{k}{j} \frac{(a^jb^{k - j}) ^ {n + 1} - a^jb^{k - j}}{a^jb^{k - j}-1}\\
\end{aligned}
$$
这样其实就可以使用二次剩余做了,这题就做完了。
别忘记特判如果 $a^jb^{k - j} = 1$ 那么答案就是 $n$ 了。
拓展
但是如果说 $\sqrt 5$ 没有二次剩余呢?
我们考虑进行扩域,唯一难的地方就是除法怎么做。
我们考虑分母部分可以表示成 $x + \sqrt 5 y$ 的形式,那么我们只要上下同时乘 $x - \sqrt 5 y$ 即可。
分母就进行了有理化,之后直接就可以求整数的逆元了。
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 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
| #include <bits/stdc++.h> using namespace std; namespace Legendgod { namespace Read {
#ifdef Fread const int Siz = (1 << 21) + 5; char *iS, *iT, buf[Siz]; #define gc() ( iS == iT ? (iT = (iS = buf) + fread(buf, 1, Siz, stdin), iS == iT ? EOF : *iS ++) : *iS ++ ) #define getchar gc #endif template <typename T> void r1(T &x) { x = 0; char c(getchar()); int f(1); for(; !isdigit(c); c = getchar()) if(c == '-') f = -1; for(; isdigit(c); c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48); x *= f; } template <typename T, typename...Args> void r1(T &x, Args&...arg) { r1(x), r1(arg...); } #undef getchar }
using namespace Read;
const int maxn = 100000 + 5; const int mod = 1e9 + 9;
long long n, K; struct Node { int x, y; Node(int a = 0, int b = 0) : x(a), y(b) {} int operator == (const Node &a) { return x == a.x && y == a.y; } Node operator + (const Node &a) { return Node((x + a.x) % mod, (y + a.y) % mod); } Node operator - (const Node &a) { return Node((x - a.x + mod) % mod, (y - a.y + mod) % mod); } Node operator * (const Node &a) { int tmp1 = (1ll * a.x * x % mod + 1ll * a.y * y % mod * 5 % mod) % mod; int tmp2 = (1ll * a.x * y % mod + 1ll * a.y * x % mod) % mod; return Node(tmp1, tmp2); } }t1[maxn], t2[maxn], pw5[maxn];
int pw1[maxn], fac[maxn], finv[maxn];
int ksm(int x,int mi) { int res(1); while(mi) { if(mi & 1) res = 1ll * res * x % mod; mi >>= 1; x = 1ll * x * x % mod; } return res; }
Node ksm(Node x,int mi) { Node res(1, 0); while(mi) { if(mi & 1) res = res * x; mi >>= 1; x = x * x; } return res; }
void init(int x) { pw1[0] = 1; for(int i = 1; i <= x; ++ i) pw1[i] = 1ll * pw1[i - 1] * (mod - 1) % mod; int inv2 = ksm(2, mod - 2); t1[0].x = 1, t2[0].x = 1; t1[1] = Node(inv2, inv2); t2[1] = Node(inv2, mod - inv2); for(int i = 2; i <= x; ++ i) t1[i] = t1[i - 1] * t1[1], t2[i] = t2[i - 1] * t2[1]; fac[0] = 1; for(int i = 1; i <= x; ++ i) fac[i] = 1ll * fac[i - 1] * i % mod; finv[x] = ksm(fac[x], mod - 2); for(int i = x - 1; i >= 0; -- i) finv[i] = 1ll * finv[i + 1] * (i + 1) % mod; pw5[0].x = 1; pw5[1] = Node(0, ksm(5, mod - 2)); for(int i = 2; i <= x; ++ i) pw5[i] = pw5[i - 1] * pw5[1]; }
int C(int a,int b) { if(a < b || a < 0 || b < 0) return 0; return 1ll * fac[a] * finv[b] % mod * finv[a - b] % mod; }
signed main() { int i, j, T; init(maxn - 5); r1(T); while(T --) { r1(n, K);
Node ans; for(int j = 0; j <= K; ++ j) { Node tmp(1ll * pw1[K - j] * C(K, j) % mod); Node tmp1 = t1[j] * t2[K - j]; if(tmp1.x == 1 && tmp1.y == 0) { ans = ans + tmp * Node(n % mod, 0); continue; }
Node tmp2; Node tmp3 = tmp1; tmp3.x = (tmp3.x - 1 + mod) % mod; tmp3.y = mod - tmp3.y;
tmp2 = ksm(tmp1, (n + 1) % (mod - 1)) - tmp1; tmp2 = tmp2 * tmp3;
tmp3 = tmp3 * Node((tmp1.x - 1 + mod) % mod, tmp1.y);
int in = ksm(tmp3.x, mod - 2); ans = ans + tmp * tmp2 * Node(in, 0); } ans = ans * pw5[K];
printf("%d\n", ans.x); } return 0; }
}
signed main() { return Legendgod::main(), 0; }
|