题目地址: https://leetcode.com/problems/wiggle-subsequence/description/
题目描述:
Asequence of numbers is called a wiggle sequence if the differences between successive numbers strictly alternate between positive and negative. The first difference (if one exists) may be either positive or negative. A sequence with fewer than two elements is trivially a wiggle sequence.
Forexample, [1,7,4,9,2,5] is a wiggle sequence because the differences (6,-3,5,-7,3) are alternately positive and negative. In contrast, [1,4,7,2,5] and [1,7,4,5,5] are not wiggle sequences, the first because its first two differences are positive and the second because its last difference is zero.
Given a sequence of integers, return the length of the longest subsequence that is a wiggle sequence. A subsequence is obtained by deleting some number of elements (eventually, also zero) from the original sequence, leaving the remaining elements in their original order.
Example 1:
Input: [1,7,4,9,2,5]
Output: 6
Explanation: The entire sequence is a wiggle sequence.
Example 2:
Input: [1,17,5,10,13,15,10,5,16,8]
Output: 7
Explanation: There are several subsequences that achieve this length. One is [1,17,10,13,10,16,8].
Example 3:
Input: [1,2,3,4,5,6,7,8,9]
Output: 2
Follow up:
Canyou do it in O(n) time?
题目大意
如果一个数组里面,相邻的两个数字的差是正负交替的,那么认为这个是波动序列。求输入的数组里面最长的波动序列长度。
解题方法
明显的DP问题,本来的想法是用个二维DP,可是提交了几遍只通过了部分测试用例。才去看的别人的两个DP数组的解法。
定义了一个记录递增的DP数组inc,一个记录递减的DP数组dec,这两个DP数组分别保存的是开头元素是递增、递减的最长波动序列长度。对于每个位置,从头遍历,如果当前的元素比前面的元素大,应该更新递增数组,否则,如果比前面的数字小,那么应该更新递减数组。
时间复杂度是O(N^2),空间复杂度是O(N).
class Solution(object):
def wiggleMaxLength(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n = len(nums)
if n <= 1:
return n
inc, dec = [1] * n, [1] * n
for x in range(n):
for y in range(x):
if nums[x] > nums[y]:
inc[x] = max(inc[x], dec[y] + 1)
elif nums[x] < nums[y]:
dec[x] = max(dec[x], inc[y] + 1)
return max(inc[-1], dec[-1])
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
其实不需要从头遍历,只需要知道前面元素对应的最长递增和递减数组即可。
时间复杂度是O(N),空间复杂度是O(N).
class Solution(object):
def wiggleMaxLength(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n = len(nums)
if n <= 1:
return n
inc, dec = [1] * n, [1] * n
for x in range(1, n):
if nums[x] > nums[x - 1]:
inc[x] = dec[x - 1] + 1
dec[x] = dec[x - 1]
elif nums[x] < nums[x - 1]:
inc[x] = inc[x - 1]
dec[x] = inc[x - 1] + 1
else:
inc[x] = inc[x - 1]
dec[x] = dec[x - 1]
return max(inc[-1], dec[-1])
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
简单分析代码就可以看出,每个元素都只和它之前的元素相关,因此,只需要使用两个变量即可。
时间复杂度是O(N),空间复杂度是O(1).
class Solution(object):
def wiggleMaxLength(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n = len(nums)
if n <= 1:
return n
inc, dec = 1, 1
for x in range(1, n):
if nums[x] > nums[x - 1]:
inc = dec + 1
elif nums[x] < nums[x - 1]:
dec = inc + 1
return max(inc, dec)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
参考资料:
https://leetcode.com/articles/wiggle-subsequence/ http://www.cnblogs.com/grandyang/p/5697621.html http://bookshadow.com/weblog/2016/07/21/leetcode-wiggle-subsequence/
DDKK.COM 弟弟快看-教程,程序员编程资料站,版权归原作者所有
本文经作者:负雪明烛 授权发布,任何组织或个人未经作者授权不得转发