JZOJ7177-鱼跃龙门

Description&Data Constraint

$1\le T\le 100,1\le n\le 10^{12}$

Solution

简单来说，就是求满足 $\frac{x(x+1)}{2}\%n=0$ 的最小的 $x$ 。

推导一下：

$n|\frac{x(x+1)}{2}$

$2n|x(x+1)$

枚举 $a|2n$，令 $b=\frac{2n}{a}$

设 $ap=x+1,bq=x$，联立得到 $ap-bq=1$。

要求 $bq$ 的最小值，扩欧经典题。

有解满足 $\gcd (a,b)=1$，因此只用判断 $2n$ 的每个质因数是否给 $a$，要么不给，要么全给。

时间复杂度 $O(\text{求质因数的时间}+2^{\text{质因子个数}}\times \log )$

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define MX 500000
#define N 10000000
using namespace std;
int t,bz[MX];
ll n,num,ans,p[N],prime[MX];
void exgcd(ll a,ll b,ll &x,ll &y)
{
	if (b==0)
	{
		x=1;
		y=0;
		return;
	}
	exgcd(b,a%b,x,y);
	ll u=y,v=x-(a/b)*y;
	x=u;
	y=v;
}
void dg(ll k,ll x)
{
	if (k>num)
	{
		ll y=n/x;
		ll xx,yy;
		exgcd(x,y,xx,yy);
		xx=-xx;
		if (xx<=0) xx+=y;
		ans=min(ans,x*xx);
		return;
	}
	dg(k+1,x);
	dg(k+1,x*p[k]);
}
int main()
{
	for (int i=2;i<=MX;++i)
	{
		if (!bz[i]) prime[++prime[0]]=i;
		for (int j=1;j<=prime[0]&&i*prime[j]<=MX;++j)
		{
			bz[i*prime[j]]=1;
			if (i%prime[j]==0) break;
		}
	}
	scanf("%d",&t);
	while (t--)
	{
		ans=12345679887654321;
		scanf("%lld",&n);
		num=0;n*=2;
		ll nn=n;
		for (int i=1;i<=prime[0];++i)
		{
			if (nn%prime[i]==0)
			{
				p[++num]=1;
				while (nn%prime[i]==0)
				{
					p[num]*=prime[i];
					nn/=prime[i];
				}
			}
		}
		if (nn>1) p[++num]=nn;
		dg(1,1);
		printf("%lld\n",ans);
	}	
	return 0;
}