LeetCode Sort Colors

最新推荐文章于 2019-04-13 19:25:13 发布

原创最新推荐文章于 2019-04-13 19:25:13 发布 · 307 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍了一种高效的颜色排序算法，该算法能在O(N)的时间复杂度内完成排序，并且仅使用常数级额外空间。通过三个指针left、i、right进行元素交换，将数组中的0、1、2分别放到正确的位置。

思路：
时间复杂度O(N)，空间复杂度O(1)。
类似荷兰国旗问题。按0，1，2分成3个区域。

c++ code:

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int left = 0;
        int right = nums.size() - 1;
        int i = 0;
        while( i <= right) {
            if(nums[i] == 0) {
                swap(nums[left], nums[i]);
                left++;
                i++;
            }else if(nums[i] == 1) {
                i++;
            }else {
                swap(nums[right], nums[i]);
                right--;
            }
        }
    }
};

java code:

public class Solution {
    private void swap(int[] nums, int i, int j) {
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }

    public void sortColors(int[] nums) {
        int left = 0, right = nums.length - 1;
        int i = 0;
        while(i <= right) {
            if(nums[i] == 0) {
                swap(nums, i, left);
                ++i;
                ++left;
            }else if(nums[i] == 1) {
                ++i;
            }else {
                swap(nums, i, right);
                --right;
            }
        }
        return;
    }
}