【算法训练】LeetCode#1470 重新排列数组

一、描述

1470. 重新排列数组

给你一个数组 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]

二、思路

如果只是简单构建一个新的数组通过双指针的方式写入数据有些low了；

尝试写一个原地算法：

• 从xn开始遍历，xn所在位置为n-1，其小于n，所以其目标位置为2*n，将其放入2*n；
• 对于2*n位置，其大于n，因此其目标位置为(2*n-n)*2+1；
• 循环往复，当位置回到n-1时，数组排列完毕。
• 失败了，没有明确的结束状态。

三、解题

Java

public class LeetCode1470 {
    public int[] shuffle(int[] nums, int n) {
        int[] ans = new int[2*n];
        int i = 0 , j = n;
        int loc = 0;
        while(loc<2*n){
          ans[loc++] = nums[i++];
          ans[loc++] = nums[j++];
        }
        return ans;
    }
}

C++

vector<int> shuffle(vector<int>& nums, int n) {
    vector<int> ans(2*n);
    for(int i = 0 ; i < n ; i++){
        ans[2*i] = nums[i];
        ans[2*i+1] = nums[i+n];
    }
    return ans;
}