题目描述:
给你一个数组 nums ,数组中有 2n 个元素,按 [x1,x2,…,xn,y1,y2,…,yn] 的格式排列。请你将数组按 [x1,y1,x2,y2,…,xn,yn] 格式重新排列,返回重排后的数组。
提示:
1 <= n <= 500
nums.length == 2n
1 <= nums[i] <= 10^3
示例 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]
代码如下:
class Solution {
public:
vector<int> shuffle(vector<int>& nums, int n) {
int size=nums.size();
vector<int>v;
stack<int>s1;
stack<int>s2;
for(int i=n-1;i>=0;i--){
s1.push(nums[i]);
}
for(int j=size-1;j>=n;j--){
s2.push(nums[j]);
}
for(int k=0;k<size;k++){
if(k%2==0){
v.push_back(s1.top());
s1.pop();
}
else{
v.push_back(s2.top());
s2.pop();
}
}
return v;
}
};
class Solution {
public:
vector<int> shuffle(vector<int>& nums, int n) {
vector<int>v;
for(int i=0;i<n;i++){
v.push_back(nums[i]);
v.push_back(nums[i+n]);
}
return v;
}
};
执行结果: