【網易編程題】序列維護

題目:
小易在維護數據的時候遇到一個需求,具體來說小易有一系列數據,這些數據了構成一個長度爲n的數字序列,接下來小易會在這個序列上進行q次操作。
每次操作有一個查詢的數字x,小易需要將序列數據中所有大於等於x的數字都減一,並輸出在本次操作中有多少個數字被減一了。
小易犯了難,希望你能幫幫他。


思路:
這個題本身並不難,但是容易超時。稍微對數組處理下,即對數組排下序,比較失敗後提前break跳出循環就不會超時了。


#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int cmp(const int &a, const int &b){
    return a > b;
}

int main(){
    vector<int> vec;
    int n, q;
    cin >> n;
    cin >> q;
    
    for (int i = 0; i < n; ++i){
        int t;
        cin >> t;
        vec.push_back(t);
    }
    sort(vec.begin(), vec.end(), &cmp);
    for (int i = 0; i < q; ++i){
        int x;
        cin >> x;
        int count = 0;
        for (int j = 0; j < vec.size(); ++j){
            if (vec[j] >= x){
                --vec[j];
                ++count;
            }else{
                break; // 比較失敗提前跳出去
            }
        }
        cout << count << endl;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章