算法学习1920. 基于排列构建数组（java / c / c++ / python / go / rust）

Posted 2021-10-16 二当家的白帽子

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

---

文章目录

• 1920. 基于排列构建数组：
• 样例 1：
• 样例 2：
• 提示：
• 分析
• 题解
• • java
  • c
  • c++
  • python
  • go
  • rust
• 原题传送门

---

1920. 基于排列构建数组：

给你一个 从 0 开始的排列 nums（下标也从 0 开始）。请你构建一个 同样长度 的数组 ans ，其中，对于每个 i（0 <= i < nums.length），都满足 ans[i] = nums[nums[i]] 。返回构建好的数组 ans 。

从 0 开始的排列 nums 是一个由 0 到 nums.length - 1（0 和 nums.length - 1 也包含在内）的不同整数组成的数组。

样例 1：

输入：
	nums = [0,2,1,5,3,4]
输出：
	[0,1,2,4,5,3]
解释：
	数组 ans 构建如下：
	ans = [nums[nums[0]], nums[nums[1]], nums[nums[2]], nums[nums[3]], nums[nums[4]], nums[nums[5]]]
	    = [nums[0], nums[2], nums[1], nums[5], nums[3], nums[4]]
	    = [0,1,2,4,5,3]

样例 2：

输入：
	nums = [5,0,1,2,3,4]
输出：
	[4,5,0,1,2,3]
解释：
	数组 ans 构建如下：
	ans = [nums[nums[0]], nums[nums[1]], nums[nums[2]], nums[nums[3]], nums[nums[4]], nums[nums[5]]]
	    = [nums[5], nums[0], nums[1], nums[2], nums[3], nums[4]]
	    = [4,5,0,1,2,3]

提示：

• 1 <= nums.length <= 1000
• 0 <= nums[i] < nums.length
• nums 中的元素 互不相同

---

分析

题目描述既是答案，直接新建一个同样大小的数组，遍历一遍，按顺序赋值即可。

但是这样需要占更多内容空间，而且单纯这样做，根本算不上算法吧？

怎样能做到不占用新的内存空间呢？提示中，每个数的范围都在[0,999]之间，这是个关键。

---

题解

java

按照题意的直接处理法

class Solution {
    public int[] buildArray(int[] nums) {
        final int n = nums.length;
        
        int[] ans = new int[n];
        
        for (int i = 0; i < n; ++i) {
            ans[i] = nums[nums[i]];
        }
        
        return ans;
    }
}

不开辟新的空间，直接在原数组处理

class Solution {
    public int[] buildArray(int[] nums) {
        final int n         = nums.length;
        final int highValue = 1000;

        // 将结果值附加到数组
        for (int i = 0; i < n; ++i) {
            // 将原值+结果
            // 一个值同时存储了原值和结果值
            nums[i] += highValue * (nums[nums[i]] % highValue);
        }

        // 将原值剔除，仅保留结果
        for (int i = 0; i < n; ++i) {
            nums[i] /= highValue;
        }

        return nums;
    }
}

---

c

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* buildArray(int* nums, int numsSize, int* returnSize){
    const int highValue = 1000;

    for (int i = 0; i < numsSize; ++i) {
        nums[i] += highValue * (nums[nums[i]] % highValue);
    }

    for (int i = 0; i < numsSize; ++i) {
        nums[i] /= highValue;
    }

    *returnSize = numsSize;

    return nums;
}

---

c++

class Solution {
public:
    vector<int> buildArray(vector<int>& nums) {
        const int highValue = 1000;
        int n = nums.size();

        for (int i = 0; i < n; ++i) {
            nums[i] += highValue * (nums[nums[i]] % highValue);
        }
        
        for (int i = 0; i < n; ++i) {
            nums[i] /= highValue;
        }

        return nums;
    }
};

---

python

from typing import List

class Solution:
    def buildArray(self, nums: List[int]) -> List[int]:
        highValue = 1000
        n = len(nums)
        for i in range(n):
            nums[i] += highValue * (nums[nums[i]] % highValue)
        for i in range(n):
            nums[i] //= highValue
        return nums

---

go

func buildArray(nums []int) []int {
	highValue := 1000
	n := len(nums)

	for i := 0; i < n; i++ {
		nums[i] += highValue * (nums[nums[i]] % highValue)
	}

	for i := 0; i < n; i++ {
		nums[i] /= highValue
	}

	return nums;
}

---

rust

impl Solution {
    pub fn build_array(nums: Vec<i32>) -> Vec<i32> {
        let mut nums = nums;
        
        let high_value = 1000;
        let n = nums.len();
        
        for i in 0..n {
            nums[i] += high_value * (nums[nums[i] as usize] % high_value);
        }

        for i in 0..n {
            nums[i] /= high_value;
        }

        return nums;
    }
}

---

原题传送门

---