leetcode学习笔记(1920.基于排列构建数组)

110 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

本题考察数列相关知识,为简单题1920.基于排列构建数组

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 中的元素 互不相同

代码

class Solution:
    def buildArray(self, nums):
        n = len(nums)
        ans = []
        for i in range(n):
            ans.append(nums[nums[i]])
        return ans

解析

这个题目就是属于看起来挺难理解,如果是我自己用笔来推nums[nums[i]]会感觉很绕很不好想,稍微不注意就会出错,还会越写越烦躁,但是如果写代码来实现这个过程就会很简单的就可以得出最后的数列。这就是计算机比人类更善于处理逻辑性强的东西的一个直观体现吧,不得不承认它确实是又快又好又简单啊。
具体实现方法很简单,直接一个for循环,构造一个新的数列就可以解决了。

来源:力扣(LeetCode)
链接:leetcode.cn/problems/bu…