LeetCode 2149. 按符号重排数组（双指针）

文章目录

- 1. 题目
- 2. 解题

1. 题目

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

你需要重排 nums 中的元素，使修改后的数组满足下述条件：

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

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

示例 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 * 10^5
nums.length 是 偶数
1 <= |nums[i]| <= 10^5
nums 由 相等 数量的正整数和负整数组成

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rearrange-array-elements-by-sign
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

双指针找到第一个正数和负数
分别往后挪动

class Solution {
public:vector<int> rearrangeArray(vector<int>& nums) {int pos = -1, neg = -1, n = nums.size();for(int i = 0; i < n; ++i){if(pos==-1 && nums[i] > 0)pos = i;if(neg==-1 && nums[i] < 0)neg = i;if(pos!=-1 && neg!=-1)break;}vector<int> ans(n);for(int i = 0; i < n; ++i){ans[i] = (i%2==0) ? nums[pos++] : nums[neg++];while(i%2==0 && pos<n && nums[pos]<0)pos++;while(i%2==1 && neg<n && nums[neg]>0)neg++;}return ans;}
};