力扣每日一题0829-1470. 重新排列数组

181 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第29天,点击查看活动详情

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

思路一

因为题目限制了每一个元素 nums[i] 最大只有可能是 1000,这就意味着每一个元素只占据了 10 个 bit。(2^10 - 1 = 1023 > 1000)

而一个 int 有 32 个 bit,所以我们还可以使用剩下的 22 个 bit 做存储。实际上,每个 int,我们再借 10 个 bit 用就好了。

因此,在下面的代码中,每一个 nums[i] 的最低的十个 bit(0-9 位),我们用来存储原来 nums[i] 的数字;再往前的十个 bit(10-19 位),我们用来存储重新排列后正确的数字是什么。

在循环中,我们每次首先计算 nums[i] 对应的重新排列后的索引 j,之后,取 nums[i] 的低 10 位(nums[i] & 1023),即 nums[i] 的原始信息,把他放到 nums[j] 的高十位上。

最后,每个元素都取高 10 位的信息(e >> 10),即是答案。

一次遍历

ans\textit{ans} 表示结果数组,数组 nums\textit{nums}ans\textit{ans} 的长度都是 2n2n。对于 0i<n0 \le i < n,重新排列数组的规则如下:

  • nums[i]nums[i] 填到 ans[2×i]\textit{ans}[2 \times i]
  • nums[i+n]nums[i+n] 填到 ans[2×i+1]\textit{ans}[2 \times i + 1]

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

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;
};

复杂度分析

  • 时间复杂度:O(n)O(n),其中 nn 是给定的参数。需要遍历长度为 2n2n 的数组 nums\textit{nums} 一次将数组重新排列,每个元素重新排列的时间是 O(1)O(1)
  • 空间复杂度: O(1)O(1)。注意返回值不计入空间复杂度。