From 1ad5b3270c9298f59a230fe91066ede0824ffd52 Mon Sep 17 00:00:00 2001 From: "98shim@gmail.com" <98shim@gmail.com> Date: Sat, 3 Oct 2020 12:03:27 +0530 Subject: [PATCH] Added a solution --- problems/src/array/median.cpp | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 problems/src/array/median.cpp diff --git a/problems/src/array/median.cpp b/problems/src/array/median.cpp new file mode 100644 index 00000000..bd6fb6f6 --- /dev/null +++ b/problems/src/array/median.cpp @@ -0,0 +1,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; + } +};