-
Notifications
You must be signed in to change notification settings - Fork 826
/
Copy pathmedian.cpp
32 lines (32 loc) · 815 Bytes
/
median.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
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
map<int,int>mp;
for(int i=0;i<nums1.size();i++){
mp[nums1[i]]++;
}
for(int i=0;i<nums2.size();i++){
mp[nums2[i]]++;
}
vector<int>ans;
for(auto i:mp){
for(int j=0;j<i.second;j++){
ans.push_back(i.first);
}
}
// for(int i=0;i<ans.size();i++){
// cout<<ans[i]<<' ';
// }
// cout<<'\n';
double a;
int n=ans.size();
if(n%2==1){
a=(double)ans[n/2];
}
else{
cout<<ans[n/2]<<' '<<ans[n/2-1]<<'\n';
a=((double)ans[n/2]+(double)ans[(n/2)-1])/2.0;
}
return a;
}
};