摆动序列cpp

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        vector<int> tmp;
        int flag = 2;
        for(int i = 0; i < nums.size(); i ++){
            if(tmp.empty()){
                tmp.push_back(nums[i]);
            }else{
                if(nums[i] > tmp.back()){
                    if(flag == 1){
                        tmp.pop_back();
                    }
                    tmp.push_back(nums[i]);
                    flag = 1;
                } else if(nums[i] < tmp.back()){
                    if(flag == 0){
                        tmp.pop_back();
                    }
                    tmp.push_back(nums[i]);
                    flag = 0;
                }
            }
        }
        return tmp.size();
    }
};