使用双指针技巧巧解leetcode 2149：以符号为准重排数组元素（python代码）

2023-10-01 00:20:38

前言

在前几篇文章中，我们已经学习了双指针的多种用法，并且成功地解决了leetcode的众多题目。今天，我们将继续学习双指针技巧，并使用它来解决leetcode 2149：以符号为准重排数组元素。

题目介绍

leetcode 2149 的题目如下：

给定一个整数数组 nums，重排数组元素，使得所有正数前面全是负数。

返回将数组重新排列后的结果。

示例 1：
输入：nums = [3,1,-2,-5,2,-3]
输出：[-2,-3,-5,3,1,2]
解释：结果数组中的正数前面全是负数。

示例 2：
输入：nums = [1,-1]
输出：[1,-1]
解释：所有正数前面全是负数。

思路分析

为了解决这个问题，我们可以使用双指针技巧。具体步骤如下：

定义两个指针，分别指向数组的开头和结尾。
当两个指针相遇时，算法结束。
如果左指针指向负数，右指针指向正数，则交换两个元素的值。
如果左指针指向正数，右指针指向负数，则继续移动两个指针。
如果左指针指向负数，右指针指向正数，则交换两个元素的值，并继续移动两个指针。

Python代码

def rearrange_array(nums):
  """
  重排数组元素，使得所有正数前面全是负数。

  参数：
    nums: 输入的整数数组

  返回：
    将数组重新排列后的结果
  """

  left = 0
  right = len(nums) - 1

  while left < right:
    if nums[left] < 0 and nums[right] > 0:
      nums[left], nums[right] = nums[right], nums[left]
      left += 1
      right -= 1
    elif nums[left] > 0 and nums[right] < 0:
      left += 1
      right -= 1
    elif nums[left] > 0 and nums[right] > 0:
      right -= 1
    else:
      left += 1

  return nums


# 测试代码
nums = [3, 1, -2, -5, 2, -3]
print(rearrange_array(nums))  # [-2, -3, -5, 3, 1, 2]

nums = [1, -1]
print(rearrange_array(nums))  # [1, -1]