給定一個(gè)數(shù)組,將數(shù)組中的元素向右移動(dòng) k 個(gè)位置,其中 k 是非負(fù)數(shù)。
示例 1:
輸入: [1,2,3,4,5,6,7] 和 k = 3
輸出: [5,6,7,1,2,3,4]
解釋:
向右旋轉(zhuǎn) 1 步: [7,1,2,3,4,5,6]
向右旋轉(zhuǎn) 2 步: [6,7,1,2,3,4,5]
向右旋轉(zhuǎn) 3 步: [5,6,7,1,2,3,4]
示例 2:
輸入: [-1,-100,3,99] 和 k = 2
輸出: [3,99,-1,-100]
解釋:
向右旋轉(zhuǎn) 1 步: [99,-1,-100,3]
向右旋轉(zhuǎn) 2 步: [3,99,-1,-100]
說(shuō)明:
盡可能想出更多的解決方案,至少有三種不同的方法可以解決這個(gè)問(wèn)題。
要求使用空間復(fù)雜度為 O(1) 的原地算法。
我已開始想的空間復(fù)雜度不是O(1)的算法
private static void rotateArray(int[] nums, int k) {
int arrayLength = nums.length;
if (arrayLength <= 1) {
return;
}
int j = k % arrayLength;
if (j == 0) {
return;
}
int[] newArray = Arrays.copyOf(nums, arrayLength);
for (int i = 0; i < arrayLength; i++) {
if ((j - i) > 0) {
nums[i] = newArray[arrayLength - j + i];
} else {
nums[i] = newArray[i - j];
}
}
}
空間復(fù)雜度為O(1)的算法:
public static void rotate_1(int[] nums, int k) {
int n = nums.length;
if (n <= 1) {
return;
}
k %= n;
for (int i = 0; i < k; i++) {
int temp = nums[n - 1];
for (int j = n - 1; j > 0; j--) {
nums[j] = nums[j - 1];
}
nums[0] = temp;
}
}
注意,將 k 取數(shù)組長(zhǎng)度的余數(shù),這一點(diǎn)很關(guān)鍵,因?yàn)榻o的 k 的值有可能 大于或等于數(shù)組長(zhǎng)度,這種情況也是成立的,需要考慮進(jìn)去。實(shí)際上,題目中,所給的示例的解釋步驟,其實(shí)就是算法的思路了(兩層循環(huán),第一層循環(huán)的最大次數(shù)是 k 值取數(shù)組長(zhǎng)度的余數(shù))。