LeetCode--初級演算法--陣列篇--第三題--旋轉陣列
阿新 • • 發佈:2019-01-23
題目:
給定一個數組,將陣列中的元素向右移動 k 個位置,其中 k 是非負數。
示例
示例 1:
輸入: [1,2,3,4,5,6,7] 和 k = 3
輸出: [5,6,7,1,2,3,4]
解釋:
向右旋轉 1 步: [7,1,2,3,4,5,6]
向右旋轉 2 步: [6,7,1,2,3,4,5]
向右旋轉 3 步: [5,6,7,1,2,3,4]
示例 2:
輸入: [-1,-100,3,99] 和 k = 2
輸出: [3,99,-1,-100]
解釋:
向右旋轉 1 步: [99,-1,-100,3]
向右旋轉 2 步: [3,99,-1,-100]
說明:
儘可能想出更多的解決方案,至少有三種不同的方法可以解決這個問題。
要求使用空間複雜度為 O(1) 的原地演算法。
程式碼:
方法一:
public static void rotate(int[] nums, int k) {
if (nums == null || k == 0 || k == nums.length) {
return;
}
int temp = 0;
while ( k > 0) {
for (int i = nums.length - 1 ;i >= 0 ; i--) {
temp = nums[i];
nums[i] = nums[i - 1];
nums[i-1] = temp;
}
k --;
}
}
好吧,寫到這,對於一個從來沒有看過演算法,而且數學還不好的我來說,已經是極限了,結果尷尬的是提交LeetCode程式碼時,居然超時不通過。尷尬啊,然後就在網上搜大佬的想法。
方法二:
下面這中和我想法差不多,就是比我的更加優化一點
public static void rotate(int[] nums, int k) {
if (nums == null || k == 0 || k == nums.length) {
return;
}
int temp = 0;
while ( k > 0) {
temp = nums[nums.length - 1];
for (int i = nums.length - 2;i >= 0 ; i--) {
nums[i + 1] = nums[i];
}
nums[0] = temp;
k --;
}
}
方法三:
這個方法我就有點看不懂啦(自己好菜),取餘的方法,我還是懵懵了,雖然能照著敲出來,但是還沒有學到思想。
public static void rotate(int[] nums, int k) {
if (nums == null || k == 0 || k == nums.length) {
return;
}
k = k % nums.length;
reverse(nums, 0, nums.length - k - 1);
reverse(nums, nums.length - k, nums.length - 1);
reverse(nums, 0, nums.length - 1);
}
private static void reverse(int[] nums, int start, int end) {
while (start < end) {
int temp = nums[start];
nums[start] = nums[end];
nums[end] = temp;
start++;
end--;
}
}