An integer array original
is transformed into a doubled array changed
by appending twice the value of every element in original
, and then randomly shuffling the resulting array.
Given an array changed
, return original
if changed
is a doubled array. If changed
is not a doubled array, return an empty array. The elements in original
may be returned in any order.
Example 1:
Input: changed = [1,3,4,2,6,8] Output: [1,3,4] Explanation: One possible original array could be [1,3,4]: - Twice the value of 1 is 1 * 2 = 2. - Twice the value of 3 is 3 * 2 = 6. - Twice the value of 4 is 4 * 2 = 8. Other original arrays could be [4,3,1] or [3,1,4].
Example 2:
Input: changed = [6,3,0,1] Output: [] Explanation: changed is not a doubled array.
Example 3:
Input: changed = [1] Output: [] Explanation: changed is not a doubled array.
Constraints:
1 <= changed.length <= 105
0 <= changed[i] <= 105
Related Topics:
Array, Hash Table, Greedy, Sorting
Similar Questions:
Sort the array A
. Keep removing the smallest element n
and 2 * n
from the array, and put n
into the answer until A
becomes empty. Anytime we can't do the removal, we return empty array.
Note: Don't use s.count(2 * n) == 0
to test if 2 * n
is in the multiset
since it's an O(N)
operation for multiset
.
// OJ: https://leetcode.com/problems/find-original-array-from-doubled-array/
// Author: github.com/lzl124631x
// Time: O(NlogN)
// Space: O(N)
class Solution {
public:
vector<int> findOriginalArray(vector<int>& A) {
if (A.size() % 2) return {};
multiset<int> s(begin(A), end(A));
vector<int> ans;
while (s.size()) {
int n = *s.begin();
s.erase(s.begin());
auto it = s.find(2 * n);
if (it == end(s)) return {};
s.erase(it);
ans.push_back(n);
}
return ans;
}
};
We can keep a frequency map in map<int, int> m
, and remove elements of the same value in batch.
Note: need some caution for 0
because 2 * 0 == 0
.
// OJ: https://leetcode.com/problems/find-original-array-from-doubled-array/
// Author: github.com/lzl124631x
// Time: O(NlogK) where `N` is the length of `A`, and `K` is the number of unique elements in `A`
// Space: O(N)
class Solution {
public:
vector<int> findOriginalArray(vector<int>& A) {
if (A.size() % 2) return {};
map<int, int> m;
for (int n : A) m[n]++;
vector<int> ans;
while (m.size()) {
auto [n, cnt] = *m.begin();
m.erase(n);
if (n == 0) {
if (cnt % 2) return {}; // count of `0` is odd.
for (; cnt; cnt -= 2) ans.push_back(n);
} else {
if (m[2 * n] < cnt) return {}; // not enough `2n` available.
if ((m[2 * n] -= cnt) == 0) m.erase(2 * n);
while (cnt--) ans.push_back(n);
}
}
return ans;
}
};
Or
// OJ: https://leetcode.com/problems/find-original-array-from-doubled-array/
// Author: github.com/lzl124631x
// Time: O(N + KlogK) where `N` is the length of `A`, and `K` is the number of unique elements in `A`
// Space: O(N)
class Solution {
public:
vector<int> findOriginalArray(vector<int>& A) {
if (A.size() % 2) return {};
unordered_map<int, int> m;
for (int n : A) m[n]++;
vector<int> nums, ans;
for (auto [n, cnt] : m) nums.push_back(n);
sort(begin(nums), end(nums));
for (int n : nums) {
if (m[2 * n] < m[n]) return {};
for (int i = 0; i < m[n]; ++i, --m[2 * n]) ans.push_back(n);
}
return ans;
}
};
// OJ: https://leetcode.com/problems/find-original-array-from-doubled-array/
// Author: github.com/lzl124631x
// Time: O(NlogN)
// Space: O(1) extra space
class Solution {
public:
vector<int> findOriginalArray(vector<int>& A) {
if (A.size() % 2) return {};
sort(begin(A), end(A));
int N = A.size(), j = 0; // `j` is the write pointer in `ans` array
vector<int> ans;
ans.reserve(N / 2);
for (int i = 0; i < N; ++i) {
if (j == ans.size()) ans.push_back(A[i]); // If `j` is at the end of `ans`, this `A[i]` must be in the `ans`.
else if (A[i] == ans[j] * 2) ++j; // If this `A[i] == ans[j] * 2`, we've found the doubled number correponding to `ans[j]`, so we move `j` forward.
else ans.push_back(A[i]); // Otherwise, this has to be in the `ans`.
if (ans.size() > N / 2) return {};
}
return ans;
}
};