2007. 从双倍数组中还原原数组Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
一个整数数组 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
# 思路
穷举所有可能
# 解法
class Solution {
public int[] findOriginalArray(int[] changed) {
int n = changed.length;
if (n % 2 == 1) return new int[0];
int[] arr = new int[100001];
int[] res = new int[n / 2];
for (int x : changed) arr[x]++;
if (arr[0] % 2 == 1) return new int[0];
int index = arr[0] / 2;
for (int i = 1; i < arr.length; i++) {
int x = arr[i];
if (x != 0) {
if (i * 2 > 100001 || arr[i * 2] < x) return new int[0];
else {
arr[i * 2] -= x;
while (x != 0) {
res[index] = i;
index++;
x--;
}
}
}
}
return res;
}
}
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
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
# 总结
- 分析出几种情况,然后分别对各个情况实现