1250-检查「好数组」

Raphael Liu Lv10

给你一个正整数数组 nums,你需要从中任选一些子集,然后将子集中每一个数乘以一个 任意整数 ,并求出他们的和。

假如该和结果为 1,那么原数组就是一个「 好数组 」,则返回 True;否则请返回 False

示例 1:

**输入:** nums = [12,5,7,23]
**输出:** true
**解释:** 挑选数字 5 和 7。
5*3 + 7*(-2) = 1

示例 2:

**输入:** nums = [29,6,10]
**输出:** true
**解释:** 挑选数字 29, 6 和 10。
29*1 + 6*(-3) + 10*(-1) = 1

示例 3:

**输入:** nums = [3,6]
**输出:** false

提示:

  • 1 <= nums.length <= 10^5
  • 1 <= nums[i] <= 10^9

前言

本题解涉及到数论中的「裴蜀定理」(具体证明可参考「裴蜀定理」百度百科 )。

「裴蜀定理」的内容为:对于不全为零的任意整数 a 和 b,记 g = \gcd(a,b),其中 \gcd(a, b) 为 a 和 b 的最大公约数,则对于任意整数 x 和 y 都满足 a\times x+b \times y 是 g 的倍数,特别地,存在整数 x 和 y 满足 a \times x + b \times y = g。

「裴蜀定理」也可以推广到多个整数的情况。对于不全为零的任意 n 个整数 a_1, a_2, \dots a_n,记这 n 个数的最大公约数为 g,则对于任意 n 个整数 x_1, x_2, \dots x_n 都满足 \sum_{i = 1}^n a_i \times x_i 是 g 的倍数。一个重要的推论是:正整数 a_1 到 a_n 的最大公约数是 1 的充分必要条件是存在 n 个整数 x_1 到 x_n 满足 \sum_{i = 1}^n a_i \times x_i = 1。

方法一:数论

思路与算法

题目给出一个正整数数组 nums,现在我们需要从中任选一些子集,然后将子集中的每一个数都乘以一个任意整数并求出他们的和,如果该和的结果为 1,那么原数组就是一个「好数组」。现在我们需要判断数组 nums 是否是一个「好数组」。由「裴蜀定理」可得,题目等价于求 nums 中的全部数字的最大公约数是否等于 1,若等于 1 则原数组为「好数组」,否则不是。

求 nums 中全部数字的最大公约数的方法为,我们设初始为 x = \textit{nums}[0],然后对于每一个数 nums}[i],0 < i < n,我们更新 x = \gcd(x, \textit{nums}[i])。遍历完全部数字后,x 即为数组 nums 中全部的元素的最大公约数。然后判断其是否等于 1 即可。在实现过程中我们也可以进一步做优化:如果遍历过程中出现最大公约数等于 1 的情况,则由于 1 和任何正整数的最大公约数都是 1,此时可以提前结束遍历。

代码

[sol1-Python3]
1
2
3
class Solution:
def isGoodArray(self, nums: List[int]) -> bool:
return reduce(gcd, nums) == 1
[sol1-Java]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
public boolean isGoodArray(int[] nums) {
int divisor = nums[0];
for (int num : nums) {
divisor = gcd(divisor, num);
if (divisor == 1) {
break;
}
}
return divisor == 1;
}

public int gcd(int num1, int num2) {
while (num2 != 0) {
int temp = num1;
num1 = num2;
num2 = temp % num2;
}
return num1;
}
}
[sol1-C#]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public class Solution {
public bool IsGoodArray(int[] nums) {
int divisor = nums[0];
foreach (int num in nums) {
divisor = GCD(divisor, num);
if (divisor == 1) {
break;
}
}
return divisor == 1;
}

public int GCD(int num1, int num2) {
while (num2 != 0) {
int temp = num1;
num1 = num2;
num2 = temp % num2;
}
return num1;
}
}
[sol1-C++]
1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
public:
bool isGoodArray(vector<int>& nums) {
int divisor = nums[0];
for (int num : nums) {
divisor = gcd(divisor, num);
if (divisor == 1) {
break;
}
}
return divisor == 1;
}
};
[sol1-C]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
int gcd(int num1, int num2) {
while (num2 != 0) {
int temp = num1;
num1 = num2;
num2 = temp % num2;
}
return num1;
}

bool isGoodArray(int* nums, int numsSize) {
int divisor = nums[0];
for (int i = 0; i < numsSize; i++) {
divisor = gcd(divisor, nums[i]);
if (divisor == 1) {
break;
}
}
return divisor == 1;
}
[sol1-JavaScript]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
var isGoodArray = function(nums) {
let divisor = nums[0];
for (const num of nums) {
divisor = gcd(divisor, num);
if (divisor === 1) {
break;
}
}
return divisor === 1;
}

const gcd = (num1, num2) => {
while (num2 !== 0) {
const temp = num1;
num1 = num2;
num2 = temp % num2;
}
return num1;
};
[sol1-Golang]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
func isGoodArray(nums []int) bool {
g := 0
for _, x := range nums {
g = gcd(g, x)
if g == 1 {
return true
}
}
return false
}

func gcd(a, b int) int {
for a != 0 {
a, b = b%a, a
}
return b
}

复杂度分析

  • 时间复杂度:O(n + \log m),其中 n 为数组 nums 的长度,m 为数组 nums 中的最大数,其中求单次最大公约数的时间复杂度为 O(\log m),由于在每次求两个数的最大公约数时其中一个数保持单调不增,所以求总的公约数的时间复杂度为 O(\log m)。
  • 空间复杂度:O(1)。仅使用常量空间。
 Comments
On this page
1250-检查「好数组」