CF161B Discounts

题目传送门

题意

你要去买 $n$ 个东西，你有 $k$ 个购物车，然而买的东西有两种，为凳子和铅笔，如果这个购物车里面有凳子，那么里面最便宜的物品打五折。

解题方法

我们的目标就是让打折的钱数尽量多，然而我们发现如果有凳子，打折的价格只会是价格小于等于这个凳子价格的，那么我们就把前 $k-1$大的凳子单独拿出来打折，剩下的放在最后的购物车里面，当然如果凳子数小于等于$k-1$，把后面的慢慢放满就可以了

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct node
{
	int num,pos;
};
node a[20001];
node b[20001];
struct trans
{
	int tot;
	int s[2001];
}ans[2001];
int tot1,tot2;
bool comp(node x,node y)
{
	return x.num>y.num;
}
int main()
{
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		int x,zkt;
		scanf("%d%d",&x,&zkt);
		if(zkt==1)
		{
			tot1++;
			a[tot1].num=x;
			a[tot1].pos=i;
		}
		else
		{
			tot2++;
			b[tot2].num=x;
			b[tot2].pos=i;
		}
	}
	sort(a+1,a+tot1+1,comp);
	double anstot=0;
	for(int i=1;i<=min(k-1,tot1);i++)
	{
		ans[i].tot++;
		ans[i].s[ans[i].tot]=a[i].pos;
		anstot+=a[i].num*0.5;
	}
	if(tot1<=k-1)
	{
		for(int i=1;i<=tot2;i++)
		{
			anstot+=b[i].num;
			ans[min(tot1+i,k)].tot++;
			ans[min(tot1+i,k)].s[ans[min(tot1+i,k)].tot]=b[i].pos;
		}
	}
	else
	{
		int minn=1e9;
		for(int i=k;i<=tot1;i++)
		{
			anstot+=a[i].num;
			ans[k].tot++;
			ans[k].s[ans[k].tot]=a[i].pos;
			minn=min(a[i].num,minn);
		}
		for(int i=1;i<=tot2;i++)
		{
			anstot+=b[i].num;
			ans[k].tot++;
			ans[k].s[ans[k].tot]=b[i].pos;
			minn=min(b[i].num,minn);
		}
		anstot-=minn*0.5;
	}
	printf("%.1lf\n",anstot);
	for(int i=1;i<=k;i++,puts(""))
	{
		cout<<ans[i].tot<<" ";
		for(int j=1;j<=ans[i].tot;j++)
		  cout<<ans[i].s[j]<<" ";
	}
	return 0;
}