酷代码 AI
菜单
全部AI  /  热门  /  AI写代码神器
服务商

题目描述 如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为摆动序列 。第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。 例如,[1, 7, 4, 9, 2, 5]是一个摆动序列,因为差值(6, -3, 5, -7, 3)是正负交替出现的。 相反,[1, 4, 7, 2, 5]和[1, 7, 4, 5, 5]不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。 子序列可以通过从原始序列中删除一些(也可以不删除)元素来获得,剩下的元素保持其原始顺序。 给定n个整数构成的序列,输出子序列中最长的摆动序列长度。 输入描述 第一行一个整数 n 第二行 n个整数a1,a2,...,an 输出描述 输出一个整数,表示子序列中最长的摆动序列长度。 样例1 输入 6 1 7 4 9 2 5 输出 6 样例2 输入 10 1 17 5 10 13 15 10 5 16 8 输出 7 提示 1 <= n <= 1000 0 <= ai <= 1000 请用C++完成此代码

```cpp #include <iostream> #include <vector> using namespace std; int wiggleMaxLength(vector<int>& nums) { int n = nums.size(); if (n < 2) return n; int up = 1, down = 1; for (int i = 1; i < n; ++i) { if (nums[i] > nums[i-1]) { up = down + 1; } else if (nums[i] < nums[i-1]) { down = up + 1; } } return max(up, down); } int main() { int n; cin >> n; vector<int> nums(n); for (int i = 0; i < n; ++i) { cin >> nums[i]; } cout << wiggleMaxLength(nums) << endl; return 0; } ``` 代码解释: 1. 使用动态规划思想,维护两个状态变量`up`和`down`,分别表示以当前元素结尾时最后一步是上升或下降的最长摆动序列长度。 2. 遍历数组时,当当前元素大于前一个元素时,更新`up`为`down + 1`;当当前元素小于前一个元素时,更新`down`为`up + 1`。 3. 最终结果为`up`和`down`中的较大值。 三条额外需要注意的逻辑点: - 当输入数组长度小于2时直接返回数组长度,因为单元素或空数组本身就是摆动序列。 - 相等的相邻元素不会影响`up`和`down`的值,因此会被自动跳过。 - 该算法时间复杂度为O(n),空间复杂度为O(1),是最优解法。 [2025-07-19 17:37:51 | AI写代码神器 | 373点数解答]

相关提问