對數組中的數字 1 和 2 進行排序，使得數字 一、2 分別位於前、後部分

問題描述：假設某個數組中只有數字 1 和 2，進行排序，使得數字 1 位於數組前部分，數字 2 位於後部分。

這道算法題其實不是很難，使用各類排序算法應該都能解出，可是若要考慮性能問題，那就得選擇一種算法複雜度最低的解法。這裏我使用雙指針的方法來解答該題，時間複雜度爲 O(n)。

1. 解法步驟

（1）設置一個頭指針、一個尾指針，頭指針首先指向數組的第一個元素（索引爲 0），而尾指針則指向數組的最後一個元素（索引爲 len - 1，假定數組的長度爲 len）；

（2）而後比較這兩個一前一後元素的大小；

• 若兩值不相等，則較小的 1 放在前面，較大的 2 放在後面，頭指針日後移動一步，尾指針向前移動一步；
• 若兩值相等且都等於 1，則頭指針日後移動一步，尾指針不移動；
• 若兩值相等且都等於 2，則尾指針往前移動一步，頭指針不移動；

（3）接着再次比較頭、尾指針指向元素的大小，決定是否交換值以及移動指針；

（4）依照以上步驟進行指針移動、元素大小比較，即可使得數字 1 位於數組前部分，數字 2 位於數組後部分。

2. 注意點：上面循環進行操做的條件是頭指針索引值小於尾指針索引值。

3. 書寫的代碼以下：

function sortOneTwoInArr (arr) {
  var len = arr.length;
  var head = 0;
  var tail = len - 1;
  /* 遍歷數組，對 1 和 2 進行排序 */
  while (head < tail) {
    // 若頭、尾指針指向的元素大小相等則只移
    // 動一個指針，不然同時移動兩個指針
    if (arr[head] === arr[tail]) {
      if (arr[head] === 1) {
        head++;
      } else if (arr[head] === 2) {
        tail--;
      }
    } else {
      if (arr[head] > arr[tail]) {
        [arr[head], arr[tail]] = [arr[tail], arr[head]];
      }
      head++;
      tail--;
    }
  }
  return arr;
}

/* 測試用例 */
var arr1 = [];
var arr2 = [1];
var arr3 = [2];
var arr4 = [1, 2, 1, 2];
var arr5 = [1, 1, 2, 2];
var arr6 = [1, 2, 2, 1, 1];
var arr7 = [2, 2, 1, 1, 2];
console.log(sortOneTwoInArr(arr6));         // [1, 1, 1, 2, 2]
複製代碼