给你一个整数数组 target 。一开始,你有一个数组 A ,它的所有元素均为 1 ,你可以执行以下操作:
x
为你数组里所有元素的和0 <= i < target.size
的任意下标 i ,并让 A 数组里下标为 i
处的值为 x
。如果能从 A
开始构造出目标数组 target
,请你返回 True ,否则返回 False 。
示例 1:
输入:target = [9,3,5]
输出:true
解释:从 [1, 1, 1] 开始
[1, 1, 1], 和为 3 ,选择下标 1
[1, 3, 1], 和为 5, 选择下标 2
[1, 3, 5], 和为 9, 选择下标 0
[9, 3, 5] 完成
示例 2:
输入:target = [1,1,1,2]
输出:false
解释:不可能从 [1,1,1,1] 出发构造目标数组。
示例 3:
输入:target = [8,5]
输出:true
提示:
N == target.length
1 <= target.length <= 5 * 10^4
1 <= target[i] <= 10^9
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/construct-target-array-with-multiple-sums 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
bool isPossible(vector<int>& target) {
int i, num;
long sum = 0, s;
priority_queue<int> q;//默认大顶堆
for(i = 0; i < target.size(); ++i)
{
sum += target[i];//总和
q.push(target[i]);
}
while(q.top() != 1)
{
s = sum-q.top();//剩余的和
num = q.top()-s;//栈顶-s,应该为1或者比1大的数
if(num < 1)//小于1则false
return false;
q.pop();//弹出栈顶
q.push(num);//把剩余的数再push进去
sum -= s;//和减少了s
}
return true;
}
};
超时原因:queue
的数据类型int
溢出了,改为long
class Solution {
public:
bool isPossible(vector<int>& target) {
long sum = 0, s, i, num;
priority_queue<long> q;//默认大顶堆
for(i = 0; i < target.size(); ++i)
{
sum += target[i];//总和
q.push(target[i]);
}
while(!q.empty() && q.top() != 1)
{
s = sum-q.top();//剩余的和
num = q.top()-s;//栈顶-s,应该为1或者比1大的数
if(num < 1)//小于1则false
return false;
q.pop();//弹出栈顶
if(num != 1)//等于1就不用再放进去了,节省时间
q.push(num);
sum -= s;//和减少了s
}
return true;
}
};
leetcode该题的数据有点弱,上面解法在例子 [1000000000, 1]
时超时。
比如对于 [5, 9, 31]
而言,31 - 14 = 17
还是最大数,不如干脆 31 - 14 * 2 = 9
更改如下,增加倍数scale
class Solution {
public:
bool isPossible(vector<int>& target) {
if(target.size() == 1)
return target[0] == 1;
long sum = 0, s, i, num, tp, scale;
priority_queue<long> q;//默认大顶堆
for(i = 0; i < target.size(); ++i)
{
sum += target[i];//总和
q.push(target[i]);
}
while(q.top() != 1)
{
tp = q.top();
q.pop();
s = sum-tp;//剩余的和
scale = max(1,int((tp-q.top())/s));//至少1倍
num = tp-scale*s;//栈顶-n*s,应该为1或者比1大的数
if(num < 1)//小于1则false
return false;
q.push(num);
sum -= scale*s;//和减少了n*s
}
return true;
}
};