51nod1145 斐波那契序列

题面

题目链接

解题思路

需要一个前置知识$fib[i]\mathrm{m}\mathrm{o}\mathrm{d}10^k$是有周期的，周期是$6\ast 10^k$。具体证明见51nod1195的题解。
考虑按位倍增处理答案。先处理出所有的位置$i,fib[i]\equiv n(mod10)$。
然后考虑如何扩展到mod100。我们知道所有的位置$j,fib[j]\equiv n(mod100)$都满足$j\equiv i(mod60)$。
因此可以考虑从i的集合扩展到j的集合。
如此扩展18次后可以得到所有小于$6\ast 10^{19}$的答案，取最小的即可。
复杂度不会算，总之很快就是了，跑了31ms。

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;

typedef long long ll;
typedef __int128 Int;

const Int MOD = 1e18;

void write(Int x) {
	if (x > 9) write(x / 10);
	putchar(x % 10 + '0');
}

Int fu[5][5], co[5][5], res[5][5];

void cal(Int a[][5], Int b[][5]) {
	for (int i = 0; i < 2; i++)
		for (int j = 0; j < 2; j++)
			for (int k = 0; k < 2; k++)
				fu[i][j] += a[i][k] * b[k][j];
	for (int i = 0; i < 2; i++)
		for (int j = 0; j < 2; j++)
			a[i][j] = fu[i][j] % MOD, fu[i][j] = 0;
}

Int get(Int n) {
	if (n == 0) return 0;
	res[0][1] = res[1][0] = co[1][1] = 0;
	res[0][0] = res[1][1] = co[0][0] = co[0][1] = co[1][0] = 1;
	while (n > 0) {
		if (n & 1) cal(res, co);
		cal(co, co); n /= 2;
	}
	return res[1][0];
}

ll n;
ll dp[110];

Int solve(ll n) {
	dp[1] = 1;
	for (int i = 2; i < 60; i++) dp[i] = (dp[i - 1] + dp[i - 2]) % 10;
	vector<Int> V, G;
	for (int i = 0; i < 60; i++) if (dp[i] == n % 10) V.push_back(i);
	Int r = 10;
	for (Int i = 2; i <= 18; i++) {
		r *= 10;
		G.clear();
		for (Int x : V) {
			while (x < r * 6) {
				if (get(x) % r == n % r) G.push_back(x);
				x += r / 10 * 6;
			}
		}
		swap(V, G);
	}
	if (V.size() == 0) { puts("No Solution"); return -1; }
	sort(V.begin(), V.end());
	return V[0];
}

int main() {
	scanf("%lld", &n);
	Int ans = solve(n);
	if (ans >= 0) write(ans + 1), puts("");
	return 0;
}
``