算法学习1470. 重新排列数组（java / c / c++ / python / go / rust）

Posted 2021-11-24 二当家的白帽子

非常感谢你阅读本文~
欢迎【👍点赞】【⭐收藏】【📝评论】~
放弃不难，但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子 https://le-yi.blog.csdn.net/ 博客原创~

---

文章目录

• 1470. 重新排列数组：
• 样例 1
• 样例 2
• 样例 3
• 提示
• 分析
• 题解
• • java
  • c
  • c++
  • python
  • go
  • rust
• 原题传送门：https://leetcode-cn.com/problems/shuffle-the-array/

---

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]

提示

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

---

分析

• 这道算法题是简单题，所以又可以重拳出击了。
• 返回的结果不算在空间复杂度上，所以直接开辟新的空间赋值返回结果就可以了。
• 但是二当家的发现可以难为自己，于是又灵机一动，原地排列，我真是个小机灵鬼。
• 想要原地排列就一定要存储中间结果，原来的位置被占以后，需要知道原来位置的数字。
• 提示里说每个数字最大不超过1000，而且都是正数，所以只需要用10个二进制位就够了。
• 参数传递进来的数组一般都是int型数组，int型一般都是32位，够存题目中的2个数了。

---

题解

java

class Solution {
    public int[] shuffle(int[] nums, int n) {
        // 申请新的数组，这样用空间换时间
		int[] ans = new int[n * 2];
		for (int i = 0; i < n; ++i) {
			ans[2 * i] = nums[i];
			ans[2 * i + 1] = nums[n + i];
		}
		return ans;
    }
}

class Solution {
    public int[] shuffle(int[] nums, int n) {
        // 原地排列，时间换空间
		for (int i = 0; i < n; ++i) {
			// 把重排列的值左移10位
			nums[2 * i] |= (nums[i] & 0b1111111111) << 10;
			nums[2 * i + 1] |= (nums[n + i] & 0b1111111111) << 10;
		}
		// 将结果右移10位，低10位的原值没用了
		for (int i = 0; i < n; ++i) {
			nums[2 * i] >>= 10;
			nums[2 * i + 1] >>= 10;
		}
		return nums;
    }
}

---

c

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* shuffle(int* nums, int numsSize, int n, int* returnSize){
    // 申请新的数组，这样用空间换时间
    int *ans = (int *) malloc(sizeof(int) * numsSize);
    for (int i = 0; i < n; ++i) {
        ans[2 * i] = nums[i];
        ans[2 * i + 1] = nums[n + i];
    }
    *returnSize = numsSize;
    return ans;
}

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* shuffle(int* nums, int numsSize, int n, int* returnSize){
     // 原地排列，时间换空间
    for (int i = 0; i < n; ++i) {
        // 把重排列的值左移10位
        nums[2 * i] |= (nums[i] & 0b1111111111) << 10;
        nums[2 * i + 1] |= (nums[n + i] & 0b1111111111) << 10;
    }
    // 将结果右移10位，低10位的原值没用了
    for (int i = 0; i < n; ++i) {
        nums[2 * i] >>= 10;
        nums[2 * i + 1] >>= 10;
    }
    *returnSize = numsSize;
    return nums;
}

---

c++

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

class Solution {
public:
    vector<int> shuffle(vector<int>& nums, int n) {
        // 原地排列，时间换空间
        for (int i = 0; i < n; ++i) {
            // 把重排列的值左移10位
            nums[2 * i] |= (nums[i] & 0b1111111111) << 10;
            nums[2 * i + 1] |= (nums[n + i] & 0b1111111111) << 10;
        }
        // 将结果右移10位，低10位的原值没用了
        for (int i = 0; i < n; ++i) {
            nums[2 * i] >>= 10;
            nums[2 * i + 1] >>= 10;
        }
        return nums;
    }
};

---

python

class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        # 申请新的数组，这样用空间换时间
        ans = []
        for i in range(n):
            ans.append(nums[i]) 
            ans.append(nums[n + i])
        return ans

class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        # 原地排列，时间换空间
        for i in range(n):
            # 把重排列的值左移10位
            nums[2 * i] |= (nums[i] & 0b1111111111) << 10;
            nums[2 * i + 1] |= (nums[n + i] & 0b1111111111) << 10
        # 将结果右移10位，低10位的原值没用了
        for i in range(n):
            nums[2 * i] >>= 10
            nums[2 * i + 1] >>= 10
        return nums

---

go

func shuffle(nums []int, n int) []int {
    // 申请新的数组，这样用空间换时间
	ans := make([]int, 0, n*2)
	for i := 0; i < n; i++ {
		ans = append(ans, nums[i], nums[n+i])
	}
	return ans
}

func shuffle(nums []int, n int) []int {
    // 原地排列，时间换空间
	for i := 0; i < n; i++ {
		// 把重排列的值左移10位
		nums[2*i] |= (nums[i] & 0b1111111111) << 10
		nums[2*i+1] |= (nums[n+i] & 0b1111111111) << 10
	}
	// 将结果右移10位，低10位的原值没用了
	for i := 0; i < n; i++ {
		nums[2*i] >>= 10
		nums[2*i+1] >>= 10
	}
	return nums
}

---

rust

impl Solution {
    pub fn shuffle(nums: Vec<i32>, n: i32) -> Vec<i32> {
        let n: usize = n as usize;
        // 申请新的数组，这样用空间换时间
        let mut ans = Vec::new();
        (0..n).for_each(|i| {
            ans.push(nums[i]);
            ans.push(nums[n + i]);
        });
        ans
    }
}

impl Solution {
    pub fn shuffle(nums: Vec<i32>, n: i32) -> Vec<i32> {
        let mut nums = nums;
        let n = n as usize;
        // 原地排列，时间换空间
        (0..n).for_each(|i| {
            // 把重排列的值左移10位
            nums[2 * i] |= (nums[i] & 0b1111111111) << 10;
            nums[2 * i + 1] |= (nums[n + i] & 0b1111111111) << 10;
        });
        // 将结果右移10位，低10位的原值没用了
        (0..n).for_each(|i| {
            nums[2 * i] >>= 10;
            nums[2 * i + 1] >>= 10;
        });
        nums
    }
}

---

---

原题传送门：https://leetcode-cn.com/problems/shuffle-the-array/

---