题目:
给你一个正整数数组 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
代码:
方法一——自己写,超时:
class Solution {
public:
int gcd(int a,int b){
if(a<b)return gcd(b,a);
return b==0?a:gcd(b,a-b);
}
bool isGoodArray(vector<int>& nums) {
int len=nums.size();
if(len<=0)return false;
sort(nums.begin(),nums.end());
int last=nums[0];
if(last==1)return true;
for(int i=1;i<len;i++){
int temp=gcd(last,nums[i]);
if(temp==1)return true;
else last=temp;
}
return false;
}
};
思路:ax+by=1,if gcd(a,b)=1,扩展到多个数字,两两求最大公因数,看看是不是1;这种方法会超时;
方法二——加上一些技巧后,不超时的版本:
class Solution {
public:
int gcd(int a,int b){
if(a<b)return gcd(b,a);
return b==0?a:gcd(b,a%b);
}
bool isGoodArray(vector<int>& nums) {
int len=nums.size();
if(len<=0)return false;
sort(nums.begin(),nums.end());
int last=nums[0];
if(last==1)return true;
for(int i=1;i<len;i++){
int temp=gcd(last,nums[i]);
if(temp==1)return true;
else last=temp;
}
return false;
}
};
思路:把gcd中的a-b换为a%b,可以大大节省时间。