3379. 转换数组

题目描述

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

解法

方法一

Python3JavaC++GoTypeScript

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

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] != 0 ? nums[(i + nums[i] % n + n) % n] : 0;
        }
        return ans;
    }
}

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 + nums[i] % n + n) % n] : 0;
        }
        return ans;
    }
};

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

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 + (nums[i] % n) + n) % n] : 0);
    }
    return ans;
}

3379. 转换数组

题目描述

解法

方法一

评论