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 class Solution_5 {

    public void rotate(int[] nums, int k) {
        for(int i = 0; i < k; i++){
            int temp = nums[nums.length-1];
            for(int j = nums.length-1; j > 0; j--){
                nums[j] = nums[j-1];
            }
            nums[
0] = temp; } } public static void main(String[] args) { int[] nums = {1,2,3,4,5,6,7}; int k = 3; Solution_5 solution_5 = new Solution_5(); solution_5.rotate(nums, k); System.out.println(Arrays.toString(nums)); } }