每日leetcode

1470. 重新排列数组 - 力扣（LeetCode）

题目

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

思路

1. 定义一个结果数组，将i从0开始遍历到n-1，每次将第i位和第n+i位插入数组，最后返回即可。

代码实现

class Solution {
public:
    vector<int> shuffle(vector<int>& nums, int n) {
        vector<int> ans;
        for(int i = 0; i < n; ++i) {
            ans.push_back(nums[i]);
            ans.push_back(nums[n+i]);
        }
        return ans;
    }
};

复杂度分析

• 时间复杂度：O(n)。
• 空间复杂度：O(n)/O(1)（考虑输出/不考虑输出）。

官方题解

• 还是有优化的角度的，如果知道最后输出数组的规模的话，应该在一开始就分配好空间，vector每次插入会预先分配一定的地址空间，如果满了的话还会重新分配，均摊复杂度是O(1)的，但是实际的开销肯定是不如一开始直接建好的，因为实际开销其实是O(log n)的。
• 复现：

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