题目
2765. 最长交替子数组
分析
为了得到数组 nums
中的最长交替子数组的长度,需要分别计算以每个下标结尾的最长交替子数组的长度。为了方便处理,计算过程中需要考虑长度等于 1 的最长交替子数组,再返回结果时判断最长交替子数组的长度是否大于 1。
以下标 0 结尾的最长交替子数组的长度等于 1。对于 i>0 ,计算以下标 i 结尾的最长交替子数组的长度时需要考虑以下标 i-1 结尾的最长交替子数组的长度、当前相邻元素之差 curDiff = nums[i] - nums[i-1] 与上一个相邻元素之差 preDiff = nums[i-1]-nums[i-2],分别考虑以下情况:
① 如果 preDiff * curDiff = -1,则可以将 nums[i] 添加到以 nums[i-1]结尾的最长交替子数组的末尾,使最长交替子数组的长度加 1,因此以下标 i 结尾的最长交替子数组的长度等于以下标 i-1 结尾的最长交替子数组的长度+1。
② 如果 preDiff * curDiff != -1 且 curDiff = 1,则以下标 i 结尾的最长交替子数组包含 nums[i-]和nums[i],因此以下标 i 结尾的最长交替子数组的长度等于 2。
③ 如果 preDiff * curDiff != 1 且 curDiff !=1 ,则以下标 i 结尾的最长交替子数组包含 nums[i],因此以下标 i 结尾的最长交替子数组的长度等于 1。
特别的,当 i=1 时,preDiff 不存在,因此一定不满足 preDiff * curDiff = -1,可以将 preDiff 初始化为一个绝对值大于 1 的整数。
遍历所有大于 0 的下标 i 之后,即可得到最长交替子数组的长度。如果最长交替子数组的长度大于 1,则返回最长交替子数组的长度,否则返回 -1。
代码
class Solution {
public int alternatingSubarray(int[] nums) {
int maxLength = -1;
int curLength = 1;
int preDiff = -2;
int curDiff = 1;
for(int i = 1;i < nums.length;i ++) {
curDiff = nums[i] - nums[i - 1];
if(preDiff * curDiff == -1) {
curLength ++;
}else if(curDiff == 1) {
curLength = 2;
}else {
curLength = 1;
}
maxLength = Math.max(maxLength,curLength);
preDiff = curDiff;
}
return maxLength > 1 ? maxLength : -1;
}
}
参考
参考题解