leetcode 283 Move Zeros; 27 Remove Elements; 26

Posted 2021-01-25 爱学英语的程序媛

 

int arr[] = {0,1,0,3,12};
//把数组的值赋给vector
vector<int> vec(arr, arr+sizeof(arr)/sizeof(int));

解法一：

时间复杂度O(n)

空间复杂度O(1)

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int k = 0;   //nums中，[0,...k)的元素均为非0元素
        //遍历到第i个元素后，保证[0,...i)中所有非0元素
        //都按照顺序排列在[0,...k)中
        for(int i=0;i<nums.size();i++){
            if(nums[i]){
                nums[k++] = nums[i];
            }
        }
        //将nums剩余的位置放置为0
        for(int i=k;i<nums.size();i++)
            nums[i] = 0;
    }
};

解法二：将非0元素与0元素交换位置，其中k指向非零元素的位置,且为了不让两个0元素之间相互交换位置，则增加一个判断条件( i != k)

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int k = 0;   //nums中，[0,...k)的元素均为非0元素
        //遍历到第i个元素后，保证[0,...i)中所有非0元素
        //都按照顺序排列在[0,...k)中
        //同时，[k,...i]为0
        for(int i=0;i<nums.size();i++){
            if(nums[i]){
                if(i!=k)
                    swap(nums[k++] , nums[i]);
                else
                    k++;
            }
        }
    }
};

 

我用了一个比较简便的解法，使用了vector的erase()函数直接删除等于val的元素（相当于下标自动加了一，即表示的是删除元素的下一个元素），剩余的元素个数可以直接由size()得到。

需注意的：不能使用remove()的原因是：它是将等于val的元素放到vector的尾部，返回新的end()值（非val部分的end），但并不减少vector的size。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int k = 0; 
        int i = 0;
        while(i<nums.size()){
            if(nums[i] == val)
                nums.erase(nums.begin()+i);    //删除下标为i的元素
            else 
                i++;
        }
        return nums.size();
    }
};

所以要考虑上述三个问题。

 

 

 

 解法一：快慢指针。用两个指针，慢指针来记录不重复元素的个数，快指针遍历整个数组，若慢指针指向的元素不等于快指针指向的元素，则赋值。

注意：判断当数组的长度为0时返回0，否则容易出现野指针报错。

 

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.empty())
            return 0;
        int count=0;   //记录不重复元素的个数
        int j=1;   //遍历整个vector
        while(j<nums.size()){
            if(nums[count] != nums[j]){
                count++;
                nums[count] = nums[j];
            }
            j++;
        }
        return count+1;
    }
};

解法二：（24s）

和解法二的思路相似，执行时间较多，比较相邻两个元素是否相同，若相同则i++；若不相同则将nums[i]赋给nums[k]。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if (nums.empty()) return 0;
        int k = 1;
        for (int i = 1; i < nums.size(); ++i)
        {
            if (nums[i] != nums[i - 1])
            {
                 nums[k++] = nums[i];   k++的作用是最终的k是数组的长度
            }
        }
        return k;
    }
};

 

 

 思路：k记录最多重复两次的数组下标，若（ 下标为i的元素不等于k） 或者（i等于k 但是 k和k-1不相等）  则把下标为i的元素赋给k+1的元素

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.size()<=2) return nums.size();
        int k=1;   //k记录最多重复两次的数组下标
        for(int i=2;i<nums.size();i++){
            if(nums[i]!=nums[k] || (nums[i]==nums[k] && nums[k]!=nums[k-1]) )
                nums[++k] = nums[i];
        }
        return k+1;
    }
};