leetcode 2 MEDIAN OF TWO SORTED ARRAYS

There are two sorted arrays A and B of size mand n respectively. Find the median of the two sorted arrays. The overall runtime complexity should be O(log (m+n)). 

class Solution {
public:
    double findMedianSortedArrays(int A[], int m, int B[], int n) {
		int i=0;
		int j=0;
		double ans;
		int * temp=new int[m+n];

		if(m==0)
		{
			if(n%2==0) return (B[n>>1]+B[(n>>1)-1])/2.0; 
			else return B[n>>1];
		}
		if(n == 0)
		{
			if(m%2==0) return (A[m>>1]+A[(m>>1)-1])/2.0; 
			else return A[m>>1];
		}
		while((i+j)<=((m+n)>>1))
		{
			if(i<m&&j<n)
			{
				if(A[i]<B[j])
				{
					temp[i+j]=A[i];
					i++;
				}
				else
				{
					temp[i+j]=B[j];
					j++;
				}
			}
			else if(i==m&&j<n)
			{
				temp[i+j]=B[j];
				j++;
			}
			else if(i<m&&j==n)
			{
				temp[i+j]=A[i];
				i++;
			}
		}
		if((m+n)%2==0) ans=(temp[(m+n)>>1]+temp[((m+n)>>1)-1])/2.0; 
		else ans=temp[(m+n)>>1];
		delete [] temp;
		return ans;
    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值