題目:
小易在維護數據的時候遇到一個需求,具體來說小易有一系列數據,這些數據了構成一個長度爲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;
}
}