两个数组(数论、GCD)

文章参考了我学长的博客两个数组

给定两个数组,蒜头君可以执行任意次以下操作:

  • 从第一个数组中取一个数 x,从第二个数组中取一个数 y,用 x + y或则x−y 替换 x。
    问能否使得第一个数组中元素都变成相同的数?

输入格式
第一行一个整数 T,表示数据组数。

对于每组数据:

第一行两个整数 n, m,表示两个数组的长度。

第二行 n 个整数,表示第一个数组 a。

第三行 m 个整数,表示第二个数组 b。

输出格式
对每组数据输出一行,如果能够使第一个数组的元素都相同,输出 “Yes”,否则输出 “No”。

数据范围
对于 40% 的数据,1 ≤ n , m ≤ 10^3

对于 100% 的数据,1 ≤ n , m ≤ 10^5

对于 100% 的数据,T ≤ 20,1 ≤ a[i] , b[i] ≤ 10^9

样例输入

1
5 2
1 3 5 7 9
2 3

样例输出

Yes

这一题使用了贝祖定理(或裴属定理),先简单的提一下这个定理:

如果a,b是整数且gcd(a,b) = d,那么对于任意的x,y(可以是负数),a * x + b * y 都一定是d的倍数,特别的一定存在两个整数x,y(其中一个很可能是负数),使得a * x + b * y == d成立

回到这一题,以两个特例来得到一个结论:

  • 如果数组b中只有一个元素x,那么数组a中的元素可以任意加上或减去x,即,a[i] += k * x(k为任意整数),因此当数组a中的元素都同余于x时输出"Yes"
  • 如果数组b中只有两个元素x,y,那么同理可得a[i] += c * x + d * y(c,d为任意整数)根据贝祖定理可得c * x + d * y == k * gcd(x,y),即a[i] += k * d因此当数组a中的元素均同余于gcd(x,y)时输出"Yes"
  • 那么当数组 b 中有很多元素时,计算出整个数组b的gcd,如果数组a中的元素均同余于gcd时输出"Yes"

AC代码

#include <iostream>
#include <algorithm>

using namespace std;

int main()   
{
	int T;
	cin >> T;
	
	while(T -- )
	{
		int n,m;
		int flag = true;
		cin >> n >> m;
		int a[n],b[m];
		
		for(int i = 0; i < n; i++) cin >> a[i];
		
		int gcd;//整个数组的最大公约数
		cin >> gcd;//输入数组 b 的第一个元素  
		 
		for(int i = 1; i < m; i++)
		{
			cin >> b[i];
			gcd = __gcd(gcd,b[i]);//遍历数组 b,使用库函数__gcd()求得最大公约数 	
		}
		
		for(int i = 1; i < n; i++)
		{
			if(a[i] % gcd != a[i-1] % gcd)
			{
				flag = false;
			}
		}
		if(flag) cout << "Yes" << endl;
		else cout << "No" << endl;
	}
	
	return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值