209. 长度最小的子数组

每天一道Rust-LeetCode(2020-02-12)

坚持每天一道题,刷题学习Rust.

题目描述

给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组。如果不存在符合条件的连续子数组,返回 0。

示例:

输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。
进阶:

如果你已经完成了O(n) 时间复杂度的解法, 请尝试 O(n log n) 时间复杂度的解法。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-size-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

首先求累加和 O(N)
然后再从头开始找到第一个大于等于s的位置
记下来第一个最小子数组.
假设start指向0,end指向大于s的位置
最终start从头快走到尾,end是从头走到尾,所以复杂度是O(N)

解题过程

struct Solution {}
impl Solution {
    pub fn min_sub_array_len(s: i32, nums: Vec<i32>) -> i32 {
        let mut nums = nums;
        if nums.len() == 0 {
            return 0;
        }
        for i in 1..nums.len() {
            nums[i] = nums[i - 1] + nums[i];
        }
        if nums[nums.len() - 1] < s {
            return 0;
        }
        if nums[0] >= s {
            return 1;
        }
        let mut start: i32 = -1;
        let mut start_value = 0;
        let mut end = 0;
        let mut min_len = std::i32::MAX;
        while end < nums.len() {
            if nums[end] - start_value < s {
                end += 1;
                continue;
            }
            let l = end as i32 - start;
            if l < min_len {
                min_len = l;
            }
            start += 1;
            start_value = nums[start as usize];
        }
        min_len as i32
    }
}
#[cfg(test)]
mod test {
    use super::*;
    #[test]
    fn test() {
        assert_eq!(Solution::min_sub_array_len(7, vec![2, 3, 1, 2, 4, 3]), 2);
        assert_eq!(Solution::min_sub_array_len(15, vec![1, 2, 3, 4, 5]), 5);
    }
}

一点感悟

其他

欢迎关注我的github,本项目文章所有代码都可以找到.