题目描述
给你一个数组 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
算法分析
模拟
- 两个指针
i
,j
同时走,i
从0
开始走到n - 1
,j
从n
开始走到2n - 1
,先把i
的元素放进ans[]
中,再把j
的元素放进ans[]
中,依次循环
时间复杂度 $O(n)$
Java 代码
class Solution {
public int[] shuffle(int[] nums, int n) {
int[] ans = new int[2 * n];
int k = 0;
for(int i = 0,j = n;j < 2 * n;i ++,j ++)
{
ans[k ++] = nums[i];
ans[k ++] = nums[j];
}
return ans;
}
}