E1. Reading Books (easy version)[贪心,模拟]

$注意把这些分成三类$

$Ⅰ.Alice和Bob都喜欢的$

$Ⅱ.只有Alice喜欢的$

$Ⅲ.只有Bob喜欢的$

$那么不妨先不考虑Ⅰ这一类,因为暂时我们还难以抉择$

$从Ⅱ类选出最小的k本书,从Ⅲ类选出最小的k本书$

$如果不足k本数,那么拿最少的一些Ⅰ类来填充直到都满足有k本书$

$那么现在已经满足都有书了,如果还要减小答案$

$必然是拿当前最划算的Ⅰ类去替换Bob和Alice各自选的代价最大的书$

$划算就替换$

剩下的就是您的模拟功底了吖~~~~

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
struct p{
	int t,a,b;
}s[maxn],q[maxn],w[maxn];//分别代表一二三类别 
int top1,top2,top3,ans;
bool com(p a,p b){
	return a.t<b.t;
}
void over(){
	cout<<-1;exit(0);
}
int main()
{
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		p f;
		cin>>f.t>>f.a>>f.b;
		if(f.a&&f.b)	s[++top1]=f;
		else if(f.a)	q[++top2]=f;
		else if(f.b)	w[++top3]=f;
	}
	sort(q+1,q+1+top2,com);
	sort(w+1,w+1+top3,com);
	sort(s+1,s+1+top1,com);
	for(int i=1;i<=k;i++)	ans=ans+(q[i].t+w[i].t);//先选掉k本书(如果够) 
	top2=min(top2,k);top3=min(top3,k);//看看当前选了几本书 
	int lq=top2,lw=top3;
	int minn=min(top2,top3),be=1;
	if(minn<k)//如果没有选满,那就去填充 
	{
		if(k-minn>top1)	over();//书不够的,输出-1 
		be=k-minn+1;
		for(int i=1;i<=k-minn;i++)//至少要添加这么多
		{
			if(lq<k&&lw<k)	lq++,lw++,ans+=s[i].t;
			else if(lq<k)
			{
				lq++;
				if(top3)	ans=ans-w[top3--].t+s[i].t;	
			}	
			else if(lw<k)
			{
				lw++;
				if(top2)	ans=ans-q[top2--].t+s[i].t;
			}
		}
	}
	for(int i=be;i<=top1;i++)//每次拿最划算的去替换 
	{
		int hao=q[top2].t+w[top3].t;
		if(hao>s[i].t)
		{
			ans=ans+s[i].t-hao;
			if(top2)	top2--;
			if(top3)	top3--;
		}	
	} 
	cout<<ans;
}