Lintcode 搜索二维矩阵 II

本文介绍了一种高效算法,用于在特定条件下搜索m×n矩阵中的值及其出现次数。该矩阵的每行每列都已排序且无重复元素。提供两种算法实现方案:二分查找法与改进线性搜索法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

写出一个高效的算法来搜索m×n矩阵中的值,返回这个值出现的次数。

这个矩阵具有以下特性:

  • 每行中的整数从左到右是排序的。
  • 每一列的整数从上到下是排序的。
  • 在每一行或每一列中没有重复的整数。

样例

考虑下列矩阵:

[

    [1, 3, 5, 7],

    [2, 4, 7, 8],

    [3, 5, 9, 10]

]

给出target = 3,返回 2



方法一    二分法快速查找:时间复杂度高(没有利用这个矩阵的特点)。

class Solution {
public:
    /**
     * @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
     */
    bool BSearch(vector<vector<int> > matrix,int target,int k,int n){
        int first,end,mid;
        first=0;
        end=n-1;
     //   cout<<first;
       
        while(first<end){
            mid=(first+end)/2;
            if(matrix[k][mid]==target)
            return true;
            else if(matrix[k][mid]<target){
                first=mid+1;
            }
            else
            end=mid-1;
        }
        if(matrix[k][end]==target)
        return true;
        else
        return false;
    }
    int searchMatrix(vector<vector<int> > &matrix, int target) {
        // write your code here
        int sum=0;
        if(matrix.empty())
        return 0;
        int m=matrix.size();
        int n=matrix[0].size();
        int i=0;

        
        while(i<m&&matrix[i][0]<=target){
            
            if(matrix[i][n-1]>=target){
          //  cout<<n<<endl;
          //  return m;
           if(BSearch(matrix,target,i,n)) 
           
           sum++;}
            i++;
        }
        return sum;
    }
};

方法二    只需要找到最小一行中第一个比target大的数然后先向左搜索,每次遇见小于target的则保持列不变,行+1,时间复杂度为O(m+n)。

class Solution {
public:
    /**
     * @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
     */
    int searchMatrix(vector<vector<int> > &matrix, int target) {
        // write your code here
        int sum=0;
        if(matrix.empty())
        return 0;
        int m=matrix.size();
        int n=matrix[0].size();
        int i=0,j=n-1; 
        for(;i<m;i++){
            if(matrix[i][0]<=target&&matrix[i][n-1]>=target)
            break;
        }
        for(;j>=0;j--){
            if(i>=m)
            break;
            if(matrix[i][j]<=target)
            break;
        }
      
        while(i<m&&j>=0){
            if(matrix[i][j]==target)
            {
                sum++;
                j--;
                i++;
                
            }
          else  if(matrix[i][j]>target)
            j--;
           else if(matrix[i][j]<target)
            i++;
            
        }
        return sum;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值