一个整数数组 original
可以转变成一个 双倍 数组 changed
,转变方式为将 original
中每个元素 值乘以 2 加入数组中,然后将所有元素 随机打乱 。
给你一个数组 changed
,如果 change
是 双倍 数组,那么请你返回 original
数组,否则请返回空数组。original
的元素可以以 任意 顺序返回。
示例 1:
输入:changed = [1,3,4,2,6,8] 输出:[1,3,4] 解释:一个可能的 original 数组为 [1,3,4] : - 将 1 乘以 2 ,得到 1 * 2 = 2 。 - 将 3 乘以 2 ,得到 3 * 2 = 6 。 - 将 4 乘以 2 ,得到 4 * 2 = 8 。 其他可能的原数组方案为 [4,3,1] 或者 [3,1,4] 。
示例 2:
输入:changed = [6,3,0,1] 输出:[] 解释:changed 不是一个双倍数组。
示例 3:
输入:changed = [1] 输出:[] 解释:changed 不是一个双倍数组。
提示:
1 <= changed.length <= 105
0 <= changed[i] <= 105
C++
class Solution {
public:
vector<int> findOriginalArray(vector<int>& changed) {
int n=changed.size();
unordered_map<int,int> mp;
for(int i=0;i<n;i++) {
mp[changed[i]]++;
}
sort(changed.begin(),changed.end());
vector<int> res;
for(int i=0;i<n;i++) {
if(mp[changed[i]]>0) {
mp[changed[i]]--;
if(mp[changed[i]*2]>0) {
res.push_back(changed[i]);
mp[changed[i]*2]--;
} else {
return vector<int>();
}
}
}
return res;
}
};
java
class Solution {
public int[] findOriginalArray(int[] changed) {
ArrayList<Integer> res = new ArrayList<>();
Map<Integer, Integer> mp = new HashMap<>();
int n = changed.length;
Arrays.sort(changed);
for (int i = 0; i < n; i++) {
mp.put(changed[i], mp.getOrDefault(changed[i], 0) + 1);
}
for (int i = 0; i < n; i++) {
if (mp.get(changed[i]) > 0) {
mp.put(changed[i], mp.get(changed[i]) - 1);
if (mp.containsKey(changed[i] * 2) && mp.get(changed[i] * 2) > 0) {
mp.put(changed[i] * 2, mp.get(changed[i] * 2) - 1);
res.add(changed[i]);
} else {
return new int[0];
}
}
}
int[] ans = new int[n / 2];
for (int i = 0; i < ans.length; i++) {
ans[i] = res.get(i);
}
return ans;
}
}
python
class Solution:
def findOriginalArray(self, changed: List[int]) -> List[int]:
n = len(changed)
ct = Counter()
changed.sort()
res = []
for i in range(n):
ct[changed[i]] += 1
for i in range(n):
if ct[changed[i]] > 0:
ct[changed[i]] -= 1
if ct[changed[i] * 2] > 0:
ct[changed[i] * 2] -= 1
res.append(changed[i])
else:
return []
return res