这是做的第二道LeetCode题目,立个flag,远比第一题难度大了很多,后来才发现是difficult难度的题目,确实有很多需要记录总结的地方。
此题是算法基础那本书中的题目,隐隐记得有个很难想的解决方法,使用了中间的数不断比较的方法解决问题;本着自己可能不会写好的原则用了一个自己觉得好想的方法:
通过顺序比较两个数组(m+n)/2次,得到中位数,复杂度O(n)。特别强调此方法虽然好想,但是实现起来有非常多的特殊情况要考虑,再加上奇偶的考虑会让情况非常多,通过多次提交,对多种情况的值进行测试才得到完善。如果哪位同学有更好的实现方法欢迎讨论,特别是针对这种方法的特殊情况的考虑过程~
代码展示如下:
#include
#include
#include
using namespace std;
class Solution {
public:
double findMedianSortedArrays(vector
& nums1, vector
& nums2) { double result; int key1 = 0;//key1 for count nums1 int key2 = 0;//key2 for count nums2 //测试样例中有空的向量 if (nums1.empty() || nums2.empty()) { if (nums1.empty()) { int n = nums2.size(); result = (nums2[(n+1) / 2-1] + nums2[(n + 2) / 2-1]) / 2.0; } else { int m = nums1.size(); result = (nums1[(m +1) / 2-1] + nums1[(m + 2) / 2-1]) / 2.0; } } else { int m = nums1.size(); int n = nums2.size(); result = nums1[0]; //中位数是第(m+n+1)/2和(m+n+2)/2的均值,不论总和是奇数偶数 //这里比较了(m+n+1)/2次 for (int i = 0; i < ((m + n + 1) / 2); i++) { //key1==m 或者 key2==n 表示向量中的最后一个数值被比较且较小,则下一个数应该在对应的向量中。 if (key1 == m) { result = nums2[key2]; key2++; } else if (key2 == n) { result = nums1[key1]; key1++; } else if (nums1[key1] > nums2[key2]) { result = nums2[key2]; key2++; } else { result = nums1[key1]; key1++; } } //如果总和是奇数则可以直接返回result,比奇偶同时考虑要简单一些 if (((m + n) % 2) != 0) { return result; } //如果总和是偶数,需要找到下一个数值; //key1==m 或者 key2==n 表示向量中的最后一个数值被比较且较小,则下一个数应该在对应的向量中。 else if (key1 == m) { result = (result + nums2[key2]) / 2.0; } else if (key2 == n) { result = (result + nums1[key1]) / 2.0; } else if (nums1[key1] > nums2[key2]) { result = (result + nums2[key2])/2.0; } else { result = (result + nums1[key1]) / 2.0; } } return result; } }; void main(void) { Solution sol; int a[] = {4}; int b[] = { 1,2,3}; int len1 = sizeof(a) / sizeof(a[0]); int len2 = sizeof(b) / sizeof(b[0]); //vector
nums1; //vector
nums2; vector
nums1(&a[0], &a[len1]); vector
nums2(&b[0], &b[len2]); double result; result = sol.findMedianSortedArrays(nums1, nums2); cout << result << endl; system("pause"); }
需要考虑的主要问题有:
1、向量可能赋值为空,需要最先判断。
2、计数key可能到达向量的最后一个值,例如key1==m(key2==n),这是需要明确下一个值一定再另外一个向量中,或者让key2和key1-1比较后赋值result。
3、总和个数的奇偶可导致不同的得到result的策略
具体考虑的过程可参考代码注释
后续更新算法基础书中提供的log(n)的复杂度的解决方法