【Leetcode】2765. 最长交替子数组

题目

2765. 最长交替子数组

题目：给你一个下标从 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] <= 10⁴

思路

本题的约束条件较低，因此可以检查每个子数组是否符合给定条件。
题目中要求的是连续子数组，先枚举子数组的长度然后遍历验证就可以了。

代码

class Solution {
public:
    int alternatingSubarray(vector<int>& nums) {
    int res = 0; 
    int n = nums.size();

    for (int i = 1; i < n; ) {
        if (nums[i] == nums[i - 1] + 1) {
            int t[2] {nums[i - 1], nums[i]};
            int r = i + 1;
            while (r < n && nums[r] == t[(r - i - 1) % 2]) r++;
            res = max(res, r - i + 1);
            i = r;
        } else {
            i++;
        }
    }
    return res == 0 ? -1 : res;
    }
};

结果