【BZOJ】4002: [JLOI2015]有意义的字符串

题意

求$\left \lfloor \left( \frac{b+\sqrt{d}}{2} \right)^n \right \rfloor \pmod {7528443412579576937} \(，\)\left( 0 \le n \le 10^{18}, 0 < b^2 \le d < (b+1)^2 \le 10^{18}, b \mbox{ mod } 2 = 1, d \mbox{ mod } 4=1 \right) $

分析

发现这个并不好算，而如果是\(\left( \frac{b-\sqrt{d}}{2} \right)^n\)那么就好算了。于是又想到数列的特征方程得到的解\(a_n = c_1 x_1^n + c_2 x_2^n\)，于是我们搞搞。直接将\(c_1 = c_2 = 1\)，则变成\(a_n = x_1^n + x_2^n\)，而我们知道\(x_1、x_2\)是特征方程的两个解，和上面那个形式极为相似，于是我们继续假设。即\(x_1 = \left( \frac{b+\sqrt{d}}{2} \right), x_2 = \left( \frac{b-\sqrt{d}}{2} \right)\)。则\(a_{n+2} = pa_{n+1} + qa_{n}\)中，\(p = x_1 + x_2, q = - x_1 x_2\)，因此得到\(a_{n+2} = ba_{n+1} - \frac{b^2-d}{4}a_{n}\)

题解

根据上面这个递推式，我们容易算出其中两项，容易得到\(a_1 = b, a_2 = \frac{b^2+d}{2}\)。而发现通项求出来的是整数，因此我们用矩阵乘法求出\(a_n\)即可。最后再根据条件特判一下\(\left( \frac{b-\sqrt{d}}{2} \right)^n\)即可，即\(ans = a_n - [b^2 \neq d \land n是偶数]\)

注意n=0要特判...

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ll;
typedef ll mtx[2][2];
const ll mo=7528443412579576937ull, Lim=1e9;
inline void CK(ll &c) {
if(c>=mo)
c-=mo;
}
inline ll mul(ll a, ll b) {
if(a<=Lim && b<=Lim) {
return a*b;
}
if(a<b) {
swap(a, b);
}
ll c=0;
for(; b; b>>=1, CK(a<<=1)) {
if(b&1) {
CK(c+=a);
}
}
return c;
}
void mul(mtx a, mtx b, mtx c, int la, int lb, int lc) {
static mtx t;
memset(t, 0, sizeof t);
for(int i=0; i<la; ++i) {
for(int j=0; j<lc; ++j) {
for(int k=0; k<lb; ++k) {
CK(t[i][j]+=mul(a[i][k], b[k][j]));
}
}
}
memcpy(c, t, sizeof t);
}
ll b, d, n;
bool spj(ll n) {
if(n==1) {
printf("%lld\n", (ll)((((double)b+sqrt(d))/2.0)));
}
else if(n==2) {
printf("%lld\n", (b*b+d)/2);
}
return n<=2;
}
mtx a, c;
int main() {
scanf("%lld%lld%lld", &b, &d, &n);
if(spj(n)) {
return 0;
}
ll t1=b, t2=(d-b*b)/4;
CK(t1), CK(t2);
a[0][0]=t1, a[0][1]=1;
a[1][0]=t2, a[1][1]=0;
c[0][0]=c[1][1]=1;
for(ll tt=n-2; tt; tt>>=1, mul(a, a, a, 2, 2, 2)) {
if(tt&1) {
mul(c, a, c, 2, 2, 2);
}
}
ll a2=(b*b+d)/2, a1=b;
CK(a1), CK(a2);
ll ans;
CK(ans=mul(a2, c[0][0])+mul(a1, c[1][0]));
if(b*b!=d && (n&1)==0) {
if(ans==0) {
ans=mo-1;
}
else {
ans--;
}
}
printf("%llu\n", ans);
return 0;
}