LeetCode. 75. 颜色分类（荷兰国旗问题/双指针）

最新推荐文章于 2023-02-02 21:15:58 发布

xylitolz

最新推荐文章于 2023-02-02 21:15:58 发布

阅读量795

点赞数

分类专栏： LeetCode 文章标签： leetcode 排序算法算法 java

本文链接：https://blog.csdn.net/xylitolz/article/details/124611623

版权

该博客介绍了如何使用单指针和双指针方法解决75号LeetCode问题——颜色分类。在单指针方法中，通过两次遍历将数组分为0、1和2的三个部分；双指针方法则通过一次遍历，同时交换0和2，最终实现数组的排序。两种方法的时间复杂度均为O(n)，空间复杂度为O(1)。

摘要由CSDN通过智能技术生成

题目

75. 颜色分类

在这里插入图片描述

方法一：单指针

本题是经典的「荷兰国旗问题」，由计算机科学家 Edsger W. Dijkstra 首先提出

考虑对数组进行两次遍历。在第一次遍历中，将数组中所有的 0 交换到数组的头部。在第二次遍历中，将数组中所有的 1 交换到头部的 0 之后。此时，所有的 2 都出现在数组的尾部，这样就完成了排序。

算法流程：

使用一个指针 $\textit{ptr}$ 表示「头部」的范围，ptr 中存储了一个整数，表示数组 nums 的[0,ptr-1]都属于「头部」。ptr 的初始值为 0，表示还没有数处于「头部」
第一次遍历：
- 从左向右遍历整个数组，如果找到了 0，那么就需要将 0 与「头部」位置ptr的元素进行交换，并将「头部」向后扩充一个位置(即ptr++)。在遍历结束之后，所有的 0 都被交换到「头部」的范围，并且「头部」只包含 0
第二次遍历：
- 从「头部」开始，从左向右遍历整个数组，如果找到了 1，那么就需要将 1 与「头部」位置的元素进行交换，并将「头部」向后扩充一个位置。在遍历结束之后，所有的 1 都被交换到「头部」的范围，并且都在 0 之后，此时 2 只出现在「头部」之外的位置，因此排序完成

class Solution {
   
    public void sortColors(int[] nums) {
   
        int n = nums.length;
        if (n == 1) {
   
            return ;
        }
        int ptr = 0;
        // 第一次遍历
        for (int i = 0; i < n; i++) {
   
            if (nums[i] == 0) {
   
                swap(nums, ptr, i)