每日一道演算法：旋轉陣列

zhangdeTalk發表於2020-02-05
演算法陣列

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

解法一：暴力法

思路分析：

遍歷兩次，先旋轉k次，每次將陣列旋轉一個元素（將陣列中每個元素和最後一個元素值進行交換）

PHP程式碼實現：

/**
 * @param Integer[] $nums
 * @param Integer $k
 * @return NULL
 */
function rotate(&$nums, $k) {
    for($i=0;$i<$k;$i++){
        $last = $nums[count($nums)-1];
        for($j=0;$j<count($nums);$j++){
            $temp = $nums[$j];
            $nums[$j] = $last;
            $last = $temp;
        }
    }
    return $nums;
}
使用：
$nums = [1,2,3,4,5,6,7,10];
$k = 3;
var_dump(rotate($nums,$k));

複雜度分析：

時間複雜度： O(k*n)
每個元素都被移動1步:O(n）,k次:O(k)
空間複雜度： O(1)
沒有額外空間被使用
當陣列長度大一點時，很容易超出時間限制，明顯是一個很粗暴的解決方法

解法二：反轉法

思路分析：

當我們旋轉k次後，k%n個尾部元素會移動到陣列前頭，剩餘的元素被往後移動
首先將所有陣列反轉，接著再把前k個元素反轉，最後再反轉後n-k個元素
假設 nums = [1,2,3,4,5,6,7] n=7 且 k=3
反轉所有元素：[7,6,5,4,3,2,1]
反轉前k個元素：[5,6,7,4,3,2,1]
反轉後n-k個元素：[5,6,7,1,2,3,4] 此時即為結果

PHP程式碼實現：

/**
 * @param Integer[] $nums
 * @param Integer $k
 * @return NULL
 */
function rotate(&$nums, $k) {
    $k %=count($nums);
    $nums = reverse($nums,0,count($nums)-1);
    $nums = reverse($nums,0,$k-1);
    $nums = reverse($nums,$k,count($nums)-1);
    return $nums;
}
/**
 * @param $nums
 * @param $start
 * @param $end
 * @return mixed
 */
function reverse($nums,$start,$end){
    while($start<$end){
        $temp = $nums[$start];
        $nums[$start] = $nums[$end];
        $nums[$end] = $temp;
        $start++;
        $end--;
    }
    return $nums;
}

複雜度分析：

時間複雜度： O(n)
n個元素被反轉了3次，即3*O(n)，還是O(n)
空間複雜度：O(1)
沒有額外空間被使用

解法三：環狀替換

思路分析：

 把陣列當成環形的，把每個元素放到其後k個位置
 比如[1,2,3,4,5,6,7,8]，k=2
 可以把這8個元素依次放在一個八邊形的各個頂點，從元素為1的頂點開始依次往前移動2個位置，最後的陣列[7,8,1,2,3,4,5,6]即為結果

每日一道演算法：旋轉陣列

PHP程式碼實現：

/**
 * @param Integer[] $nums
 * @param Integer $k
 * @return NULL
 */
function rotate(&$nums, $k) {
    $k = $k%count($nums);
    $count = 0;
    for($start=0;$count<count($nums);$start++){
        $current = $start;//當前位置的指標
        $prev = $nums[$start];
        do {
            $next = ($current+$k)%count($nums);
            $temp = $nums[$next];
            $nums[$next] = $prev;
            $prev = $temp;
            $current = $next;
            $count++;//控制迴圈的次數
        } while($start != $current);//開始位置和當前位置不同一個位置才進行元素移動
    }
    return $nums;
}

複雜度分析：

時間複雜度：O(n)
只遍歷了每個元素一次
空間複雜度：O(1)
使用了常數個額外空間

解法四：迴圈遍歷（k次）+雜湊函式

思路分析：

先擷取出前n-k個元素到一個新陣列，迴圈遍歷k次（從最後一個元素倒序遍歷），把每次迴圈的指標位置對應的元素插到新陣列前邊

PHP程式碼實現：

/**
 * @param Integer[] $nums
 * @param Integer $k
 * @return NULL
 */
function rotate(&$nums, $k) {
    $start = count($nums)-$k;
    $res = array_slice($nums,0,$start);
    for($i=count($nums)-1;$i>=$start;$i--){
        array_unshift($res,$nums[$i]);
    }
    return $res;
}

複雜度分析：

時間複雜度： O(n)，其中n為k
只遍歷了k次
空間複雜度： O(n)
使用了新陣列空間

解法五：雜湊函式

思路分析：

先擷取出前n-k個元素到一個新陣列，再擷取後k個元素到另外一個新陣列，最後倒序合併這兩個陣列即可

PHP程式碼實現：

/**
 * @param Integer[] $nums
 * @param Integer $k
 * @return NULL
 */
function rotate(&$nums, $k) {
    $res_start = array_slice($nums,0,count($nums)-$k);
    $res_end   = array_slice($nums,count($nums)-$k,$k);
    return array_merge($res_end,$res_start);
}

複雜度分析：

時間複雜度：O(n)
空間複雜度： O(n)
使用了新陣列空間

github

以後每次題解都會上傳到這個專案

LeetCode_PHP：https://github.com/zhangdejian/LeetCode_PHP

題目來源

力扣（LeetCode）：https://leetcode-cn.com/problems/rotate-ar...

本作品採用《CC 協議》，轉載必須註明作者和本文連結

阿德

相關文章