题目描述
给你一个数组 nums
,数组中有 2n
个元素,按 [x1,x2,...,xn,y1,y2,...,yn]
的格式排列。
请你将数组按 [x1,y1,x2,y2,...,xn,yn]
格式重新排列,返回重排后的数组。
样例
输入:nums = [2,5,1,3,4,7], n = 3
输出:[2,3,5,4,1,7]
解释:由于 x1=2, x2=5, x3=1, y1=3, y2=4, y3=7,所以答案为 [2,3,5,4,1,7]
输入:nums = [1,2,3,4,4,3,2,1], n = 4
输出:[1,4,2,3,3,2,4,1]
输入:nums = [1,1,2,2], n = 2
输出:[1,2,1,2]
限制
1 <= n <= 500
nums.length == 2n
1 <= nums[i] <= 10^3
算法
(模拟) $O(n)$
- 定义一个新数组,按照题目描述将数组重排。
时间复杂度
- 每个位置遍历一次,故时间复杂度为 $O(n)$。
空间复杂度
- 需要额外 $O(n)$ 的空间存储答案。
C++ 代码
class Solution {
public:
vector<int> shuffle(vector<int>& nums, int n) {
vector<int> res(2 * n);
for (int i = 0, j = 0; i < n; i++) {
res[j++] = nums[i];
res[j++] = nums[i + n];
}
return res;
}
};