Build Array from Permutation — Day 103(Python)

Photo by Sam Moqadam on Unsplash

Today, we will look into another easy problem.

1920. Build Array from Permutation

Given a zero-based permutation nums (0-indexed), build an array ans of the same length where ans[i] = nums[nums[i]] for each 0 <= i < nums.length and return it.

A zero-based permutation nums is an array of distinct integers from 0 to nums.length - 1 (inclusive).

Example 1:

Input: nums = [0,2,1,5,3,4]
Output: [0,1,2,4,5,3]
Explanation:
The array ans is built as follows:
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]

Example 2:

Input: nums = [5,0,1,2,3,4]
Output: [4,5,0,1,2,3]
Explanation: The array ans is built as follows:
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]

Constraints:

  • 1 <= nums.length <= 1000
  • 0 <= nums[i] < nums.length
  • The elements in nums are distinct.

Follow-up: Can you solve it without using an extra space (i.e., O(1) memory)?

We are given a zero-based permutation, which means in our array we have numbers from 0 to length(nums)-1. We need to return an array whose elements are positioned as nums[nums[i]].

class Solution:
def buildArray(self, nums: List[int]) -> List[int]:
output = []
for n in nums:
output.append(nums[n])
return output

Complexity

Time Complexity

Since we are just traversing through the array once, the time complexity is O(N).

Space Complexity

We are using extra space to store our result, hence space complexity is O(N).

Let us try to solve the follow-up question.

class Solution:
def buildArray(self, nums: List[int]) -> List[int]:
return( nums[nums[i]] for i in range(len(nums)))

Here we are running for-loop inside the return statement itself.

Time Complexity

Since we are just traversing through the array once, the time complexity is O(N).

Space Complexity

We are not using any extra space, and hence the space complexity will be O(1).

--

--

--

Software Engineer. Find me @ www.linkedin.com/in/annamariya-jt

Love podcasts or audiobooks? Learn on the go with our new app.

Recommended from Medium

Building Infrastructure for the Metaverse | PODE and the Player Creator Portal

How to Use Terraform with GKE: a Step-by-Step Guide to Deploying Your First Cluster

Introducing Chaos Platform 2.0 for Azure

How to Build a Car Racing Game in AR in 15 Minutes or Less

I’m Officially Vim User Now!

Create Custom Templates and Pages for WordPress

Creating Enemy AI

Exception handling : 3 ways illustrated

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Annamariya Tharayil

Annamariya Tharayil

Software Engineer. Find me @ www.linkedin.com/in/annamariya-jt

More from Medium

[LeetCode]#2078. Two Furthest Houses With Different Colors

3 Reasons For Choosing Python as Your First Programming Language

Number of Good Pairs — Day 101(Python)

Implementing Linked Lists in Python