LeetCode刷题6——重新排列数组2020.10.24

题目

给你一个数组 nums ，数组中有 2n 个元素，按 [x1,x2,…,xn,y1,y2,…,yn] 的格式排列。

请你将数组按 [x1,y1,x2,y2,…,xn,yn] 格式重新排列，返回重排后的数组。

示例 1：

输入：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]
示例 2：

输入：nums = [1,2,3,4,4,3,2,1], n = 4
输出：[1,4,2,3,3,2,4,1]
示例 3：

输入：nums = [1,1,2,2], n = 2
输出：[1,2,1,2]

思路

按照排列规则，分为两类，第一类是原数组的i/2应该赋值给新数组0,2,4…等项；第二类为原数组的i/2+n项应该赋值给1,3,5…等项。代码如下
class Solution {
public:
vector shuffle(vector& nums, int n) {
vector renums(2n);
for(int i=0;i<2n;i=i+2){
renums[i]=nums[i/2];
renums[i+1]=nums[n+i/2];
}
return renums;
}
};