sgu171 Sarov zones 贪心

    一个竞赛，有K个赛区，每个赛区固定Ni个人参赛，并且每个赛区有个出线分数Qi,现共有N名选手（SUM(Ni)==N），每名选手有一个分数P，体重W，现在问，让所有出线选手的体重和尽可能大的情况下，每个选手应该在哪个赛区参赛？

    YY了个贪心居然能过....对每名选手按体重降序排序，对每个选手找到一个Q最大的并且满足Q<Pi的赛区，若找不到就去Q最大的赛区当炮灰...然后输出答案即可,,

#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
int n,m,p,q,k;

struct node
{
    int res,q,id;
    bool operator<(const node &a) const
    {
        return q>a.q;
    }
}b[120];

struct stu
{
    int id,p,w;
}s[26000];

int ans[26000];

bool cmp(stu x,stu y)
{
    if (x.w==y.w) return x.p<y.p;
    return x.w>y.w;
}
bool can(stu ss)
{
    int j=1;
    while (j<=k && (b[j].res==0 || b[j].q>=ss.p)) j++;
    if (j>k) return false;
    b[j].res--;
    ans[ss.id]=b[j].id;
    return true;
}

int main()
{
//    freopen("in.txt","r",stdin);
    memset(ans,0,sizeof ans);
    scanf("%d",&k);
    n=0;
    for (int i=1; i<=k; i++)
    {
        scanf("%d",&b[i].res);
        n+=b[i].res;
        b[i].id=i;
    }
    for (int i=1; i<=k; i++)
    {
        scanf("%d",&b[i].q);
    }
    sort(b+1,b+1+k);
    for (int i=1; i<=n; i++)
    {
        scanf("%d",&s[i].p);
        s[i].id=i;
    }
    for (int i=1; i<=n; i++)
    {
        scanf("%d",&s[i].w);
    }
    sort(s+1,s+1+n,cmp);
    for (int i=1; i<=n; i++)
    {
        if (!can(s[i]))
        {
            int j=1;
            while (b[j].res==0 && j<=k) j++;
            b[j].res--;
            ans[s[i].id]=b[j].id;
        }
    }
    for (int i=1; i<=n; i++)
    cout<<ans[i]<<" ";
    return 0;
}