算法,思路来自:https://blog.csdn.net/zip_fan/article/details/47340287(这篇博客的代码有小错误)
定义结构体(数组):左端,右端,原来的位置(答案所求)
对于每个样例
清空队列
清空放答案的数组
读入左端,读入右端,对结构体数组按照左端较小在前进行排序
用cnt,和答案数组,存已经邀请的人
用i存当前扫到的结构体位置
循环:
队列不为空时,
询问当前队列里,右端最小的数是否比cnt大,如果是,就放入答案,
其余的结构体仍在优先队列里。
从use往后扫结构体数组,把左端<=当前人数的结构体放入优先队列列
(队列右端较小的在堆顶)
队列为空就出循环,
输出cnt
把答案输出来,再把其余数输出来
#include<stdio.h>
#include<queue>
#include<algorithm>
using namespace std;
#define maxn 100000
struct what
{
int left,right;
int where;
friend bool operator<(what x,what y)
{
return x.right>y.right;
}
}soda[maxn+10];
bool compare(const what x,const what y)
{
return x.left<y.left;
}
int answer[maxn+10];
bool sign[maxn+10];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
priority_queue<what>q;
for(int i=1;i<=n;i++)
{
scanf("%d",&soda[i].left);
soda[i].where=i;
sign[i]=false;
}
for(int i=1;i<=n;i++)
scanf("%d",&soda[i].right);
sort(soda+1,soda+n+1,compare);
if(soda[1].left!=0)
{
printf("0\n");
for(int i=0;i<n;i++)
{
if(i!=n-1)
printf("%d ",i+1);
else
printf("%d\n",i+1);
}
continue;
}
bool start=true;
int cnt=0,i=1;
while(start==true||!q.empty())
{
//printf("1");
if(!q.empty())
{
if(q.top().right>=cnt)
{
start=false;
answer[cnt++]=q.top().where;
}
q.pop();
}
for(i=i;soda[i].left<=cnt&&i<=n;i++)
{
if(soda[i].left<=cnt)
q.push(soda[i]);
else
break;
}
}
printf("%d\n",cnt);
if(cnt==n)
{
for(i=0;i<cnt;i++)
{
if(i!=cnt-1)
printf("%d ",answer[i]);
else
printf("%d\n",answer[i]);
}
}
else
{
for(i=0;i<cnt;i++)
{
if(i!=cnt-1)
printf("%d ",answer[i]);
else
printf("%d",answer[i]);
sign[answer[i]]=true;
}
for(i=1;i<=n;i++)
{
if(sign[i]==false)
printf(" %d",i);
}
printf("\n");
}
}//
return 0;
}