CodeForces 332C Students' Revenge

题目

题意

给出n对数对(a,b)，A从其中选出p个数对，B再从中A给他的数对中挑出k个数对，B的方案为：首先满足k个数对的b之和最大，如果有多组方案，再考虑a之和最小。现在问A应该从中选出哪p个数对，使得在B的方案下得到的a之和最大，若有多组方案，再考虑b之和最大

题解

很容易想到贪心，但是题目的限制条件很多，所以坑也多，很容易满足一个条件又不满足另一个了。

1）在B的方案下，B首先会考虑不选b最小的那p-k个数对，所以先按照b从小到大排序，标记出这p-k个数对

2）那么A肯定会在剩下的数对中间选出a最大的那k个数对，按照a从大到小排序，因为他们的b都比较大，所以B肯定会选择这些数对，这样就满足了a之和最大的要求。

3）现在要在剩下的n-k个数中间选出不影响a之和的最大值且b最大的k-p个数对了。设第2)步中选出的数对的b的最小值为bmin，其实就是再选出p-k个b不大于bmin的数即可，因为之前按照b从小到大排过序了，只需记录一下第2)步中的输出的数对的最小下标就可以了。

坑：test 3：没有考虑a之和取到最大值有多种方案的话，选b之和最大的方案

        test 8：a和b全部为1的情况

        test 6：同test 3

        test 31和test 29：巨坑，我第3)步的方法是，再按b排一次序，但是这样的出来的数组可能和第1)步排出来的顺序不一样！！！

代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 100005

int n,p,k;
struct node
{
int id,a,b,pos;
}o[maxn],tt[maxn];
bool cmp1(node x,node y)
{
if (x.b==y.b) return x.a>y.a;
return x.b<y.b;
};
bool cmp2(node x,node y)
{
if (x.a==y.a) return x.b>y.b;
return x.a>y.a;
};
int main()
{
//freopen("/home/christinass/code/in.txt","r",stdin);
scanf("%d%d%d",&n,&p,&k);
for (int i=0;i<n;i++)
{
scanf("%d%d",&o[i].a,&o[i].b);
o[i].id=i+1;
}
sort(o,o+n,cmp1);
for (int i=0;i<n;i++) o[i].pos=i;
for (int i=0;i<n;i++) tt[i].a=o[i].a,tt[i].b=o[i].b,tt[i].id=o[i].id;
sort(o+p-k,o+n,cmp2);
int t=n;
for (int i=p-k;i<=p-1;i++)
{
printf("%d ",o[i].id);
t=min(o[i].pos,t);
}
for (int i=1;i<=p-k;i++)
printf("%d ",tt[t-i].id);
printf("\n");
return 0;
}