Leetcode 75 Sort colors

xxxmmc2024-11-14 9:46

题意：荷兰国旗问题，给一个数组[0,0,2,1,0]，构造成[0,0,0,1,2]的形式，分成三块

https://leetcode.com/problems/sort-colors/description/

题解：

在任意时刻，i 左边的数都是 0，k 右边的数都是 2，而 i 到 j 之间的数都是 1。

想象有三个指针， i , j , k i, j, k i,j,k 维护 [ 0 , i ) [0,i) [0,i)为0，维护$[i,j)为1，[j, nums.size()]为2

想象有三个指针在动，i代表起始位置，k代表末尾位置，j遍历整个数组，移动j，当j的值指向的数字为0，的时候那么跟i交换，移动的过程中j >=i

cpp 复制代码

class Solution {
public:
    void sortColors(vector<int>& nums) {
        for(int i = 0, j = 0, k = nums.size()-1; k >= j;) {
            if(!nums[j]) {
                swap(nums[i++], nums[j++]);
            } else if( nums[j] == 2) {
                swap(nums[j],nums[k--]);
            } else j++;
        }
    }
};

时间复杂度 O ( n ) O(n) O(n)

空间复杂度 O ( 1 ) O(1) O(1)