Easy

题目描述

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

提示：

  • 1 <= n <= 500
  • nums.length == 2n
  • 1 <= nums[i] <= 10^3

解题思路

这是一个简单的数组重排问题。我们需要将原数组从中间分割，前半部分是 [x1,x2,...,xn]，后半部分是 [y1,y2,...,yn]，然后交替取元素组成新数组。

方法分析：

  1. 双指针法：使用两个指针分别指向前半部分和后半部分的开始位置，交替取元素。这是最直观的方法。

  2. 直接索引法：通过数学公式直接计算每个位置应该放置的元素。对于结果数组的偶数位置放置前半部分元素，奇数位置放置后半部分元素。

  3. 原地修改法：可以通过位运算等技巧实现原地修改，但会增加复杂度，对于此题不推荐。

推荐解法：双指针法

  • 创建结果数组，使用两个指针 i 和 j 分别指向位置 0 和 n
  • 交替从两个指针位置取元素填入结果数组
  • 时间复杂度 O(n)，空间复杂度 O(n)，代码简洁易懂

代码实现

class Solution {
public:
    vector<int> shuffle(vector<int>& nums, int n) {
        vector<int> result(2 * n);
        int i = 0, j = n;
        for (int k = 0; k < 2 * n; k++) {
            if (k % 2 == 0) {
                result[k] = nums[i++];
            } else {
                result[k] = nums[j++];
            }
        }
        return result;
    }
};

class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        result = []
        i, j = 0, n
        for k in range(2 * n):
            if k % 2 == 0:
                result.append(nums[i])
                i += 1
            else:
                result.append(nums[j])
                j += 1
        return result

public class Solution {
    public int[] Shuffle(int[] nums, int n) {
        int[] result = new int[2 * n];
        int i = 0, j = n;
        for (int k = 0; k < 2 * n; k++) {
            if (k % 2 == 0) {
                result[k] = nums[i++];
            } else {
                result[k] = nums[j++];
            }
        }
        return result;
    }
}

/**
 * @param {number[]} nums
 * @param {number} n
 * @return {number[]}
 */
var shuffle = function(nums, n) {
    const result = [];
    for (let i = 0; i < n; i++) {
        result.push(nums[i], nums[i + n]);
    }
    return result;
};
在 LeetCode 上运行： 力扣中文版 LeetCode 国际版

复杂度分析

复杂度类型双指针法
时间复杂度O(n)
空间复杂度O(n)