面试题21: 二维数组搜索

题目: 设计一个算法对 m*n 矩阵进行搜索,这个矩阵拥有如下属性.

    1)每行的数都是从左到右排序好的.

    2)每行的首数大于上行的尾数.

其实这题就是用二分查找算法,二分查找一般是在一维数组上进行搜索,解决这道题的关键就是如何将二维数组映射成一维数组.

映射关系为: k=i*n+j,j<n,k为在一维数组的中的下标值,i,j分别为二维数组中的行和列.根据前面的可以推导出i=k/n,j=k%n.

代码实现:

package main

import "fmt"

const M  = 3
const N  = 3
func searchMatrix(matrix interface{}, target int) bool {
	a, m, n := matrix.([M][N]int), M, N
	low, high, mid := 0, m * n -1, 0

	for low < high {
		mid = (low + high) / 2
		if a[mid / n][mid % n] == target {
			return true
		} else if a[mid / n][mid % n] < target {
			low = mid + 1
		} else {
			high = mid - 1
		}
	}
	return false
}

func main() {
	r := searchMatrix([M][N]int{{1,2,3},{4,5,6},{7,8,9}}, 3)
	fmt.Println(r)
}