万万哇 发表于 2025-4-15 11:13:16

【力扣hot100题】(097)颜色分类

https://i-blog.csdnimg.cn/direct/d1fe1c840ba64a849033a7d5802dcb03.png
第一眼感觉有点像确定不同元素个数的排序,但是要保证时间复杂度不超过o(n),即只扫描一遍。
没思路于是看批评区,看到一种很快速的做法,虽然面试不保证通过就是了……
class Solution {
public:
    void sortColors(vector<int>& nums) {
      int count0=0;
      int count1=0;
      int count2=0;
      for(int i=0;i<nums.size();i++){
            if(nums==0) count0++;
            else if(nums==1) count1++;
            else count2++;
      }
      for(int i=0;i<nums.size();i++){
            if(count0-->0) nums=0;
            else if(count1-->0) nums=1;
            else nums=2;
      }
    }
}; 但想法还是很有意思的,记录一下。
之后还是没思路,于是看了答案……
答案是使用指针法,设置指针指向前面0元素的末端,遍历数组时发现有0元素就与指针互换,然后指针后移一位。
按同样方法使所有2排到末了去。
如许要遍历两边数组,但是很通俗易懂。
class Solution {
public:
    void sortColors(vector<int>& nums) {
      int ptr=0;
      for(int i=0;i<nums.size();i++){
            if(nums==0){
                swap(nums,nums);
                ptr++;
            }
      }
      ptr=nums.size()-1;
      for(int i=nums.size()-1;i>=0;i--){
            if(nums==2){
                swap(nums,nums);
                ptr--;
            }
      }
    }
}; 于是答案又给出了双指针的进阶做法,只遍历一遍,然后每次查看是否为0和2,互换就行。
这里我遇到了一点困难,不知道为什么会出错,这是我一开始的错误解法:
class Solution {
public:
    void sortColors(vector<int>& nums) {
      int ptr0=0;
      int ptr2=nums.size()-1;
      for(int i=0;i<nums.size();i++){
            if(nums==0){
                swap(nums,nums);
                ptr0++;
            }
            else if(nums==2){
                swap(nums,nums);
                ptr2--;
            }
            if(ptr2==i) return ;
      }
    }
}; 后来想想终于想到了为什么出错,由于单指针时0是从前往后遍历,2是从后往前遍历,这里改成双指针是都是从前往后遍历,会出现互换的两个元素都是2,互换完后直接跳过了如许的错误。
然后我决定用0和1做指针。
有点难些由于遇到0的时候也要附带互换1的指针,以是遇到0的时候互换后要把互换出去的1换回来(如果0指针在1的后面也就是有1元素)。
class Solution {
public:
    void sortColors(vector<int>& nums) {
      int ptr0=0;
      int ptr1=0;
      for(int i=0;i<nums.size();i++){
            if(nums==0){
                swap(nums,nums);
                if(ptr0<ptr1) swap(nums,nums);
                ptr0++;
                ptr1++;
            }
            else if(nums==1){
                swap(nums,nums);
                ptr1++;
            }
      }
    }
};

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: 【力扣hot100题】(097)颜色分类