- 寻找两个正序数组的中位数
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
//1、把nums1,nums2合并到nums3;
vector<int> nums3;
nums3.insert(nums3.end(),nums1.begin(),nums1.end());
nums3.insert(nums3.end(),nums2.begin(),nums2.end());
//2、去重复的元素,也可以调用去vector函数实现
int i = 0,j = 0;
vector<int> s1;
bool s = 0;
for(i = 0;i < (int)nums3.size();++i)
{
s = 1;
for(j = 0;j < (int)s1.size();++j)
{
if(nums3[i] == s1[j])
s = 0;
}
if(s)
{
s1.push_back(nums3[i]);
}
}
//3、排序
sort(s1.begin(),s1.end());
for(vector<int>::iterator it=s1.begin();it!=s1.end();++it){
cout<<*it<<" ";
}
//4、计算中位数
int lens = s1.size();
if(lens%2 != 0)
{
return s1[lens/2];
}
else
{
int l = lens/2;
cout<<"("<<s1[l]<<"+"<<s1[l-1]<<")/"<<2<<endl;
double dd = ((double)s1[l]+(double)s1[l-1])/2;
return dd;
}
return 0.0;
}
int main(void){
vector<int> nums1,nums2;
int i = 0,s = 0;
cout<<"input 6组int数据:";
for(i = 0;i < 6;++i)
{
cin>>s;
nums1.push_back(s);
}
cout<<"input 再次6组int数据:";
for(i = 0;i < 6;++i)
{
cin>>s;
nums2.push_back(s);
}
cout<<"dd = "<<findMedianSortedArrays(nums1,nums2)<<endl;
return 0;
}
iotek@iotekclass:~/pwz1903C++/Depa/面试题/方凌计算机$ ./a.out
input 6组int数据:1 3 2 2 1 3
input 再次6组int数据:3 1 1 2 2 3
1 2 3 dd = 2
iotek@iotekclass:~/pwz1903C++/Depa/面试题/方凌计算机$ ./a.out
input 6组int数据:2 3 4 5 6 7
input 再次6组int数据: 8 9 10 11 12 0
0 2 3 4 5 6 7 8 9 10 11 12 (7+6)/2
dd = 6.5
iotek@iotekclass:~/pwz1903C++/Depa/面试题/方凌计算机$