1. 程式人生 > >隨筆-旋轉陣列(翻轉,迴圈交換)

隨筆-旋轉陣列(翻轉,迴圈交換)

給定一個數組,將陣列中的元素向右移動 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) 的原地演算法。

方法一:翻轉

先整體翻轉,然後需要右移幾位,翻轉幾位,最後翻轉剩餘的陣列元素!

class Solution {
    public int [] reverse(int []nums,int st,int end){
        while(st<end){
            int tmp=nums[st];
            nums[st]=nums[end];
            nums[end]=tmp;
            st++;
            end--;
        }
        return nums;
    }
    public void rotate(int[] nums, int k) {
        k=k%nums.length;
        reverse(nums,0,nums.length-1);
        reverse(nums,0,k-1);
        reverse(nums,k,nums.length-1);
        
    }
}

方法二:迴圈交換

思路:每右移一次,所有元素右移,最後一位到第一位;

class Solution {
    public void rotate(int[] nums, int k) {
        k=k%nums.length;
        while(k>0){
            int num=nums[nums.length-1];
            for(int i=nums.length-1;i>0;i--){
                nums[i]=nums[i-1];
            }
            nums[0]=num;
            k--;
        }
    }
}