码迷,mamicode.com
首页 > 编程语言 > 详细

剑指offer - 二维数组中的查找

时间:2019-12-13 14:17:36      阅读:87      评论:0      收藏:0      [点我收藏+]

标签:turn   offer   bre   就是   col   本地   class   solution   for   

  这题我总觉得 oj 有问题。。。就算真的复杂度很高,也不可能通不过一个样例吧。。。

  第一眼没注意到每一列都从上到下递增,写出来的算法平均时间复杂度为 O(lgn * lgm),最坏为O(nlgm),思路就是对第一列做二分查找得到一个行区间,然后遍历行区间,对每一行做二分查找,但提交后有段错误(本地 MSVC 编译、运行没问题)。。。

  然后又认真看一遍题之后,经过思考各种边界情况,优化了很多小地方的查找步数,时间复杂度肯定比第一个小很多,所以有点不敢相信居然超时?

class Solution {
public:
    bool Find(int target, std::vector<std::vector<int>> array) {
        if (array.empty()) return false;
        if (array[0][0] > target) return false;
        if (array[array.size() - 1][array[0].size() - 1] < target) return false;
        int start_col = 0,
            end_col = array.size() - 1,
            mid_col = (start_col + end_col) / 2;
        while (array[start_col][0] < target
            && target < array[end_col][0])
        {
            if (array[start_col][0] < target
                && target < array[mid_col][0])
                start_col = (start_col + mid_col) / 2,
                end_col = mid_col,
                mid_col = (start_col + end_col) / 2;
            else if (array[mid_col][0] < target
                && target < array[end_col][0])
                end_col = (mid_col + end_col) / 2,
                start_col = mid_col,
                mid_col = (start_col + end_col) / 2;
            else break;
        }

        for (int i = start_col; i <= end_col; i++)
        {
            int start_row = 0,
                end_row = array[0].size() - 1,
                mid_row = (start_row + end_row) / 2;
            if (array[i][start_row] == target) return true;
            if (array[i][end_row] == target) return true;
            while (start_row < end_row)
            {
                if (target == array[i][mid_row]) return true;
                else if (array[i][start_row] <= target
                    && target <= array[i][mid_row])
                    start_row = (start_row + mid_row) / 2,
                    end_row = mid_row,
                    mid_row = (start_row + end_row) / 2;
                else if (array[i][mid_row] <= target
                    && target <= array[i][end_row])
                    end_row = (mid_row + end_row) / 2,
                    start_row = mid_row,
                    mid_row = (start_row + end_row) / 2;
                else break;
            }
        }
        return false;
    } 

    bool Find2(int target, std::vector<std::vector<int>> array)
    {
        int row = 0, col = 0;
        int row_siz = array.size(), col_siz = array[0].size();
        while (row < row_siz && col < col_siz)
        {
            if (target == array[row][col]) return true;
            else if (col + 1 < col_siz && row + 1 < row_siz
                && target < array[row + 1][col + 1]
                && target > array[row + 1][col]
                && target > array[row][col + 1]) return false;
            else if (target < array[row][col]) {
                row--;
                if (row <= 0) return false;
            }
            
            if (col + 1 < col_siz
                && row + 1 < row_siz
                && target >= array[row + 1][col + 1]) col++, row++;
            else if (col + 1 < col_siz
                && target >= array[row][col + 1]) col++;
            else if (row + 1 < row_siz
                && target >= array[row + 1][col]) row++;
            else if (col + 1 < col_siz
                && target <= array[row][col + 1]) row--;
            else break;
        }
        return false;
    }
};

  

剑指offer - 二维数组中的查找

标签:turn   offer   bre   就是   col   本地   class   solution   for   

原文地址:https://www.cnblogs.com/darkchii/p/12034516.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!