CF161BDiscounts

CF161B

题目大意;要购买\(n\)件物品,有\(A\)\(B\)两种类型,要求分成\(k\)组,其中如果其中一组含有\(A\)类物品,那么这一组最便宜的一件物品就会半价

怎么分组最小化代价？

我们应该尽量优惠的幅度尽量大

对于一个\(A\)类物品,假设他的价格为\(w\),那么我们绝对不会选择价值比它更小的,因为这样会让我们优惠的代价变小

我们选择贵的又对优惠的价格没有影响,所以

我们能够选择一个比较优的分组方案

先按照价格排序

把最贵的前\(k\)个A(不足\(k\)个就全部)分成一组

然后如果不够组数,就把剩下的尽量补全\(k\)组,然后把每个\(A\)类塞到小于他的最大的\(A\)的那组,如果不存在这样的\(A\)，就塞到最后一组

来保证优惠价格的最大化

#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cctype>
#include<vector>
#include<ctime>
#define LL long long
#define pii pair<int,int>
#define mk make_pair
#define fi first
#define se second
using namespace std;
const int N = 2e5 + 3;
inline int read(){
    int v = 0,c = 1;char ch = getchar();
    while(!isdigit(ch)){
        if(ch == '-') c = -1;
        ch = getchar();
    }
    while(isdigit(ch)){
        v = v * 10 + ch - 48;
        ch = getchar();
    }
    return v * c;
}
struct node{
    int ti;
    int ci; 
    int id;
}a[N];
bool book[N],used[N];
vector <int> G[N];
int n,k;
double ans;
inline bool cmp(node x,node y){
    return x.ci > y.ci;
}
inline bool cmpp(node x,node y){
    return x.id < y.id; 
}
int main(){
    int sum = 0,cnt;
    n = read(),k = read();
    for(int i = 1;i <= n;++i){
        a[i].ci = read(),a[i].ti = read(),sum += (a[i].ti == 1);
        a[i].id = i;
    }
    sort(a + 1,a + n + 1,cmp);
    int need = max(0,k - sum),now = 1;
    for(int i = 1;i <= n;++i){
        if(a[i].ti == 1){
            book[now] = 1;
            if(now < k) G[now++].push_back(a[i].id);
            else G[now].push_back(a[i].id);
        }
        else{
            if(need && now < k) G[now++].push_back(a[i].id),need--;
            else G[now].push_back(a[i].id);
        }
    }
    sort(a + 1,a + n + 1,cmpp);
    for(int i = 1;i <= now;++i){
        for(int j = 0;j < (int)G[i].size();++j){
            if(j == G[i].size() - 1 && book[i]) ans += (double)a[G[i][j]].ci * 0.5;
            else ans += a[G[i][j]].ci;
        }
    }
    printf("%.1lf\n",ans);
    for(int i = 1;i <= now;++i){
        printf("%d ",G[i].size());
        for(int j = 0;j < (int)G[i].size();++j) printf("%d ",G[i][j]);
        putchar('\n');
    }
    return 0;
}