Median of Two Sorted Arrays - LeetCode

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

我写的太不简洁了,也没有优化,不过AC了:

更好的算法请戳这里: http://www.cnblogs.com/longdouhzt/archive/2013/03/04/2943572.html 

还有这里: http://www.programcreek.com/2012/12/leetcode-median-of-two-sorted-arrays-java/

public class Solution {
	public double getMedian(int[] numArray) {
		int index;
		double median;
		if (numArray.length % 2 == 0)
		    median = ((double)numArray[numArray.length/2] + (double)numArray[numArray.length/2 - 1])/2;
		else
		    median = (double) numArray[numArray.length/2];
		return median;
	}
public double findMedianSortedArrays(int A[], int B[]) {
        int m = A.length;
        int n = B.length;
        int n1 = 0;
        int m1 = 0;
        int flag = 0;
        int pre = 0;
        if(m == 0 && n == 0)
        	return 0;
        else if(m == 0 && n != 0){
        	return getMedian(B);
        }
        else if(m != 0 && n == 0){
        	return getMedian(A);
        }
        else if((m + n) % 2 == 1){
		for(int i = 0; i <= (m + n)/2; i++){
			if(n1 >= n){
				m1 ++;
				flag = 1;
			}
			else if(m1 >= m){
				n1 ++;
				flag = 2;
			}
			else if(A[m1] >= B[n1]){
				n1++;
				flag = 2;
			}
			else
			{
				m1++;
				flag = 1;
			}
		}
		if(flag == 1){
			m1 = m1 - 1;
			return A[m1];
		}
		else
		{
			n1 = n1 - 1;
			return B[n1];
		}
	}
	
	else{
		for(int i = 0; i < (m + n)/2; i++){
			if(n1 >= n){
			//	pre = A[m1];
				m1 ++;
				flag = 1;
			}
			else if(m1 >= m){
				//pre = B[n1];
				n1 ++;
				flag = 2;
			}
			else if(A[m1] > B[n1]){
			//	pre = B[n1];
				n1++;
				flag = 2;
			}
			else
			{
				//pre = A[m1];
				m1++;
				flag = 1;
			}
		}
		
		if(flag == 1){
			
			pre = A[m1 - 1];
		}
		else{
			pre = B[n1 - 1];
		}
		
		if(n1 >= n){
			//	pre = A[m1];
				m1 ++;
				flag = 1;
			}
			else if(m1 >= m){
				//pre = B[n1];
				n1 ++;
				flag = 2;
			}
			else if(A[m1] > B[n1]){
			//	pre = B[n1];
				n1++;
				flag = 2;
			}
			else
			{
				//pre = A[m1];
				m1++;
				flag = 1;
			}
		
		if(flag == 1){
			m1 = m1 - 1;
			return ((double)A[m1] + (double)pre)/2;
		}
		else
		{
			n1 = n1 -1;
			return ((double)B[n1] + (double)pre)/2;
		}
	}
    }

}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值