【Lintcode】898. Leftmost One

最新推荐文章于 2024-09-12 19:02:20 发布

记录算法题解

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量176

点赞数

分类专栏： LC 二分、位运算与数学文章标签：算法 leetcode

本文链接：https://blog.csdn.net/qq_46105170/article/details/117859395

版权

LC 二分、位运算与数学专栏收录该内容

401 篇文章 2 订阅

订阅专栏

该博客讨论了一种算法问题，给定一个二维矩阵，其中每一行都包含至少一个1，并且0总是位于1的左边。任务是找到最左边的1的纵坐标。解决方案是使用二分查找，遍历每一行以找到每个行中最左边的1的位置，最后返回最小的纵坐标。算法的时间复杂度为O(m log n)，空间复杂度为O(1)。

摘要由CSDN通过智能技术生成

题目地址：

https://www.lintcode.com/problem/898/

给定一个二维 $0 - 1$ 矩阵，题目保证每行都含 $1$ ，并且 $0$ 都排在 $1$ 的左边。求出现在最左边的 $1$ 的纵下标。

可以对每行用二分。代码如下：

public class Solution {
    /**
     * @param A: The 2-dimension array
     * @return: Return the column the leftmost one is located
     */
    public int getColumn(int[][] A) {
        // Write your code here
        int res = A[0].length;
        for (int[] row : A) {
            res = Math.min(res, find(row));
        }
        
        return res;
    }
    
    private int find(int[] A) {
        int l = 0, r = A.length - 1;
        while (l < r) {
            int m = l + (r - l >> 1);
            if (A[m] == 1) {
                r = m;
            } else {
                l = m + 1;
            }
        }
        
        return l;
    }
}