1470-重新排列数组

Raphael Liu Lv10
  2023-09-13 00:47:26 2023-09-13 00:47:26 Created   2023-09-13 16:06:24 2023-09-13 16:06:24 Updated    

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

方法一：一次遍历

用 ans 表示结果数组，数组 nums 和 ans 的长度都是 2n。对于 0 \le i < n，重新排列数组的规则如下：

  • nums}[i] 填到 ans}[2 \times i]；

  • nums}[i + n] 填到 ans}[2 \times i + 1]。

根据该规则将原数组 nums 中的元素依次填入结果数组 ans 中，即可得到重新排列后的数组。

[sol1-Python3]
1
2
3
4
5
6
7
class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        ans = [0] * (2 * n)
        for i in range(n):
            ans[2 * i] = nums[i]
            ans[2 * i + 1] = nums[n + i]
        return ans
[sol1-Java]
1
2
3
4
5
6
7
8
9
10
class Solution {
    public int[] shuffle(int[] nums, int n) {
        int[] ans = new int[2 * n];
        for (int i = 0; i < n; i++) {
            ans[2 * i] = nums[i];
            ans[2 * i + 1] = nums[i + n];
        }
        return ans;
    }
}
[sol1-C#]
1
2
3
4
5
6
7
8
9
10
public class Solution {
    public int[] Shuffle(int[] nums, int n) {
        int[] ans = new int[2 * n];
        for (int i = 0; i < n; i++) {
            ans[2 * i] = nums[i];
            ans[2 * i + 1] = nums[i + n];
        }
        return ans;
    }
}
[sol1-C++]
1
2
3
4
5
6
7
8
9
10
11
class Solution {
public:
    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;
    }
};
[sol1-C]
1
2
3
4
5
6
7
8
9
int* shuffle(int* nums, int numsSize, int n, int* returnSize){
    int *ans = (int *)malloc(sizeof(int) * n * 2);
    for (int i = 0; i < n; i++) {
        ans[2 * i] = nums[i];
        ans[2 * i + 1] = nums[i + n];
    }
    *returnSize = n * 2;
    return ans;
}
[sol1-JavaScript]
1
2
3
4
5
6
7
8
var shuffle = function(nums, n) {
    const ans = new Array(2 * n).fill(0);
    for (let i = 0; i < n; i++) {
        ans[2 * i] = nums[i];
        ans[2 * i + 1] = nums[i + n];
    }
    return ans;
};
[sol1-Golang]
1
2
3
4
5
6
7
8
func shuffle(nums []int, n int) []int {
    ans := make([]int, n*2)
    for i, num := range nums[:n] {
        ans[2*i] = num
        ans[2*i+1] = nums[n+i]
    }
    return ans
}

复杂度分析

  • 时间复杂度：O(n)，其中 n 是给定的参数。需要遍历长度为 2n 的数组 nums 一次将数组重新排列，每个元素重新排列的时间是 O(1)。

  • 空间复杂度：O(1)。注意返回值不计入空间复杂度。

 Comments
On this page
1470-重新排列数组