2019牛客多校赛第六场 D.Move (枚举+multiset)

传送门

题目大意:

用k个箱子装n个大小为 a i a_i ai的物品,优先将最大满足的放入箱子中。求箱子的最小满足容量。

题目思路:

直接暴力枚举箱子的容量,看能否将所有物品放入箱子中。
可以用multiset来模拟剩余物品。(可含有重复元素的set)

se.clear()
se.insert()
se.empty()
auto se.upper_bound()
se.erase(auto)

ac代码:

#include
using namespace std;
int n,k;
int a[1005];
multiset<int>se;
bool check(int x){
    se.clear();
    for(int i=1;i<=n;i++)
        se.insert(a[i]);
    for(int i=1;i<=k;i++){
        int r=x;
        while(!se.empty()){
            auto it=se.upper_bound(r); //找到距离r最近的
            if(it!=se.begin()){
                it--;
                r-=*it;
                se.erase(it);
            }else break;
        }
    }
    if(se.size()==0) return true;
    else return false;
}

int main(){
    int T;
    scanf("%d",&T);
    for(int Case=1;Case<=T;Case++){
        scanf("%d %d",&n,&k);
        int maxs=0,sum=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            maxs=max(maxs,a[i]);
            sum+=a[i];
        }
        sum/=k;
        maxs=max(maxs,sum);
    while(!check(maxs)) maxs++;  //枚举最小容积
    printf("Case #%d: %d\n",Case,maxs);
    }

}

感谢【大佬博客】让我学到了还有multiset数据结构。

你可能感兴趣的:(训练赛)