给你一个下标从 0 开始的整数数组 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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
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;
}
};
192 ms 121 MB C++