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;
}


没有更多推荐了,返回首页