-
Notifications
You must be signed in to change notification settings - Fork 0
/
1354. Construct Target Array With Multiple Sums
62 lines (45 loc) · 1.34 KB
/
1354. Construct Target Array With Multiple Sums
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
You are given an array target of n integers. From a starting array arr consisting of n 1's, you may perform the following procedure :
let x be the sum of all elements currently in your array.
choose index i, such that 0 <= i < n and set the value of arr at index i to x.
You may repeat this procedure as many times as needed.
Return true if it is possible to construct the target array from arr, otherwise, return false.
Example 1:
Input: target = [9,3,5]
Output: true
Explanation: Start with arr = [1, 1, 1]
[1, 1, 1], sum = 3 choose index 1
[1, 3, 1], sum = 5 choose index 2
[1, 3, 5], sum = 9 choose index 0
[9, 3, 5] Done
Example 2:
Input: target = [1,1,1,2]
Output: false
Explanation: Impossible to create target array from [1,1,1,1].
Example 3:
Input: target = [8,5]
Output: true
Constraints:
n == target.length
1 <= n <= 5 * 104
1 <= target[i] <= 109
#solution
class Solution {
public boolean isPossible(int[] target) {
if (target.length == 1) return target[0] == 1;
PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
int sum = 0;
for (int t: target) {
pq.add(t);
sum += t;
}
while (pq.peek() != 1) {
int curr = pq.poll();
if (sum - curr == 1) return true;
int x = curr % (sum - curr);
sum = sum - curr + x;
if (x == 0 || x == curr) return false;
else pq.add(x);
}
return true;
}
}