摆动排序

"""
摆动排序。

给定一个未排序的数组 nums，重新排列它，使得 nums[0] < nums[1] > nums[2] < nums[3]...。
例如：
如果输入的数字是 [3, 5, 2, 1, 6, 4]
一种可能的摆动排序结果是 [3, 5, 1, 6, 2, 4]。
"""

def wiggle_sort(nums: list) -> list:
    """
    摆动排序的Python实现。
    示例：
    >>> wiggle_sort([0, 5, 3, 2, 2])
    [0, 5, 2, 3, 2]
    >>> wiggle_sort([])
    []
    >>> wiggle_sort([-2, -5, -45])
    [-45, -2, -5]
    >>> wiggle_sort([-2.1, -5.68, -45.11])
    [-45.11, -2.1, -5.68]
    """
    for i, _ in enumerate(nums):
        # 如果索引是奇数，且前一个元素大于当前元素，或者索引是偶数，且前一个元素小于当前元素
        if (i % 2 == 1) == (nums[i - 1] > nums[i]):
            # 交换前一个元素和当前元素的位置
            nums[i - 1], nums[i] = nums[i], nums[i - 1]

    return nums


if __name__ == "__main__":
    # 获取用户输入的数组元素
    print("Enter the array elements:")
    array = list(map(int, input().split()))
    
    # 打印未排序的数组
    print("The unsorted array is:")
    print(array)
    
    # 打印经过摆动排序后的数组
    print("Array after Wiggle sort:")
    print(wiggle_sort(array))