sgu 171 分类： sgu 2015-03-08 09:52 33人阅读 评论(0) 收藏

题意：
有K个城市，第i城市至多有N[i]个人，每个城市有一个属性Q[i]。
对于N=∑N[i]个人，每个人有一个属性P[i]和价值W[i]，把第i个人放进第j个城市中，当且仅当P[i]>Q[j]时，可以获得W[i]的价值，否则不获得价值。
求出满足价值和最大的人数分配方案。0<=N<=16000。

---

以上为Owaski的翻译

贪心，先放置权值大的学生，放置到满足Pi>Qj并且(Pi−Qj)最小的城市，
若没有这样的城市存在，就在其他学生放置完后，再任取一个城市放置。

可以用反证法证明这个策略是最优的。

---

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<iostream>
#include<algorithm>

#define INF 1<<30

const int MAXK = 105 ,MAXN = 16005;

int k , n;
int N[MAXK] = {0} , Q[MAXK] = {0};

struct person{int p , w ,num;}mp[MAXN];

bool cmp(const person &a,const person &b)
{
    return (a.w > b.w);
}

int site[MAXN] = {0};

bool hash[MAXN] = {false};

int main()
{
#ifndef ONLINE_JUDGE
    freopen("sgu171.in","r",stdin);
    freopen("sgu171.out","w",stdout);
#endif  

    scanf("%d",&k);

    n = 0;
    for(int i = 1; i <= k; i++){scanf("%d",&N[i]);n += N[i];}
    for(int i = 1; i <= k; i++)scanf("%d",&Q[i]);
    for(int i = 1; i <= n; i++)scanf("%d",&mp[i].p);     
    for(int i = 1; i <= n; i++)scanf("%d",&mp[i].w);
    for(int i = 1; i <= n; i++)mp[i].num = i;

    std::sort(mp + 1, mp + n + 1, cmp);

    for(int i = 1; i <= n; i++)
    {
        int t = 0 , min = INF;

        for(int j = 1; j <= k; j++)
        {
            if(N[j] && mp[i].p > Q[j])
            {
                int tmp = mp[i].p - Q[j];

                if(tmp < min) min = tmp , t = j;
            }
        }

        if( t ) 
        {
            int x = mp[i].num;

            hash[x] = true;
            site[x] = t;
            N[t] --;
        }
    }

    for(int i = 1; i <= n ;i++)
     if(hash[i] == false)
     {
        for(int j = 1; j <= k; j++)
          if(N[j]){hash[i] = true , site[i] = j, N[j] --;break;}
     }

    for(int i = 1; i <= n; i++)
       printf("%d ",site[i]);


#ifndef ONLINE_JUDGE    
    fclose(stdin);
    fclose(stdout);
#endif  
}

版权声明：本文为博主原创文章，未经博主允许不得转载。