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

题意

求$left lfloor left( frac{b+sqrt{d}}{2} ight)^n ight floor 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 ight) $

分析

发现这个并不好算，而如果是(left( frac{b-sqrt{d}}{2} ight)^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} ight), x_2 = left( frac{b-sqrt{d}}{2} ight))。则(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} ight)^n)即可，即(ans = a_n - [b^2 eq 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
", (ll)((((double)b+sqrt(d))/2.0)));
	}
	else if(n==2) {
		printf("%lld
", (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
", ans);
	return 0;
}