用于对 0、1 和 2 的数组进行排序的 Javascript 程序

这是一个用于对只包含 0、1 和 2 的数组进行排序的 Javascript 程序。该程序使用了荷兰国旗问题的解决方法，即使用三个指针：一个指向当前 0 的边界 end0，一个指向当前 2 的边界 start2，一个指向当前遍历位置 i。

代码实现

function sortArray(arr) {
  let end0 = 0;
  let start2 = arr.length - 1;
  let i = 0;
  
  while (i <= start2) {
    if (arr[i] === 0) {
      swap(arr, i, end0);
      end0++;
      i++;
    } else if (arr[i] == 2) {
      swap(arr, i, start2);
      start2--;
    } else {
      i++;
    }
  }
  return arr;
}

function swap(arr, i, j) {
  let temp = arr[i];
  arr[i] = arr[j];
  arr[j] = temp;
}

解释说明

• 程序接收一个参数 arr。
• 创建三个指针，分别是 end0、start2 和 i，初始值分别是 0、arr.length-1 和 0。
• 使用 while 循环遍历数组，当 i 大于 start2 时就可以结束循环了。
• 如果 arr[i] 等于 0，就把 arr[i] 和 arr[end0] 交换，同时将 end0 和 i 分别增加 1。
• 如果 arr[i] 等于 2，就把 arr[i] 和 arr[start2] 交换，同时将 start2 减少 1。
• 如果 arr[i] 等于 1，就直接将 i 增加 1。
• 循环结束后，返回排序后的数组。

使用示例

const arr = [2, 0, 2, 1, 1, 0];
const sorted = sortArray(arr);
console.log(sorted); // [ 0, 0, 1, 1, 2, 2 ]

结束语

该程序实现了一种简单且高效的排序算法，只需要遍历一次数组就可以完成排序，时间复杂度为 O(n)。程序使用了三个指针，较为简单，容易理解。