【exgcd】牛客练习赛 D-青蛙兔子的约会

最新推荐文章于 2024-10-02 23:40:03 发布

lamentropetion

最新推荐文章于 2024-10-02 23:40:03 发布

阅读量500

点赞数

分类专栏：数学文章标签：动态规划算法

本文链接：https://blog.csdn.net/weixin_62528401/article/details/130516387

版权

数学专栏收录该内容

116 篇文章 0 订阅

订阅专栏

文章介绍了如何使用扩展欧几里得算法(exgcd)解决形如ax+by=n的方程，并在给定的[L,R]范围内判断是否有解。通过调整x的初始值，可以将问题转化为判断x的值是否在[L,R]之间，从而优化算法复杂度到O(1)。提供了两种不同的解决方案来确定解的存在性。

摘要由CSDN通过智能技术生成

D-青蛙兔子的约会_牛客练习赛111 (nowcoder.com)

题意：

思路：

感觉和那个青蛙的约会比较像

就是列了个方程：

a*x+b*y=n

考虑用exgcd解方程

然后看x在[L,R]有没有解

做法就是先把x的最小整数解求出来，然后考虑它的通解

x=x0+b/d*t

暴力枚举t肯定不行，因为看数据范围是1e9，因此考虑O(1)算

一种做法是根据相对位置，将L和R都减去x，然后去看L和R之间有没有k*p

这个比较细节

比较这两处的l和r即可

如果l<=r就是有解，否则就是无解

Code：

#include <bits/stdc++.h>

#define int long long 

using namespace std;

const int mxn=2e6+10; 
const int mod=1e9+7;

int a,b,n,L,R,x,y;

int exgcd(int a,int b,int &x,int &y){
	if(!b){
		x=1,y=0;
		return a;
	}
	int d=exgcd(b,a%b,y,x);
	y-=(a/b)*x;
	return d;
}
int ceil(int x,int y){
	return (x+y-1)/y;
}
void solve(){
	cin>>a>>b>>n>>L>>R;
	int d=exgcd(a,b,x,y);
	if(n%d!=0){
		cout<<"NO"<<'\n';
		return;
	}
	int p=b/d;
	x=(x*n/d)%p;
	x=(x%p+p)%p;
	int l=L-x;
	if(l<0) l=0;
	else l=(L-x+p-1)/p;
	int r=R-x;
	if(r<0) r=-1;
	else r=(R-x)/p;
	if(l<=r) cout<<"YES"<<'\n';
	else cout<<"NO"<<'\n';
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int __=1;cin>>__;
	while(__--)solve();return 0; 
}

还有一种做法就是：

如果把次数x设成从0开始，那么是不是看x在不在r-l之内就行

那就在求exgcd之前青蛙先走L次，这样次数就是从0开始了，这样就是判断次数在不在[0,r-l]之间

Code：

#include <bits/stdc++.h>

#define int long long 

using namespace std;

const int mxn=2e6+10; 
const int mod=1e9+7;

int a,b,n,L,R,x,y;

int exgcd(int a,int b,int &x,int &y){
	if(!b){
		x=1,y=0;
		return a;
	}
	int d=exgcd(b,a%b,y,x);
	y-=(a/b)*x;
	return d;
}
void solve(){
	cin>>a>>b>>n>>L>>R;
	n-=L*a;
	int d=exgcd(a,b,x,y);
	if(n%d!=0){
		cout<<"NO"<<'\n';
		return;
	}
	x*=n/d;
	y*=n/d;
	int t=b/d;
	int ans=(x%t+t)%t;
	if(ans<=R-L) cout<<"YES"<<'\n';
	else cout<<"NO"<<'\n';
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int __=1;cin>>__;
	while(__--)solve();return 0; 
}