解法
思路和算法
根据题意模拟，计算结果数组 $\textit{result}$ 即可。
用 $n$ 表示数组 $\textit{nums}$ 的长度。对于 $0 \le i < n$ 的每个下标 $i$，计算 $\textit{result}[i]$ 的方法如下。
-
当 $\textit{nums}[i] > 0$ 时，$\textit{result}[i]$ 的值等于数组 $\textit{nums}$ 的下标 $i$ 向右移动 $\textit{nums}[i]$ 的下标处的值，即数组 $\textit{nums}[i]$ 的下标 $i + \textit{nums}[i]$ 对应的范围 $[0, n - 1]$ 中的下标。
-
当 $\textit{nums}[i] < 0$ 时，$\textit{result}[i]$ 的值等于数组 $\textit{nums}$ 的下标 $i$ 向左移动 $-\textit{nums}[i]$ 的下标处的值，即数组 $\textit{nums}[i]$ 的下标 $i + \textit{nums}[i]$ 对应的范围 $[0, n - 1]$ 中的下标。
-
当 $\textit{nums}[i] = 0$ 时，$\textit{result}[i]$ 的值等于数组 $\textit{nums}$ 的下标 $i$ 处的值。
上述情况可以统一表示成数组 $\textit{nums}[i]$ 的下标 $i + \textit{nums}[i]$ 对应的范围 $[0, n - 1]$ 中的下标。对于 $0 \le i < n$ 的每个下标 $i$，计算 $\textit{result}[i]$ 时为了确保得到范围 $[0, n - 1]$ 中的下标，应计算 $\textit{index} = ((i + \textit{nums}[i]) \bmod n + n) \bmod n$，则 $\textit{result}[i] = \textit{nums}[\textit{index}]$。
计算数组 $\textit{result}$ 中的所有元素之后，即可得到结果数组。
代码
###Java
class Solution {
public int[] constructTransformedArray(int[] nums) {
int n = nums.length;
int[] result = new int[n];
for (int i = 0; i < n; i++) {
int index = ((i + nums[i]) % n + n) % n;
result[i] = nums[index];
}
return result;
}
}
###C#
public class Solution {
public int[] ConstructTransformedArray(int[] nums) {
int n = nums.Length;
int[] result = new int[n];
for (int i = 0; i < n; i++) {
int index = ((i + nums[i]) % n + n) % n;
result[i] = nums[index];
}
return result;
}
}
###C++
class Solution {
public:
vector<int> constructTransformedArray(vector<int>& nums) {
int n = nums.size();
vector<int> result(n);
for (int i = 0; i < n; i++) {
int index = ((i + nums[i]) % n + n) % n;
result[i] = nums[index];
}
return result;
}
};
###Python
class Solution:
def constructTransformedArray(self, nums: List[int]) -> List[int]:
n = len(nums)
return [nums[(i + nums[i]) % n] for i in range(n)]
###C
int* constructTransformedArray(int* nums, int numsSize, int* returnSize) {
int* result = (int*) malloc(sizeof(int) * numsSize);
for (int i = 0; i < numsSize; i++) {
int index = ((i + nums[i]) % numsSize + numsSize) % numsSize;
result[i] = nums[index];
}
*returnSize = numsSize;
return result;
}
###Go
func constructTransformedArray(nums []int) []int {
n := len(nums)
result := make([]int, n)
for i := 0; i < n; i++ {
index := ((i + nums[i]) % n + n) % n
result[i] = nums[index]
}
return result
}
###JavaScript
var constructTransformedArray = function(nums) {
let n = nums.length;
let result = new Array(n);
for (let i = 0; i < n; i++) {
let index = ((i + nums[i]) % n + n) % n;
result[i] = nums[index];
}
return result;
};
###TypeScript
function constructTransformedArray(nums: number[]): number[] {
let n = nums.length;
let result = new Array(n);
for (let i = 0; i < n; i++) {
let index = ((i + nums[i]) % n + n) % n;
result[i] = nums[index];
}
return result;
};
复杂度分析
登录