当前位置：首页 > article >正文

Rust 力扣 - 1493. 删掉一个元素以后全为 1 的最长子数组

article 2024/12/28 7:09:13

文章目录

题目描述
题解思路
题解代码
题目链接

题目描述

在这里插入图片描述

题解思路

本题我们能转换成求只包含一个0的子数组的最长长度

如果数组中不存在0，则结果为数组长度减一

我们使用一个left指针指向子数组的左边界，然后我们遍历子数组的右边界，我们记录子数组中0元素的位置idx0

当右边界遍历到0时，将左边界右移到idx0 + 1，并尝试刷新子数组的最长长度

当右边界遍历结束，尝试刷新子数组的最长长度

题解代码

impl Solution {
    pub fn longest_subarray(nums: Vec<i32>) -> i32 {
        let mut ans = 0;

        let mut idx0 = -1;

        for i in 0..nums.len() {
            if nums[i] == 0 {
                idx0 = i as i32;
                break;
            }
        }

        if idx0 == -1 {
            return nums.len() as i32 - 1;
        }

        let mut left = 0;

        for i in (idx0 + 1)..nums.len() as i32 {
            if nums[i as usize] == 0 {
                ans = ans.max(i - left - 1);
                left = idx0 + 1;
                idx0 = i;
            }
        }

        ans = ans.max(nums.len() as i32 - 1 - left);

        ans
    }
}