leetcode 376. 摆动序列 python

码龄10年

关注

原创最新推荐文章于 2025-03-22 21:48:37 发布 935 阅读

WIN下使用visual studio 2022安装GMSSL命令行工具 2.1 使用管理员权限打开visual studio（非常重要,否则最后一步 nmake install会失败）右键vs2022，以管理员身份运行即可。2.4 配置环境变量,gmssl可执行程序默认会被安装到如下目录。将如下路径添加到系统环境变量即可。添加后的环境变量如下。2.编译解压后的gmssl源码,编译过程主要参考install.md中的内容,README.md中的步骤不完整。2.2 打开vs2022的命令终端,右键文件,选择在终端中打开。1.下载gmssl源码,解压到相应目录。阅读详情

题目描述：

题解：

1.如果输入nums只有一个元素返回1，如果nums有两个元素，如果两个数字不等则返回2，否则返回1.

2.计算nums数组中相邻数字的差，保存在gaps数组。最终摇摆序列的长度=max(第一个差值为正数的序列最长长度，第一个差值为负数的序列最长长度)+1，分别用p_len n_len记录第一个差值为正数和负数的序列长度，p_start，n_start分别记录差值为正数和负数的开始位置。

3.分别从p_start依次寻找差值为负，正，负....记录长度，n_start做相同操作。

class Solution:
    def wiggleMaxLength(self, nums: List[int]) -> int:
        if len(nums) == 1:
            return len(nums)
        if len(nums) == 2:
            if nums[0] != nums[1]:
                return 2
            else:
                return 1
        gaps = []
        n = len(nums)
        for i in range(n - 1):
            gaps.append(nums[i + 1] - nums[i])
        p_len = 0
        n_len = 0
        p_start = 0
        n_start = 0
        find_p = 0
        find_n = 0
        for i in range(n - 1):
            if gaps[i] > 0:
                p_start = i
                p_len = 1
                find_p = 1
                break
        for i in range(n - 1):
            if gaps[i] < 0:
                n_start = i
                n_len = 1
                find_n = 1
                break
        p_flag = -1
        for i in range(p_start + 1, n - 1):
            if gaps[i] * p_flag > 0:
                p_len = p_len + 1
                p_flag = -p_flag
        n_flag = 1
        for i in range(n_start + 1, n - 1):
            if gaps[i] * n_flag > 0:
                n_len = n_len + 1
                n_flag = -n_flag
        return max(p_len, n_len) + 1