3379. 转换数组

English Version

题目描述

给你一个整数数组 nums，它表示一个循环数组。请你遵循以下规则创建一个大小 相同 的新数组 result ：

对于每个下标 i（其中 0 <= i < nums.length），独立执行以下操作：

• 如果 nums[i] > 0：从下标 i 开始，向 右 移动 nums[i] 步，在循环数组中落脚的下标对应的值赋给 result[i]。
• 如果 nums[i] < 0：从下标 i 开始，向 左 移动 abs(nums[i]) 步，在循环数组中落脚的下标对应的值赋给 result[i]。
• 如果 nums[i] == 0：将 nums[i] 的值赋给 result[i]。

返回新数组 result。

注意：由于 nums 是循环数组，向右移动超过最后一个元素时将回到开头，向左移动超过第一个元素时将回到末尾。

 

示例 1：

示例 2：

 

提示：

• 1 <= nums.length <= 100
• -100 <= nums[i] <= 100

解法

方法一：模拟

思考

从 $i$ 出发按 $nums[i]$ 的符号移动 $|nums[i]|$ 步（环形），写入落点的值。$n \le 100$，直接算下标。

步数对 $n$ 取模时注意负数：$(i + x \bmod n + n) \bmod n$ 即可。

结果数组与原数组分离，避免覆盖尚未使用的 $nums[i]$。

我们创建一个结果数组 $\textit{ans}$，对于每个下标，根据 $nums[i]$ 的正负向右或向左移动 $|nums[i]|$ 步，计算出落脚的下标，并将该下标对应的值赋给 $\textit{ans}[i]$。

时间复杂度 $O(n)$，空间复杂度 $O(n)$。其中 $n$ 是数组 $nums$ 的长度。

Python3

class Solution:
    def constructTransformedArray(self, nums: List[int]) -> List[int]:
        n = len(nums)
        return [nums[(i + x % n + n) % n] for i, x in enumerate(nums)]

Java

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

C++

class Solution {
public:
    vector<int> constructTransformedArray(vector<int>& nums) {
        int n = nums.size();
        vector<int> ans(n);
        for (int i = 0; i < n; ++i) {
            ans[i] = nums[(i + nums[i] % n + n) % n];
        }
        return ans;
    }
};

Go

func constructTransformedArray(nums []int) []int {
	n := len(nums)
	ans := make([]int, n)
	for i, x := range nums {
		ans[i] = nums[(i+x%n+n)%n]
	}
	return ans
}

TypeScript

function constructTransformedArray(nums: number[]): number[] {
    const n = nums.length;
    const ans: number[] = [];
    for (let i = 0; i < n; ++i) {
        ans.push(nums[(i + (nums[i] % n) + n) % n]);
    }
    return ans;
}

Rust

impl Solution {
    pub fn construct_transformed_array(nums: Vec<i32>) -> Vec<i32> {
        let n = nums.len() as i32;
        let mut ans = vec![0; nums.len()];
        for (i, &x) in nums.iter().enumerate() {
            ans[i] = nums[(((i as i32 + x % n + n) % n) as usize)];
        }
        ans
    }
}