题目大意:
邀请n个人去徒步旅行。第i个人会接受邀请,当且仅当当前已经邀请的人数num满足:l[i]<=num<=r[i]。问最多能邀请多少个人参加,并输出邀请顺序。
分析:
当两个人的l[i]相同时,优先邀请r[i]小的,显然这样的方案是最优的。
于是,可以先将所有人按照l[i]递增排一下序,l[i]相同的,r[i]小的放前面。
然后顺序考虑每一个人,
若l[i]小于等于当前已经邀请的人数(左端满足要求),则将此人放入队列中。队列中的元素应当满足r[i]小的靠前,即优先邀请r[i]小的,因此使用优先队列。
然后考虑右端是否满足条件:即考虑队首元素的r[i]是否大于等于当前已经邀请的人数。
对于不满足的,从队列中删除掉。
若当前队列不为空,则邀请位于队首的那个人。
若队列为空或者邀请人数等于n,则结束算法。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<map>
#include<set>
#include<algorithm>
#include<queue>
#include<stack>
#include<cstdlib>
#include<cstring>
#include<vector>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
typedef long long LL;
typedef unsigned long long uLL;
typedef __int64 LI;
typedef unsigned __int64 uLI;
typedef unsigned int uI;
typedef double db;
#define maxn 100005
#define inf 0x3f3f3f3f
struct P{
int id,l,r;
bool operator < (const P &a) const
{
return r>a.r;
}
}p[maxn];
bool cmp(P a,P b){
if(a.l==b.l) return a.r<b.r;
return a.l<b.l;
}
bool vis[maxn];
int ans[maxn];
int main()
{
int t,i,cur,n;
cin>>t;
while(t--){
scanf("%d",&n);
for(i=1;i<=n;++i) {scanf("%d",&p[i].l);p[i].id=i;}
for(i=1;i<=n;++i) scanf("%d",&p[i].r);
priority_queue<P> q;
sort(p+1,p+1+n,cmp);
for(cur=0,i=1;cur<n;){
for(;i<=n&&p[i].l==cur;++i) q.push(p[i]);
while(!q.empty()&&q.top().r<cur) q.pop();
if(!q.empty()){
ans[cur++]=q.top().id;
q.pop();
}else break;
}
printf("%d\n",cur);
if(cur){
memset(vis,0,sizeof(vis));
printf("%d",ans[0]);
vis[ans[0]]=1;
for(i=1;i<cur;++i) {printf(" %d",ans[i]);vis[ans[i]]=1;}
for(i=1;i<=n;++i) if(!vis[i]) printf(" %d",i);
puts("");
}else{
for(i=1;i<n;++i) printf("%d ",i);
printf("%d\n",i);
}
}
return 0;
}