Sorting_Searching 有序矩阵中查找数 @CareerCup

最新推荐文章于 2014-03-04 10:27:07 发布

原创最新推荐文章于 2014-03-04 10:27:07 发布 · 1.6k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

CareerCup

当前文章被收录于：

CareerCup

72 篇文章 0 人学习

订阅专栏查看详情

CareerCup专栏

当前文章被收录于：

CareerCup专栏

41 篇文章 16 人学习

和网友分享Crack the Coding Interview一书中的解法

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一种高效的矩阵搜索算法，该算法适用于每一行和每一列都已排序的矩阵，通过从右上角或左下角开始查找，每次比较都能排除一行或一列，从而快速定位目标元素。

要点就是从右上角或者左下角开始找，每次比较能排除一行或者一列

package Sorting_Searching;

import CtCILibrary.AssortedMethods;

/**
 * 
 *Given a matrix in which each row and each column is sorted, write a method to find an element in it.

译文：

给出一个矩阵，其中每一行和每一列都是有序的，写一个函数在矩阵中找出指定的数。
 */
public class S11_6 {

	public static void main(String[] args) {
		int M = 10;
		int N = 5;
		int[][] matrix = new int[M][N];
		for (int i = 0; i < M; i++) {
			for (int j = 0; j < N; j++) {
				matrix[i][j] = 10 * i + j;
			}
		}

		AssortedMethods.printMatrix(matrix);

		for (int i = 0; i < M; i++) {
			for (int j = 0; j < M; j++) {
				int v = 10 * i + j;
				System.out.println(v + ": " + findElement(matrix, v));
			}
		}

	}
	
	public static boolean findElement(int[][] matrix, int elem) {
		int cols = matrix[0].length;
		int rows = matrix.length;
		int x = 0;				// the first row
		int y = cols-1;			// the last col
		
		while(true){
			if(x<0 || x>=rows || y<0 || y>=cols){
				return false;
			}
			if(matrix[x][y] == elem){
				return true;
			}
			if(elem > matrix[x][y]){		// 大的话，往下走
				x++;
			}
			else if(elem < matrix[x][y]){		// 小的话，往左走
				y--;
			}
		}
	}

}