1. 程式人生 > >[Leetcode]27. Remove Element

[Leetcode]27. Remove Element

使用庫函式upper,lower可以很方便的找到對應的位置刪去就好了

坑點:

1.空陣列

2.刪去的值有可能不在數組裡面 

const int x=[]{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    return 0;
}();
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        sort(nums.begin(),nums.end());
        auto s = lower_bound(nums.begin(), nums.end(),val);
        if(s == nums.end() || *s != val) return nums.size();
        auto t = upper_bound(nums.begin(), nums.end(),val);
        nums.erase(s,t);
        return nums.size();
        
    }
};