题目描述
给你一个整数数组 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
是循环数组,向右移动超过最后一个元素时将回到开头,向左移动超过第一个元素时将回到末尾。
样例
输入: 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。
输入: 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
算法
(模拟) $O(n)$
- 按题目描述模拟,三种情况可以用一个公式来计算目标的下标。
时间复杂度
- 遍历原数组一次,故时间复杂度为 $O(n)$。
空间复杂度
- 需要 $O(n)$ 的额外空间存储答案。
C++ 代码
class Solution {
public:
vector<int> constructTransformedArray(vector<int>& nums) {
const 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;
}
};