-
Notifications
You must be signed in to change notification settings - Fork 30
/
Copy pathFind Original Array From Doubled Array.cpp
47 lines (34 loc) · 1.34 KB
/
Find Original Array From Doubled Array.cpp
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
/*
Solution by Rahul Surana
***********************************************************
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.
***********************************************************
*/
#include <bits/stdc++.h>
class Solution {
public:
map<int,int> m;
vector<int> findOriginalArray(vector<int>& changed) {
if(changed.size()%2==1) return vector<int>();
vector<int> ans;
sort(changed.begin(),changed.end());
for(int i = 0 ; i < changed.size();i++){
if(changed[i]&1){
ans.push_back(changed[i]);
m[changed[i]*2] -=1;
}
else{
if(m[changed[i]] >= 0){
ans.push_back(changed[i]);
m[changed[i]*2] -=1;
}
else{
m[changed[i]] +=1;
}
}
}
if(ans.size() != changed.size()/2) return vector<int>();
return ans;
}
};