【第22期】观点:IT 行业加班,到底有没有价值?

[leetcode]75. Sort Colors

原创 2016年05月31日 00:11:52

题目地址

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

题目大意

一个含有n个整数数组,元素由0 1 2构成,排列随机。在O(n)时间复杂度,O(1)空间复杂度下,将其按升序排序。

解题思路

解一

首先遍历一遍,记录0,1,2个数,然后重新填充数组。

解二

使用两个标记为,a记录距离数组起始最近的非0元素,b记录距离数组尾最近的非2元素。遍历两个标记之间,遇到0与a标记元素交换,遇到2与b标记元素交换。

代码

解一

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if (nums.size() == 0) {
            return;
        }
        int red = 0;
        int white = 0;

        for (int i = 0; i != nums.size(); ++i) {
            if (nums[i] == 0) {
                ++red;
            }
            if (nums[i] == 1) {
                ++white;
            }
        }

        for (int i = 0; i != nums.size(); ++i) {
            if (red != 0) {
                nums[i] = 0;
                --red;
            } else if (white != 0) {
                nums[i] = 1;
                --white;
            } else {
                nums[i] = 2;
            }
        }
        return;
    }
};

解二

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if (nums.empty()) {
            return;
        }
        int size = nums.size();
        int start = 0;
        while (start < size && nums[start] == 0) {
            start++;
        }

        int end = size - 1;
        while (end >= 0 && nums[end] == 2) {
            end--;
        }

        int i = start;
        while (start < end && i <= end && start < size && end >= 0) {
            if (nums[i] == 1) {
                ++i;
                continue;
            }
            if (nums[i] == 0) {
                nums[i] = nums[start];
                nums[start] = 0;
                start++;
            } else if (nums[i] == 2){
                nums[i] = nums[end];
                nums[end] = 2;
                end--;
                if (nums[start] == 0) {
                    start++;
                    i++;
                }
            }
        }
    }
};
版权声明:本作品由掠雪墨影创作,采用知识共享署名 4.0 国际许可协议进行许可。转载请以链接形式标明本文地址。 举报

相关文章推荐

75 Sort Colors

75 Sort Colors链接:https://leetcode.com/tag/two-pointers/ 问题描述: Given an array with n objects colore...

[LeetCode]75.Sort Colors

【题目连接】 75.Sort Colors 【题目】 <p style="margin-top: 0px; margin-bottom:

程序员升职加薪指南!还缺一个“证”!

CSDN出品,立即查看!

[LeetCode]75.Sort Colors

【题目】 Given an array with n objects colored red, white or blue, sort them so that objects of the sa...

[leetcode]Sort Colors

新博文地址:<a id="cb_post_title_url" style="font-size: 1.5em; line-height: 1.5em; font-family: Verdana, Geneva, Arial, Helvetica, sans-se...

[leetcode-排序]--75. Sort Colors

Question 75. Sort Colors Given an array with n objects colored red, white or blue, sort them so th...
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)