3379.转换数组：下标取模

【LetMeFly】3379.转换数组：下标取模

力扣题目链接：https://leetcode.cn/problems/transformed-array/

给你一个整数数组 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：

输入： nums = [3,-2,1,1]

输出： [1,1,1,3]

解释：

• 对于 nums[0] 等于 3，向右移动 3 步到 nums[3]，因此 result[0] 为 1。
• 对于 nums[1] 等于 -2，向左移动 2 步到 nums[3]，因此 result[1] 为 1。
• 对于 nums[2] 等于 1，向右移动 1 步到 nums[3]，因此 result[2] 为 1。
• 对于 nums[3] 等于 1，向右移动 1 步到 nums[0]，因此 result[3] 为 3。

示例 2：

输入： nums = [-1,4,-1]

输出： [-1,-1,4]

解释：

• 对于 nums[0] 等于 -1，向左移动 1 步到 nums[2]，因此 result[0] 为 -1。
• 对于 nums[1] 等于 4，向右移动 4 步到 nums[2]，因此 result[1] 为 -1。
• 对于 nums[2] 等于 -1，向左移动 1 步到 nums[1]，因此 result[2] 为 4。

 

提示：

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

解题方法：下标取模

题目啥意思呢？题目意思是：令$result[i] = nums[i + nums[i]]$。

但是编程语言中数组$nums$可不是循环数组，所以需要将$i+nums[i]$映射到$0\sim n-1$（对于支持负数下标的编程语言python也可以映射为负数下标）。

$t$如何映射到$0\sim n-1$？$t\ %\ n$可能为负数，再加上$n$再对$n$取模就好了。

• 时间复杂度$O(len(nums))$
• 空间复杂度$O(1)$，算法返回值不计入力扣空间复杂度

AC代码

C++

/*
 * @LastEditTime: 2026-02-15 18:11:29
 */
class Solution {
public:
    vector<int> constructTransformedArray(vector<int>& nums) {
        vector<int> ans(nums.size());
        for (int i = 0, n = nums.size(); i < n; i++) {
            ans[i] = nums[((i + nums[i]) % n + n) % n];
        }
        return ans;
    }
};

Python

'''
LastEditTime: 2026-02-15 18:13:14
'''
from typing import List

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

Java

/*
 * @LastEditTime: 2026-02-15 18:16:30
 */
class Solution {
    public int[] constructTransformedArray(int[] nums) {
        int[] ans = new int[nums.length];
        for (int i = 0, n = nums.length; i < n; i++) {
            ans[i] = nums[((nums[i] + i) % n + n) % n];
        }
        return ans;
    }
}

Go

/*
 * @LastEditTime: 2026-02-15 18:15:07
 */
package main

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

Rust

/*
 * @LastEditTime: 2026-02-15 18:23:59
 */
impl Solution {
    pub fn construct_transformed_array(nums: Vec<i32>) -> Vec<i32> {
        let n = nums.len() as i32;
        let mut ans = vec![0; n as usize];
        for i in 0..(n as usize) {
            let j = ((nums[i] + i as i32) % n + n) % n;  // 如果一路usize会无法得到负数
            ans[i] = nums[j as usize];
        }
        ans
    }
}

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

千篇源码题解已开源