LeetCode 2765.最长交替子数组

吃着火锅x唱着歌2025-09-26 18:04

给你一个下标从 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

解释： 交替子数组有 [2,3]，[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^4

分组循环，交替元素的为一组，找最长的：

cpp 复制代码

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

        int ans = -1;

        int i = 0;
        while (i < n) {
            int start = i;
            ++i;

            if (i == n) {
                break;
            }

            if (nums[i] != nums[start] + 1) {
                continue;
            }

            while (i < n && 
                (((i - start) & 1) && nums[i] == nums[start] + 1 || 
                 ((i - start) & 1) == 0 && nums[i] == nums[start]) 
            ) {
                ++i;
            }

            ans = max(ans, i - start);

            // 上一组的末尾可能是下一组的开头
            --i;
        }

        return ans;
    }
};

如果nums的长度为n，则此算法时间复杂度为O(n)，空间复杂度为O(1)。