[ZROI2021]整数对

题目

题目概要
求满足如下条件的非负整数 $⟨x,y⟩$ 的个数：
$$x\le n,y\le \sqrt{p/q}\cdot x$$

数据范围与提示
多组数据，$T\le 10^5$ 且 $p,q\le 10^3$，但是 $n\le 10^9$ 。

思路

看上去非常像类欧几里得啊！然而，如果用同样的方法去推导，就会推出无数个 $\mathcal{O}(n)$ 的做法……

为啥无理数没法消除呢？因为它不能像整数那样，消去一个整数部分之后，剩下的仍然是有理数。所以说很多方法都行不通。

如果你还做过等比数列三角形，你就会往这个方向想，然后又得到了零分的高分！

现在来瞟一眼，正解是个啥……显然式子等价于
$$n+1+\sum _{i=1}^n\sum _{j=1}^{+\infty }\left[j\le \sqrt{p/q}\cdot i\right]$$
里面那个真值括号，可以移项得到 $\frac{j}{i}\le \sqrt{p/q}$ 。那么，我们 用一个有理数逼近这个无理数，就可以转化为类欧了。具体而言，我们应当找到分母不超过 $n$ 的所有分数中，最接近 $\sqrt{p/q}$ 的一个（不然就会统计漏）。

这件事听上去很有道理，又好像不容易进行。所以 $\mathsf{H}\mathsf{a}\mathsf{n}\mathsf{d}\mathsf{I}\mathsf{n}\mathsf{D}\mathsf{e}\mathsf{v}\mathsf{i}\mathsf{l}$ 用这个方法把它秒了的时候，我还不太相信。因为 分母设置为 $10^9$ 不意味着最接近 $\sqrt{p/q}$ 。可能有比它略大一点的分数，并且分母和分子都更小，更接近 $\sqrt{p/q}$ 呢？

需要利用 $\mathrm{S}\mathrm{t}\mathrm{e}\mathrm{r}\mathrm{n}\text{-}\mathrm{B}\mathrm{r}\mathrm{o}\mathrm{c}\mathrm{o}\mathrm{t}$ 树进行有理逼近。根据已有理论，我们知道只会有 $\mathcal{O}(\log n)$ 条 “链”，用 $\mathcal{O}(\log n+\log p)$ 的倍增来跳跃每一条链。

如果是有理数，真值括号变为 $[\frac{j}{i}\le \frac{a}{b}]$ 时，即有 $[j\le \frac{ai}{b}]$，那么我们只需要求
$$n+1+\sum _{i=1}^n\lfloor \frac{a\cdot i}{b}\rfloor$$
这就是类欧了。复杂度 $\mathcal{O}[T\log n(\log n+\log q)]$ 。

代码

其实我有些怠惰——按道理，分子最高可达 $n\sqrt{p/q}$，所以倍增应该把 $j$ 设置大一点。但是我还是开的 $30$ 😄

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(__int128 x){
	if(x > 9) writeint(x/10);
	putchar((char)(x-x/10*10)^48);
}

bool check(int p,int q,int_ x,int_ y){
	return __int128(x)*x*q <= __int128(y)*y*p;
}

int_ son, mom; int threshold;
void dfs(int p,int q,int_ lx=0,int_ ly=1,int_ rx=1,int_ ry=0){
	int_ mx = lx+rx, my = ly+ry;
	if(my > threshold) return ;
	if(check(p,q,mx,my)){
		for(int j=30; ~j; --j)
			if(my+(1<<j)*ry <= threshold)
			if(check(p,q,mx+(1<<j)*rx,my+(1<<j)*ry))
				mx += (1<<j)*rx, my += (1<<j)*ry;
		son = mx, mom = my;
		return dfs(p,q,mx,my,rx,ry);
	}
	for(int j=30; ~j; --j)
		if(my+(1<<j)*ly <= threshold)
		if(!check(p,q,mx+(1<<j)*lx,my+(1<<j)*ly))
			mx += (1<<j)*lx, my += (1<<j)*ly;
	return dfs(p,q,lx,ly,mx,my);
}
void approximation(int p,int q,int n){
	son = 0, mom = 1; // no number exist
	threshold = n; dfs(p,q);
}

__int128 likeGcd(__int128 a,__int128 b,__int128 c,__int128 n){
	if(n == 0) return 0; // nothing left
	__int128 m = ((n-1)*a+b)/c;
	if(a >= c || b >= c){
		__int128 f = likeGcd(a%c,b%c,c,n);
		f += (a/c)*(n*(n-1)>>1);
		f += (b/c)*n; return f;
	}
	else{
		if(!a) return 0; // since b < c
		return m*n-likeGcd(c,c-b+a-1,a,m);
	}
}

int main(){
	int T = readint(), p, q, n;
	while(T --){
		p = readint(), q = readint();
		n = readint();
		approximation(p,q,n);
		__int128 ans = likeGcd(son,son,mom,n);
		writeint(ans+n+1), putchar('\n');
	}
	return 0;
}