766. 托普利茨矩阵 AND 1004. 最大连续1的个数 III

这是我参与8月更文挑战的第31天，活动详情查看：8月更文挑战

766. 托普利茨矩阵

给你一个 m x n 的矩阵 matrix 。如果这个矩阵是托普利茨矩阵，返回 true ；否则，返回 false 。

如果矩阵上每一条由左上到右下的对角线上的元素都相同，那么这个矩阵是 托普利茨矩阵 。

输入：matrix = [[1,2,3,4],[5,1,2,3],[9,5,1,2]]
输出：true
解释：
在上述矩阵中, 其对角线为: 
"[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。 
各条对角线上的所有元素均相同, 因此答案是 True 。

解题思路

我们以第一行为对角线的起点，遍历对角线，再以第一列为起点，遍历对角线，二者结合起来就可以遍历所有的对角线

代码

class Solution {
    public boolean isToeplitzMatrix(int[][] matrix) {

        for(int i=0;i<matrix[0].length;i++)//以第一行元素为起点遍历对角线
        {

            int x=i+1,y=1;//起点坐标
            while (x<matrix[0].length&&y<matrix.length)
            {
                if(matrix[y][x]!=matrix[y-1][x-1])return false;
                x++;
                y++;
            }
        }
        for(int j=0;j<matrix.length;j++)//以第一列元素为起点遍历对角线
        {

            int x=1,y=j+1;
            while (x<matrix[0].length&&y<matrix.length)
            {
                if(matrix[y][x]!=matrix[y-1][x-1])return false;
                x++;
                y++;
            }
        }
        return true;

    }
}

1004. 最大连续1的个数 III

给定一个由若干 0 和 1 组成的数组 A，我们最多可以将 K 个值从 0 变成 1 。

返回仅包含 1 的最长（连续）子数组的长度。

示例 1：

输入：A = [1,1,1,0,0,0,1,1,1,1,0], K = 2 输出：6 解释： [1,1,1,0,0,1,1,1,1,1,1] 粗体数字从 0 翻转到 1，最长的子数组长度为 6。

解题思路

我们维护一个0个数小于k的滑动窗口，遍历所有元素，找出最大的窗口

代码

class Solution {
    public int longestOnes(int[] A, int K) {


        int l=0,r=0,n=A.length,cnt=0,res=0;
        while (r<n)
        {
            if(A[r]==0) cnt++;
            while (cnt>K)
            {
                if(A[l++]==0)
                    cnt--;
            }
            res= Math.max(res,r-l+1);
            r++;
        }
        return res;
    }
}