Newcoder 148 H.Rikka with Ants（计算几何+递归）

Description

二维平面上有两只蚂蚁初始在$(1,0)$点，有三条直线：

$1.y=0$，没有蚂蚁可以越过这条线

$2.y=\frac{a}{b}x(a,b>0)$，只有第二只蚂蚁可以越过这条线

$3.y=\frac{c}{d}x(c,d>0)$，只有第一只蚂蚁可以越过这条线

注意到蚂蚁只能往上走和往右走，且如果蚂蚁可以往上走就先往上走，不能往上走了再往右走，故两只蚂蚁的路径唯一，问被两只蚂蚁均经过的整点个数

Input

第一行一整数$T$表示用例组数，每组用例输入四个整数$a,b,c,d$

$(1\le T\le 10^5,1\le a,b,c,d\le 10^9)$

Output

如果两只蚂蚁均经过的整点有无限个则输出$-1$，否则输出整点个数，结果模$998244353$

Sample Input

5
1 1 1 1
1 2 1 1
1 3 2 1
1 100 1 99
12 34 56 78

Sample Output

-1
2
1
5049
3

Solution

显然若两条直线重合则两只蚂蚁均经过的整点有无穷个，假设$\frac{a}{b}>\frac{c}{d}$，两只蚂蚁若都经过$(x,y)$点，那么蚂蚁要么从$(x,y-1)$往上走到达$(x,y)$点，要么从$(x-1,y)$点无法到达$(x-1,y+1)$点只能往右走到达$(x,y)$点，故有
$$\{\begin{array}{l}y\le \frac{a}{b}x\\ y\le \frac{c}{d}x\\ y+1>\frac{a}{b}(x-1)\\ y+1>\frac{c}{d}(x-1)\end{array}$$
由$\frac{a}{b}>\frac{c}{d}$知只需满足$2,3$式即可，问题转化为求$y=\frac{c}{d}x$和$y+1=\frac{a}{b}(x-1)$与$x$轴围成三角形中整点的个数（不包括原点以及$y+1=\frac{a}{b}(x-1)$上的整点），这显然是有穷个点，假设两条直线内整点横坐标最大值为$X$，则轻易得到
$$X=\lfloor \frac{(a+b)d}{ad-bc}\rfloor$$
那么我们只要求出$y=\frac{c}{d}x$在$x\in [1,X]$范围内与$x$轴围成整点的个数，再减去$y+1=\frac{a}{b}(x-1)$在$x\in [1,X]$范围内与$x$轴围成的纵坐标非负的整点个数即为答案

对于第一部分，显然有
$$X+\sum _{x=1}^X\lfloor \frac{cx}{d}\rfloor$$
对于第二部分，先把直线向左平移一个单位，然后想上平移一个答案，则第二部分答案等价于$y=\frac{a}{b}x$在$x\in [0,X-1]$范围内与$x$轴围成的纵坐标为正的整点个数，也即为
$$\sum _{x=0}^{X-1}\lfloor \frac{ax}{b}\rfloor$$
令$f(a,b,c,n)=\sum _{x=0}^n\lfloor \frac{ax+b}{c}\rfloor$，则所求答案为$X+f(c,0,d,X)-f(a,0,b,X-1)$，只要求出$f$即可

1.若$a=0$，则显然$f(a,b,c,n)=(n+1)\cdot \lfloor \frac{b}{c}\rfloor$

2.若$a,b$不全小于$c$，那么把这个下取整的两块整数拿出来，也即
$$\lfloor \frac{ax+b}{c}\rfloor =\lfloor \frac{a}{c}\rfloor \cdot x+\lfloor \frac{b}{c}\rfloor +\lfloor \frac{(a\%c)\cdot x+b\%c}{c}\rfloor$$
进而有
$$f(a,b,c,n)=f(a\%c,b\%c,c,n)+\frac{n(n+1)}{2}\cdot \lfloor \frac{a}{c}\rfloor +(n+1)\cdot \lfloor \frac{b}{c}\rfloor$$
3.若$0<a<c,0\le b<c$，回到该式的几何意义，即为直线$y=\frac{ax+b}{c}$在$x\in [0,n]$范围内与$x$轴围成的区域中纵坐标为正的整点个数，由于$\frac{b}{c}<1$，故该范围没有横坐标$x=0$的整点，记该区域整点纵坐标最大值为$m$，则有
$$m=\lfloor \frac{an+b}{c}\rfloor$$
对于$[1,n]\times [1,m]$范围内的$nm$个整点，不合法整点为直线$y=\frac{ax+b}{c}$与$y$轴围成的、横坐标为正且不在直线上的整点，这些整点也即为$y=\frac{ax+b-1}{c}$与$y$轴围成的横坐标为正的整点个数，交换两个坐标轴即为$y=\frac{cx-b-1}{a}$与$x$轴在$x\in [1,m]$范围内围成区域的纵坐标为正的整点个数，也即为$f(c,c-b-1,a,m-1)$，故此时有
$$f(a,b,c,n)=nm-f(c,c-b-1,a,m-1)$$
递归求解即可，注意在计算$m$时，$an$可能会爆$longlong$，要用$\_\_int128$

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef __int128 lll;
#define mod 998244353
#define inv2 499122177 
int mul(int x,int y)
{
	ll z=1ll*x*y;
	return z-z/mod*mod;
}
int add(int x,int y)
{
	x+=y;
	if(x>=mod)x-=mod;
	return x;
}
int f(int a,int b,int c,ll n)
{
	if(a==0)return mul((n+1)%mod,b/c);
	if(a>=c||b>=c)
	{
		int res=mul(mul(mul(n%mod,(n+1)%mod),inv2),a/c);
		res=add(res,mul((n+1)%mod,b/c));
		return add(res,f(a%c,b%c,c,n));
	}
	ll m=((lll)a*n+b)/c;
	return add(mul(n%mod,m%mod),mod-f(c,c-b-1,a,m-1));
}
int main()
{
	int T,a,b,c,d;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d%d",&a,&b,&c,&d);
		if(1ll*a*d==1ll*b*c)
		{
			printf("-1\n");
			continue;
		}
		if(1ll*a*d<1ll*b*c)swap(a,c),swap(b,d);
		ll x=(1ll*(b+a)*d)/(1ll*a*d-1ll*b*c);
		int ans=add(add(x%mod,f(c,0,d,x)),mod-f(a,0,b,x-1));
		printf("%d\n",ans);
	} 
	return 0;
}