You are given a 0-indexed integer array nums
of even length consisting of an equal number of positive and negative integers.
You should rearrange the elements of nums
such that the modified array follows the given conditions:
nums
is preserved.Return the modified array after rearranging the elements to satisfy the aforementioned conditions.
Example 1:
Input: nums = [3,1,-2,-5,2,-4] Output: [3,-2,1,-5,2,-4] Explanation: The positive integers in nums are [3,1,2]. The negative integers are [-2,-5,-4]. The only possible way to rearrange them such that they satisfy all conditions is [3,-2,1,-5,2,-4]. Other ways such as [1,-2,2,-5,3,-4], [3,1,2,-2,-5,-4], [-2,3,-5,1,-4,2] are incorrect because they do not satisfy one or more conditions.
Example 2:
Input: nums = [-1,1] Output: [1,-1] Explanation: 1 is the only positive integer and -1 the only negative integer in nums. So nums is rearranged to [1,-1].
Constraints:
2 <= nums.length <= 2 * 105
nums.length
is even1 <= |nums[i]| <= 105
nums
consists of equal number of positive and negative integers.class Solution:
def rearrangeArray(self, nums: List[int]) -> List[int]:
n = len(nums)
ret = [0] * n
pos_idx, neg_idx = 0, 1
for num in nums:
if num > 0:
ret[pos_idx] = num
pos_idx += 2
else:
ret[neg_idx] = num
neg_idx += 2
return ret
문제의 조건에 맞게 구현하면 된다.
[양수 - 음수 - 양수 - 음수 ...]
이런 식으로..
규칙을 염두에 두면, 자연스레 각 idx
는 += 2
로 갱신해야 함을 알 수 있다.