2019 南昌网络赛 H. The Nth Item（齐次线性递推 + 快速幂分块打表）

---

这是二阶齐次递推式，一个做法是构造二阶矩阵然后矩阵快速幂，复杂度为 $O(8\ast q\ast \log (10^9))$，T飞。
对于齐次线性递推式，可以推导它的通项公式：
令 $F(n)=x^n$：$$F(n)-3\ast F(n-1)-2\ast F(n-2)=0$$$$=x^n-3x^{n-1}-2x^{n-2}=0$$$$x^2-3x+2=0$$可以解出两个特征根：$x_1=\frac{3+\sqrt{17}}{2},x_2=\frac{3-\sqrt{17}}{2}$

那么$F(n)$ 的通解为：$c_1\ast \frac{3+\sqrt{17}}{2}+c2\ast \frac{3-\sqrt{17}}{2}$

$c_1\ast x_1+c_2\ast x_2+..+c_n\ast x_n$是$F(n)$的通解仅当 $x_1,x_2,...,x_n$是 n 个不重复的特征根。

$F(0)=0,F(1)=1$，带入可解出 $c_1=\frac{\sqrt{17}}{17},c_2=-\frac{\sqrt{17}}{17}$

$F(n)=\frac{\sqrt{17}}{17}\ast ((\frac{3+\sqrt{17}}{2}{)}^n-(\frac{3-\sqrt{17}}{2}{)}^n)$

$\sqrt{17}$的二次同余展开为：$524399943$
那么每次询问$log(10^9)$求快速幂即可，但由于 $q$达到了 $10^7$，这样还不够快。

对于$(\frac{3+\sqrt{17}}{2}{)}^n$和$(\frac{3-\sqrt{17}}{2}{)}^n$，可以分块打表记录，预处理$n=1,2,...,\sqrt{10^9}$ 以及 $n=1\ast \sqrt{10^9},2\ast \sqrt{10^9},...,\sqrt{10^9}\ast \sqrt{10^9}$的值，即可实现O(1)查询

---

代码：

#include<bits/stdc++.h>
using namespace std;
const int p = 524399943;		//sqrt(17) mod 998244353 的二次剩余 
const int mod = 998244353;
const int maxn = 1e5 + 10;
typedef long long ll;
ll fpow(ll a,ll b) {
	ll r = 1;
	while(b) {
		if(b & 1) r = r * a % mod;
		b >>= 1;
		a = a * a % mod;
	}
	return r;
} 
ll block,num;
ll mp1[maxn],np1[maxn];
ll mp2[maxn],np2[maxn];
ll inv2,invp;
ll cal(ll x) {
	x %= (mod - 1);
	ll t1 = np1[x / block] * mp1[x % block] % mod;
	ll t2 = np2[x / block] * mp2[x % block] % mod;
	return (t1 - t2 + mod) % mod * invp % mod;
}
ll n,q;
int main() {
	inv2 = fpow(2,mod - 2);
	invp = fpow(p,mod - 2);
	ll a = (3 + p) * inv2 % mod,b = (3 - p + mod) % mod * inv2 % mod;
	//分块打表，新技能
	block = sqrt(mod);
	num = mod / block + 1;
	mp1[0] = mp2[0] = 1;
	for(int i = 1; i <= block; i++) {
		mp1[i] = mp1[i - 1] * a % mod;
		mp2[i] = mp2[i - 1] * b % mod;
	}
	np1[0] = np2[0] = 1;
	for(int i = 1; i <= num; i++) {
		np1[i] = np1[i - 1] * mp1[block]  % mod;
		np2[i] = np2[i - 1] * mp2[block]  % mod;
	}
	ll ans = 0,res = 0;
	scanf("%lld%lld",&q,&n);
	while(q--) {
		n = n ^ (ans * ans);
		ans = cal(n);
		res ^= ans;
	}
	printf("%lld\n",res);
	return 0;
}