两个数组（数论、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;
}