首頁>技術>

簡介: 演算法面試真題詳解:搜尋二維

矩陣 II描述寫出一個高效的演算法來搜尋m×n矩陣中的值,返回這個值出現的次數。這個矩陣具有以下特性:每行中的整數從左到右是排序的。每一列的整數從上到下是排序的。在每一行或每一列中沒有重複的整數。

線上評測地址:領釦題庫官網

樣例 1:輸入:[[3,4]]target=3輸出:1
樣例 2:輸入:    [      [1, 3, 5, 7],      [2, 4, 7, 8],      [3, 5, 9, 10]    ]    target = 3輸出:2

挑戰:要求O(m+n) 時間複雜度和O(1) 額外空間

演算法:搜尋/模擬

根據題意,每行中的整數從左到右是排序的,每一列的整數從上到下是排序的,在每一行或每一列中沒有重複的整數。那麼我們只要從矩陣的左下角開始向右上角找,若是小於target就往右找,若是大於target就往上找

從左下角即(n-1,0)處出發如果matrixx < target 下一步往右搜如果matrixx > target 下一步往上搜如果matrixx = target 下一步往x-1即右上角搜,因為是有序的,每一行每一列中每個數都是唯一的

複雜度分析

時間複雜度O(n+m)左下角往右上角呈階梯狀走,長度為n+m空間複雜度O(size(matrix))地圖的大小

public class Solution {    /**     * @param matrix: A list of lists of integers     * @param target: An integer you want to search in matrix     * @return: An integer indicate the total occurrence of target in the given matrix     */    public int searchMatrix(int[][] matrix, int target) {        int ans = 0;        int n = matrix.length,m = matrix[0].length;        if (n == 0 || m==0) {            return 0;        }        //從左下角往右上角走,(x,y)=(n-1,0)        int x = n - 1,y = 0;        while (x >= 0 && y < m) {            //如果matrix[x][y]等於target,則找到一個相等的值,由於嚴格排序,直接跳到右上角繼續搜            if (matrix[x][y] == target) {                ans++;                x--;                y++;            }            //如果matrix[x][y]比target大,則往上走            else if (matrix[x][y] > target) {                x--;            }            //如果matrix[x][y]比target小,則往右走            else {                y++;            }        }        return ans;    }}

原文:https://developer.aliyun.com/article/780454?spm=5176.8068049.0.0.39926d19H3TZz6&groupCode=iot

7
最新評論
  • BSA-TRITC(10mg/ml) TRITC-BSA 牛血清白蛋白改性標記羅丹明
  • 專案上線突發的一個超級大坑,MySQL導致專案無法正常訪問