Go快速查找有序二维数组的数字
作者 | 陌无崖
转载请联系授权
导语
大家肯定对数组都不陌生，今天的这道题就是关于数组的，在做这道题之前呢，先带领大家回顾一下数组的要点。
数组
数组是一块连续的内存并按照顺序存储数据，使用数组必须分配内存，因此数组的空间效率差，经常会出现空闲的区域没有得到充分利用。数组的的内存连续，根据下标在O(1)时间读/写任何元素，时间效率高。
题目描述
在一个二维数组中，每一行都按照从左到右的递增顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数输入这样的一个二维数组和整数，判断该整数是否在该二维数组中。如：
1 2 8 9
2 4 9 12
4 7 10 13
6 8 11 15解决思路
对于这样的题，我们应该尽量利用该类数组的性质，根据数组已经排好的序列，很明显我们应该在比较的过程中，不断缩小比较的区域，通常的我们的思路是从1开始比较，我们可以尝试一下，比如我们找7这个数字，1比7小，因此，7应该在1的下面或右边，这个时候出现了问题，们是因该比较右边的还是下面呢？因此这个思路不行，那如果我们从顶点9开始呢，发现9比7大，那7一定在9的右边，因此我们可以比较8，发现仍然比较大，因此我们比较2，发现比7小，那么一定在下面，因此开始比较4，然后比较7，最终找到。很明显这个思路比较容易。按照这个思路大家可以在自己的草稿纸上进行画图尝试一下。找一下规律，暂时先不要看下面的部分。
规律
通过以上的思路我们可以发现如下规律
a[i][j] > num 时,j--
a[i][j] < num 时,i++代码
找到了以上规律我们就可以轻松的完成我们的代码设计了
type S1 []int
type S2 []S1
func FindNum(data S2, row, col, num int) bool {
if len(data) == 0 || row <= 0 || col <= 0 {
return false
}
//
i := 0
j := col - 1
for {
if i < 0 || j < 0 {
break
}
if data[i][j] > num {
j--
} else if data[i][j] < num {
i++
} else {
return true
}
}
return false
}这里需要注意的是为了满足传入各种数组的要求，在定义二维数组时使用了下面的方式
type S1 []int
type S2 []S1单元测试案例
为了保证我们的代码时成功的你也可以使用以下测试案例，或者自写案例
demo1是我传入的自定义的包
func TestFindNum(t *testing.T) {
// 定义一个完整数组
// var data [][4]int
// data = make([][4]int, 4)
// fmt.Println(data)[[0 0 0 0] [0 0 0 0] [0 0 0 0] [0 0 0 0]]
// 初始化二维数组
fmt.Println("测试用例1******")
a := demo1.S2{
demo1.S1{1, 2, 8, 9},
demo1.S1{2, 4, 9, 12},
demo1.S1{4, 7, 10, 13},
demo1.S1{6, 8, 11, 15},
}
num1, num2 := 8, 3
if demo1.FindNum(a, 4, 4, num1) == true && demo1.FindNum(a, 4, 4, num2) == false {
fmt.Println("success")
} else {
fmt.Println("fail")
}
fmt.Println("测试用例2********")
a = demo1.S2{}
if demo1.FindNum(a, 4, 4, num1) == false {
fmt.Println("success")
} else {
fmt.Println("fail")
}
}本文参与 腾讯云自媒体同步曝光计划，分享自微信公众号。
原始发表：2019-12-18，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
