查找有序二维数组的中位数

题目:给定 n×n 的实数矩阵,每行和每列都是递增的,求这 n^2 个数的中位数

代码如下:

package com.cb.java.algorithms.programmingmethod.search;

/**
 * 给定 n×n 的实数矩阵,每行和每列都是递增的,求这 n^2 个数的中位数
 * 
 * @author 36184
 *
 */
public class SearchMedian {

	/**
	 * 对于n×n的数组,需要开辟一个新的同样大小的数组
	 * @param nums 要查找的数组
	 * @return
	 */
	public static int searchMedian(int[][] nums) {
		if (nums == null || nums.length <= 0) {
			return -1;
		}
		int[] temp = new int[nums.length * nums[0].length]; //开辟一个新的数组
		for (int[] num : nums) {
			for (int n : num) {
				temp[n - 1] ++;
			}
		}
		int[] temp1 = new int[temp.length];//开辟一个新的数组
		int length = temp.length / 2;
		int j = 0;
		for (int t=0;t<temp.length; t++) {
			if (temp[t] == 0) {
				continue;
			} else {
				for (int i = 1; i <=temp[t]; i++) {
					temp1[j++] =t+1;
				}
			}
		}
		if (temp1.length % 2 == 0) {
			return (temp1[length] + temp1[length - 1]) / 2;
		} else {
			return temp1[length];
		}
	}
	
	public static void main(String[] args) {
		int[][] arr = { { 1, 2, 8, 9 }, { 2, 4, 9, 12 }, { 4, 7, 10, 13 }, { 6, 8, 11, 15 } };
		int median=searchMedian(arr);
		System.out.println(median);
	}
}

猜你喜欢

转载自blog.csdn.net/u013230189/article/details/80890220