Question

2149. 按符号重排数组

English Version

题目描述

给你一个下标从 0 开始的整数数组 nums ，数组长度为 偶数 ，由数目 相等 的正整数和负整数组成。

你需要返回满足下述条件的数组 nums：

1. 任意 连续 的两个整数 符号相反
2. 对于符号相同的所有整数，保留 它们在 nums 中的 顺序 。
3. 重排后数组以正整数开头。

重排元素满足上述条件后，返回修改后的数组。

 

示例 1：

输入：nums = [3,1,-2,-5,2,-4]
输出：[3,-2,1,-5,2,-4]
解释：
nums 中的正整数是 [3,1,2] ，负整数是 [-2,-5,-4] 。
重排的唯一可行方案是 [3,-2,1,-5,2,-4]，能满足所有条件。
像 [1,-2,2,-5,3,-4]、[3,1,2,-2,-5,-4]、[-2,3,-5,1,-4,2] 这样的其他方案是不正确的，因为不满足一个或者多个条件。 

示例 2：

输入：nums = [-1,1]
输出：[1,-1]
解释：
1 是 nums 中唯一一个正整数，-1 是 nums 中唯一一个负整数。
所以 nums 重排为 [1,-1] 。

 

提示：

• 2 <= nums.length <= 2 * 105
• nums.length 是 偶数
• 1 <= |nums[i]| <= 105
• nums 由 相等 数量的正整数和负整数组成

 

不需要原地进行修改。

解法

方法一

class Solution:
  def rearrangeArray(self, nums: List[int]) -> List[int]:
    ans = [0] * len(nums)
    i, j = 0, 1
    for num in nums:
      if num > 0:
        ans[i] = num
        i += 2
      else:
        ans[j] = num
        j += 2
    return ans

class Solution {

  public int[] rearrangeArray(int[] nums) {
    int[] ans = new int[nums.length];
    int i = 0, j = 1;
    for (int num : nums) {
      if (num > 0) {
        ans[i] = num;
        i += 2;
      } else {
        ans[j] = num;
        j += 2;
      }
    }
    return ans;
  }
}

class Solution {
public:
  vector<int> rearrangeArray(vector<int>& nums) {
    vector<int> ans(nums.size());
    int i = 0, j = 1;
    for (int num : nums) {
      if (num > 0) {
        ans[i] = num;
        i += 2;
      } else {
        ans[j] = num;
        j += 2;
      }
    }
    return ans;
  }
};

func rearrangeArray(nums []int) []int {
  ans := make([]int, len(nums))
  i, j := 0, 1
  for _, num := range nums {
    if num > 0 {
      ans[i] = num
      i += 2
    } else {
      ans[j] = num
      j += 2
    }
  }
  return ans
}

function rearrangeArray(nums: number[]): number[] {
  let ans = [];
  let i = 0,
    j = 1;
  for (let num of nums) {
    if (num > 0) {
      ans[i] = num;
      i += 2;
    } else {
      ans[j] = num;
      j += 2;
    }
  }
  return ans;
}