【BZOJ1178】会议中心(APIO2009)-贪心+倍增+set
测试地址:会议中心
做法:本题需要用到贪心+倍增+set。
如果这题单纯只求最多的线段数,相信大家都会贪心求了,但是鉴于我太弱今天才会,这里还是写一下吧。
把所有线段按左端点从小到大排序,按右端点为第二关键字从大到小排序,我们首先维护一个栈,是为了把所有包含其它线段的线段删掉,因为选这些线段一定不优。删完之后,这些线段的左右端点都是单调的了。那么我们贪心从小到大选择,一旦能选某条线段就马上选,因为单调性的缘故,这个算法一定是对的。
那么这题要求字典序最小的解,怎么办呢?按照套路,我们按字典序从小到大试验线段,如果选了这条线段后还能求出最优解,就选这条线段,这样选出来的解一定是最优且字典序最小的。
问题来了,我们要如何判断选了某些线段之后还能不能求出最优解呢?注意到我们要实现一个函数,表示所有线段的端点都在区间之间的最优解,那么我们每次往解中添加一条线段,令这条线段为,而它在我们已经选出的线段中的前驱右端点为,后继左端点为,于是如果,就表示可以选这条线段。前驱和后继可以用set维护,那么现在的问题就是如何实现这个函数。
注意到解如果要最优,那么选的线段一定不能是能包含其它线段的线段,所以其实就是在我们一开始所述的贪心算法中的那些线段里选。而注意到我们贪心的过程,我们发现选了一条线段后,它之后紧接着选的线段是一定的,于是我们就可以用倍增来加速这个贪心的过程,即令为:如果一开始选了线段,那么贪心过程中取到的第条线段是哪条。那么显然有:
于是上面的函数就很好实现了,其实就是从第一条左端点大于等于的线段开始取,取的线段右端点不能超过的贪心过程,用倍增加速这一过程即可。
那么我们就做完了这一题,时间复杂度为。
以下是本人代码:
#include <bits/stdc++.h>
using namespace std;
const int inf=1000000100;
int n,st[200010],top,pos[200010],ans[200010];
int f[200010][25]={0},maxf;
struct interv
{
int l,r,id;
bool operator < (interv a) const
{
return l<a.l;
}
}q[200010];
set<interv> S;
set<interv>::iterator it;
bool cmp(interv a,interv b)
{
if (a.l!=b.l) return a.l<b.l;
else return a.r>b.r;
}
void init()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q+1,q+n+1,cmp);
top=0;
for(int i=1;i<=n;i++)
pos[q[i].id]=i;
for(int i=1;i<=n;i++)
{
while(top&&q[st[top]].r>q[i].r) top--;
st[++top]=i;
}
int now=1,nowr=0;
maxf=0;
for(int i=1;i<=top;i++)
{
f[i][0]=i;
while(q[st[now]].r<q[st[i]].l) f[now][1]=i,now++;
if (q[st[i]].l>nowr)
{
maxf++;
nowr=q[st[i]].r;
}
}
for(int i=2;i<=20;i++)
for(int j=1;j<=top;j++)
f[j][i]=f[f[f[j][i-1]][1]][i-1];
}
int calc(int x,int y)
{
if (x>y) return 0;
int l=1,r=top;
while(l<r)
{
int mid=(l+r)>>1;
if (q[st[mid]].l<x) l=mid+1;
else r=mid;
}
if (q[st[l]].l<x) return 0;
int tot=0;
for(int i=20;i>=0;i--)
if (f[l][i]&&q[st[f[l][i]]].r<=y)
{
tot+=(1<<i);
l=f[f[l][i]][1];
}
return tot;
}
void work()
{
int cnt=0;
for(int i=1;i<=n;i++)
{
int l=q[pos[i]].l,r=q[pos[i]].r,tot=0,L,R;
it=S.lower_bound(q[pos[i]]);
if (it!=S.end()) L=(*it).l;
else L=inf;
if (L<=r) continue;
tot+=calc(r+1,L-1);
if (it!=S.begin())
{
it--;
R=(*it).r;
}
else R=0;
if (R>=l) continue;
tot+=calc(R+1,l-1);
if (tot+1==calc(R+1,L-1))
{
ans[++cnt]=i;
S.insert(q[pos[i]]);
}
}
printf("%d\n",cnt);
for(int i=1;i<=cnt;i++)
printf("%d ",ans[i]);
}
int main()
{
init();
work();
return 0;
}