1. 程式人生 > >LeetCode--初級演算法--陣列篇--第三題--旋轉陣列

LeetCode--初級演算法--陣列篇--第三題--旋轉陣列

題目:

給定一個數組,將陣列中的元素向右移動 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--;
        }
    }