|
33 | 33 | <p><strong>进阶:</strong><br>
|
34 | 34 | 你能否用 O(<em>n</em>) 时间复杂度完成此题?</p>
|
35 | 35 |
|
36 |
| - |
37 | 36 | ## 解法
|
38 | 37 |
|
39 | 38 | <!-- 这里可写通用的实现逻辑 -->
|
40 | 39 |
|
| 40 | +动态规划。 |
| 41 | + |
| 42 | +设 up 表示以前 i 个元素中的某一个元素结尾的最长上升摆动序列的长度,down 表示以前 i 个元素中的某一个元素结尾的最长下降摆动序列的长度。初始 `up = 1`, `down = 1`。 |
| 43 | + |
| 44 | +从数组下标 1 开始遍历: |
| 45 | + |
| 46 | +- 若 `nums[i] > nums[i - 1]`,则需要更新最长上升摆动序列的长度:`up = max(up, down + 1)` |
| 47 | +- 若 `nums[i] < nums[i - 1]`,则需要更新最长下降摆动序列的长度:`down = max(down, up + 1)` |
| 48 | + |
| 49 | +最后返回 `max(up, down)` 即可。 |
| 50 | + |
41 | 51 | <!-- tabs:start -->
|
42 | 52 |
|
43 | 53 | ### **Python3**
|
44 | 54 |
|
45 | 55 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
46 | 56 |
|
47 | 57 | ```python
|
48 |
| - |
| 58 | +class Solution: |
| 59 | + def wiggleMaxLength(self, nums: List[int]) -> int: |
| 60 | + up = down = 1 |
| 61 | + for i in range(1, len(nums)): |
| 62 | + if nums[i] > nums[i - 1]: |
| 63 | + up = max(up, down + 1) |
| 64 | + elif nums[i] < nums[i - 1]: |
| 65 | + down = max(down, up + 1) |
| 66 | + return max(up, down) |
49 | 67 | ```
|
50 | 68 |
|
51 | 69 | ### **Java**
|
52 | 70 |
|
53 | 71 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
54 | 72 |
|
55 | 73 | ```java
|
| 74 | +class Solution { |
| 75 | + public int wiggleMaxLength(int[] nums) { |
| 76 | + int up = 1, down = 1; |
| 77 | + for (int i = 1; i < nums.length; ++i) { |
| 78 | + if (nums[i] > nums[i - 1]) { |
| 79 | + up = Math.max(up, down + 1); |
| 80 | + } else if (nums[i] < nums[i - 1]) { |
| 81 | + down = Math.max(down, up + 1); |
| 82 | + } |
| 83 | + } |
| 84 | + return Math.max(up, down); |
| 85 | + } |
| 86 | +} |
| 87 | +``` |
| 88 | + |
| 89 | +### **C++** |
| 90 | + |
| 91 | +```cpp |
| 92 | +class Solution { |
| 93 | +public: |
| 94 | + int wiggleMaxLength(vector<int>& nums) { |
| 95 | + int up = 1, down = 1; |
| 96 | + for (int i = 1; i < nums.size(); ++i) { |
| 97 | + if (nums[i] > nums[i - 1]) { |
| 98 | + up = max(up, down + 1); |
| 99 | + } else if (nums[i] < nums[i - 1]) { |
| 100 | + down = max(down, up + 1); |
| 101 | + } |
| 102 | + } |
| 103 | + return max(up, down); |
| 104 | + } |
| 105 | +}; |
| 106 | +``` |
56 | 107 |
|
| 108 | +### **Go** |
| 109 | +
|
| 110 | +```go |
| 111 | +func wiggleMaxLength(nums []int) int { |
| 112 | + up, down := 1, 1 |
| 113 | + for i := 1; i < len(nums); i++ { |
| 114 | + if nums[i] > nums[i-1] { |
| 115 | + up = max(up, down+1) |
| 116 | + } else if nums[i] < nums[i-1] { |
| 117 | + down = max(down, up+1) |
| 118 | + } |
| 119 | + } |
| 120 | + return max(up, down) |
| 121 | +} |
| 122 | +
|
| 123 | +func max(a, b int) int { |
| 124 | + if a > b { |
| 125 | + return a |
| 126 | + } |
| 127 | + return b |
| 128 | +} |
57 | 129 | ```
|
58 | 130 |
|
59 | 131 | ### **...**
|
|
0 commit comments