LEETCODE 27. 移除元素

class Solution {
public:
    void swap(int& i,int& j){
        int tmp=i;
        i=j;
        j=tmp;
    }
    int removeElement(vector<int>& nums, int val) {
        if(nums.size()==0){
            return 0;
        }
        if(nums.size()==1){
            if(nums[0]==val)
                return 0;
            else
                return 1;
        }
        int s=0;
        int q=0;
        while(q<nums.size()){
            if(nums[q]!=val){
                nums[s]=nums[q];
                s+=1;
            }
            q+=1;
        }
        return s;
    }
};