LeetCode-Python-1250. 检查「好数组」（数学）

给你一个正整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/check-if-it-is-a-good-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

利用裴蜀定理的特殊形式，也是质数的性质：

整数a和b互质当且仅当存在整数x,y使得xa+yb=1

所以只要在nums里找到一对互质的数，即可返回True。
 

Bézout's identity — Let a and b be integers with greatest common divisor d. Then, there exist integers x and y such that ax + by = d.

时间复杂度：O（N logK）, K是数组中最大值的大小。

空间复杂度：O（1）

class Solution(object):
    def isGoodArray(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        def gcd(x, y):
            while y:
                x, y = y, x % y
            return x
        g = nums[0]
        for num in nums:
            g = gcd(g, num)
        return g == 1