2024.1.23每日一题

LeetCode

最长交替子数组

2765. 最长交替子数组 - 力扣（LeetCode）

题目描述

给你一个下标从 0 开始的整数数组 nums 。如果 nums 中长度为 m 的子数组 s 满足以下条件，我们称它是一个 交替子数组 ：

• m 大于 1 。
• s1 = s0 + 1 。
• 下标从 0 开始的子数组 s 与数组 [s0, s1, s0, s1,...,s(m-1) % 2] 一样。也就是说，s1 - s0 = 1 ，s2 - s1 = -1 ，s3 - s2 = 1 ，s4 - s3 = -1 ，以此类推，直到 s[m - 1] - s[m - 2] = (-1)m 。

请你返回 nums 中所有 交替 子数组中，最长的长度，如果不存在交替子数组，请你返回 -1 。

子数组是一个数组中一段连续 非空 的元素序列。

示例 1：

输入：nums = [2,3,4,3,4]
输出：4
解释：交替子数组有 [3,4] ，[3,4,3] 和 [3,4,3,4] 。最长的子数组为 [3,4,3,4] ，长度为4 。

示例 2：

输入：nums = [4,5,6]
输出：2
解释：[4,5] 和 [5,6] 是仅有的两个交替子数组。它们长度都为 2 。

提示：

• 2 <= nums.length <= 100
• 1 <= nums[i] <= 104

思路

分组循环
灵神题解

2765. 最长交替子数组 - 力扣（LeetCode）

分组循环模板如下

n = len(nums)
i = 0
while i < n:
    start = i
    while i < n and ...:
        i += 1
    # 从 start 到 i-1 是一组
    # 下一组从 i 开始，无需 i += 1

代码

C++

class Solution {
public:
    int alternatingSubarray(vector<int> &nums) {
        int ans = -1;
        int i = 0;
        int n = nums.size();
        while(i < n- 1){
            if(nums[i+1] - nums[i] != 1){
                i++;
                continue;
            }

            int start = i;
            i += 2;
            while(i < n && nums[i] == nums[start] + (i - start) % 2){
                i++;
            }
            ans = max(ans,i - start);
            i--;
        }
        return ans;
    }
};

Java

class Solution {
    public int alternatingSubarray(int[] nums) {
        int n = nums.length;
        int i = 0;
        int ans = -1;
        while(i < n - 1){
            if(nums[i+1] - nums[i] != 1){
                i++;
                continue;
            }

            int start = i; // 记录这一组的开始位置
            i += 2; // i 和 i + 1已经满足要求，从i + 2判断
            while(i < n && nums[i] == nums[start] + (i - start) % 2){
                i++;
            }
            ans = Math.max(ans,i - start);
            i--;
        }
        return ans;
    }
}