leetcode Counting Sort

leetcode 75题

Posted by franki on August 22, 2021

leetcode 75题 颜色分类

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

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

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

思路

计数排序

代码

var sortColors = function(nums) {
    // 计数排序
    if (nums.length === 0) {
        return;
    }
    let r = 0, w = 0, b = 0;
    for (const num of nums) {
        if (num === 0) {
            nums[b] = 2;
            b++;
            nums[w] = 1;
            w++;
            nums[r] = 0;
            r++; 
        } else if (num === 1) {
            nums[b] = 2;
            b++;
            nums[w] = 1;
            w++;
        } else if (num === 2) {
            b++;
        }
    }
};

复杂度分析

  • 时间复杂度：O(N)
  • 空间复杂度：O(1)
CATALOG
    FEATURED TAGS
    前端开发 H5 JavaScript 设计模式 browser jQuery 源码分析 生活 leetcode Array Stack Queue Linked List 剑指offer Binary Search Tree Binary Tree Breadth-First Search Depth-First Search String Set Binary Search Sliding Window Backtracking Dynamic Programming Two Pointers Union Find Sort Bit Operation Recursion Map Graph Search Hash LinkedList QuickSort Trie Design MinHeap Traverse Min Heap Node.js BackEnd SQL MySQL Design Patterns Network 计算机网络 Python
    FRIENDS