leetcode周赛-3379. 转换数组

1001101_QIA2024-12-11 18:58

给你一个整数数组 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 是循环数组，向右移动超过最后一个元素时将回到开头，向左移动超过第一个元素时将回到末尾。

cpp 复制代码

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

我是fw，区分正负、处理溢出，花了很长时间，经过复盘才发现，当nums $i$ 为负数的时候，比如是-k，那么需要左移k步，那么其实就相当于右移动n-k步。

正数：result $i$ =nums $(i+(nums\[i$ %n))%n];-------->推广到：result $i$ =nums $(i+(nums\[i$ %n+n))%n];