PHP 实现颜色分类问题

PHP 实现颜色分类问题

author: he xiaodong date: 2020-06-06
颜色分类

给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意: 不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶:

  • 一个直观的解决方案是使用计数排序的两趟扫描算法。
  • 首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。 你能想出一个仅使用常数空间的一趟扫描算法吗?

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/sort-colors

解题思路

遍历数据,加以判断,然后原地交换数据。

  • 若遇到0,则与左指针指向元素交换,并将左指针右移一位
  • 若遇到2,则与右指针指向元素交换,并将右指针左移一位,因为交换过来的元素仍可能为2,所以再次对此元素进行判断
PHP 实现
  /**
    * @param Integer[] $nums
    * @return NULL
   */
    function sortColors(&$nums) {
        $left = 0;
        $right = count($nums) - 1;
        for ($i = 0; $i <= $right; $i++) {
            if ($nums[$i] === 0) {
                list($nums[$i], $nums[$left]) = [$nums[$left], $nums[$i]];
                $left++;
            }

            else if ($nums[$i] === 2) {
                list($nums[$i], $nums[$right]) = [$nums[$right], $nums[$i]];
                $right--;
                $i--;
            }
        }

        return $nums;
    }

return lengthOfLongestSubstring($s = "abcabcbb");  // output: 3

推荐买课,用右侧链接有优惠 算法面试通关40讲