题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5360
题意:要去旅游有n个人,第i个人接受邀请最少要Li个人已接受邀请,且不能超过Ri个人已接受,问最优邀请方案使得最多人去
思路:很容易想到贪心,从满足Li条件的人中选取Ri最小的人邀请,可以每次将满足条件的人进入优先队列,每次去队首就可以了
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
struct Node
{
int l,r,pos;
bool operator <(const Node &q) const
{
return r>q.r;
}
}s[100030];
int vis[100030],ans[100030];
bool cmp(Node p,Node q)
{
return p.l<q.l;
}
priority_queue<Node> que;
int main()
{
int n,t;
scanf("%d",&t);
while (t--)
{
scanf("%d",&n);
memset(vis,0,sizeof(vis));
for (int i=0;i<n;i++)
{
scanf("%d",&s[i].l);
s[i].pos=i+1;
}
for (int i=0;i<n;i++)
{
scanf("%d",&s[i].r);
}
sort(s,s+n,cmp);
int now=0,pos=0,cnt=0;
while (pos<n)
{//cout<<":";
if (s[pos].l<=now)
que.push(s[pos++]);
else
break;
}
while (true)
{
while (!que.empty())
{
Node tmp=que.top();
que.pop();
if (tmp.r>=now)
{
vis[tmp.pos]=1;
ans[cnt++]=tmp.pos;
now++;
break;
}
}
while (pos<n)
{
if (s[pos].l<=now)
que.push(s[pos++]);
else
break;
}
if (que.empty()) break;
}
printf("%d\n",cnt);
int flag=1;
for (int i=0;i<cnt;i++)
{
if (!flag) printf(" ");
flag=0;
printf("%d",ans[i]);
}
for (int i=1;i<=n;i++)
{
if (!vis[i])
{
if (!flag) printf(" ");
flag=0;
printf("%d",i);
}
}
printf("\n");
}
}