【網易編程題】小易希望把這些個積木變成嚴格遞增的

題目:
在這裏插入圖片描述


思路:
很簡單。對於每一堆,你只需要留下最少的積木就可以了,第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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章