浙大PAT甲級 1080

結構體排序以及一些特殊情況的考慮。用兩個截止數組分別表示最後一位入取到該學校的兩個分數值。

如果這次遍歷與上一位的rank相同，則判斷該兩門成績是否與該學校的兩個截止數組相同，如果相同，無聊是否滿額，都繼續招人。

AC代碼：

#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<list>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 999999999
using namespace std;
int sch[100];
struct node
{
    int id;
    int ge;
    int gi;
    vector<int> v;
};
set<int> zhao[100];
int jie1[100];
int jie2[100];
node t[40000];
bool cmp(node x,node y)
{
    if(x.ge+x.gi!=y.ge+y.gi)
        return x.ge+x.gi>y.ge+y.gi;
    else
    {
        return x.ge>y.ge;
    }
}
int main()
{
   int n,m,k;
   scanf("%d %d %d",&n,&m,&k);
   for(int i=0;i<m;i++)
   {
       scanf("%d",&sch[i]);
   }
   for(int i=0;i<n;i++)
   {
       t[i].id=i;
       scanf("%d %d",&t[i].ge,&t[i].gi);
       for(int j=0;j<k;j++)
       {
           int d;
           scanf("%d",&d);
           t[i].v.push_back(d);
       }
   }
   sort(t,t+n,cmp);

   int uu=t[0].v.front();
   zhao[uu].insert(t[0].id);
   sch[uu]--;
   jie1[uu]=t[0].ge;
   jie2[uu]=t[0].gi;
   for(int i=1;i<n;i++)
   {
       int flag=0;
       if(t[i].ge==t[i-1].ge&&t[i].gi==t[i-1].gi)
       {
           flag=1;
       }
       if(flag==0)
       {
           for(int j=0;j<t[i].v.size();j++)
           {
               int tmp=t[i].v[j];
               if(sch[tmp]>0)
               {
                   zhao[tmp].insert(t[i].id);
                   jie1[tmp]=t[i].ge;
                   jie2[tmp]=t[i].gi;
                   sch[tmp]--;
                   break;
               }
           }
       }
       else
       {
            for(int j=0;j<t[i].v.size();j++)
           {
               int tmp=t[i].v[j];
               if(jie1[tmp]==t[i].ge&&jie2[tmp]==t[i].gi)
               {
                   zhao[tmp].insert(t[i].id);
                   sch[tmp]--;
                   break;
               }
               else
               {
                   if(sch[tmp]>0)
                   {
                       zhao[tmp].insert(t[i].id);
                       sch[tmp]--;
                       break;
                   }
               }
           }
       }
   }
   for(int i=0;i<m;i++)
   {
       if(zhao[i].empty())
       {
           printf("\n");
       }
       else
       {
           set<int>::iterator it=zhao[i].begin();
           printf("%d",*it);
           it++;
           for(;it!=zhao[i].end();it++)
           {
               printf(" %d",*it);
           }
           printf("\n");
       }
   }
}