数组01--二维数组中的查找

原创

昕光xg 2022-08-29 09:56:40 博主文章分类：数据结构与算法 ©著作权

文章标签 数组查找剑指offer 二维数组中的查找二维数组数组 文章分类 运维

©著作权归作者所有：来自51CTO博客作者昕光xg的原创作品，请联系作者获取转载授权，否则将追究法律责任

数组01--二维数组中的查找-jz01

题目概述
解析&参考答案
注意事项
说明

题目概述

算法说明
在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。
请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
测试用例
输入：
7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
输出：
true

解析&参考答案

解析
从数组的右上角的元素开始比较，若和目标值相同则结束，若小于目标值则列向左移动一步，若大于目标值则行向上移动一步；
依次循查找，直到程序结束。
参考答案

vim jz01.go
package main

import (
    "fmt"
    "time"
)

func Find(target int, array [][]int) bool {
    row := len(array)
    col := len(array[0])
    result := false
    if row == 0 || col == 0 {
        return false
    }
    for i := 0; i < row; i++ {
        for j := 0; j < col; j++ {
            if array[i][j] == target {
                result = true
            }
        }
    }
    return result
}

func FindV2(target int, array [][]int) bool {
    result := false
    row := len(array)
    col := len(array[0])
    if row == 0 || col == 0 {
        return false
    }
    r, c := 0, col-1
    for r < row && c >= 0 {
        if array[r][c] == target {
            result = true
            break
        } else if array[r][c] > target {
            c--
        } else {
            r++
        }
    }
    return result
}

func main() {
    target := 7
    start := time.Now().UnixNano()
    array := [][]int{{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}}
    // ret := Find(target, array)
    ret := FindV2(target, array)
    fmt.Println(ret)
    fmt.Printf("take: %v ns", time.Now().UnixNano()-start)
}

vim jz01_test.go
package main

import "testing"

var tests = []struct {
    target int
    array  [][]int
    result bool
}{
    {7, [][]int{{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}}, true},
    {5, [][]int{{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}}, false},
    {1, [][]int{{}}, false},
}

func TestFind(t *testing.T) {
    for _, tt := range tests {
        if actual := Find(tt.target, tt.array); actual != tt.result {
            t.Errorf("Find %d in %v, got %v; expected %v", tt.target, tt.array, Find(tt.target, tt.array),
                tt.result)
        }
    }
}

func TestFindV2(t *testing.T) {
    for _, tt := range tests {
        if actual := FindV2(tt.target, tt.array); actual != tt.result {
            t.Errorf("Find %d in %v, got %v; expected %v", tt.target, tt.array, FindV2(tt.target, tt.array),
                tt.result)
        }
    }
}