題目:
思路:
很簡單。對於每一堆,你只需要留下最少的積木就可以了,第i堆積木最少要i個木塊纔可以保證嚴格單調遞增。如果多了,把多的往揹包裏塞;如果少了,把從揹包裏拿出來補充這個堆,若不夠補充,則失敗。
#include <iostream>
#include <vector>
using namespace std;
int main(){
int n;
int heap_num;
double block_num;
cin >> n;
for (int k = 0; k < n; ++k){
bool flag = true;
cin >> heap_num >> block_num;
vector<double> heap;
for (int i = 0; i < heap_num; ++i){
double t;
cin >> t;
heap.push_back(t);
}
for (int i = 0; i < heap.size(); ++i){
if (heap[i] == i){
continue;
}else if(heap[i] > i){
block_num += heap[i] - i;
heap[i] = i;
}else{
if (block_num >= (i - heap[i])){
block_num -= (i - heap[i]);
heap[i] = i;
}else{
flag = false;
break;
}
}
}
if (flag){
cout << "YES" <<endl;
}else{
cout << "NO" << endl;
}
}
return 0;
}