Description
给出n个二维点对,求LIS长度和编号字典序最小的LIS(x非增,y非减)
Input
多组用例,每组用例第一行为一整数n表示点数,第二行n个整数xi表示n个点的第一维,第三行n个整数yi表示n个点的第二维,以文件尾结束输入
(n<=50000,1<=xi,yi<=10^9)
Output
对于每组用例,第一行输出LIS长度ans,第二行ans个整数表示编号字典序最小的LIS的编号
Sample Input
5
5 4 3 2 1
6 7 8 9 10
2
1 2
3 4
Sample Output
5
1 2 3 4 5
1
1
Solution
三维偏序关系,CDQ分治即可,比较麻烦的是最小字典序,以dp[i]表示以第i个元素开始的LIS长度,类似一般的求以i结尾的LIS长度可知此处需要倒过来做CDQ,即先CDQ(mid+1,r),然后根据[mid+1,r]更新[l,mid],之后再CDQ(l,mid),那么CDQ(1,n)可以得到所有的dp值,则LIS长度ans=max(dp[i]),之后从前往后扫,每次找dp[i]==ans&&(pre==0||(p[pre].x>=p[i].x&&p[pre].y<=p[i].y))的i,其中pre为LIS上一个元素的编号,初始化为0,每次找到一个i就ans–,pre=i,这样就可以得到字典序最小的LIS了
Code
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 111111
int n,h[maxn],tot,dp[maxn];
struct node
{
int x,y,id;
bool operator <(const node &a)const
{
if(y!=a.y)return y<a.y;
if(x!=a.x)return x>a.x;
return id<a.id;
}
}p[maxn];
int cmp(node a,node b)
{
return a.id<b.id;
}
struct BIT
{
#define lowbit(x) (x&(-x))
int b[maxn];
void init()
{
memset(b,0,sizeof(b));
}
void update(int x,int v)
{
while(x<=tot)
{
b[x]=max(b[x],v);
x+=lowbit(x);
}
}
int query(int x)
{
int ans=0;
while(x)
{
ans=max(ans,b[x]);
x-=lowbit(x);
}
return ans;
}
void clear(int x)
{
while(x<=tot)
{
b[x]=0;
x+=lowbit(x);
}
}
}bit;
void CDQ(int l,int r)
{
if(l==r)return ;
int mid=(l+r)>>1;
CDQ(mid+1,r);
sort(p+l,p+mid+1);
sort(p+mid+1,p+r+1);
int j=r;
for(int i=mid;i>=l;i--)
{
for(;j>mid&&p[j].y>=p[i].y;j--)
bit.update(p[j].x,dp[p[j].id]);
dp[p[i].id]=max(dp[p[i].id],bit.query(p[i].x)+1);
}
for(int i=mid+1;i<=r;i++)bit.clear(p[i].x);
sort(p+l,p+mid+1,cmp);
CDQ(l,mid);
}
int main()
{
while(~scanf("%d",&n))
{
tot=0,bit.init();
for(int i=1;i<=n;i++)scanf("%d",&p[i].x),h[++tot]=p[i].x;
for(int i=1;i<=n;i++)scanf("%d",&p[i].y),h[++tot]=p[i].y;
sort(h+1,h+tot+1);
for(int i=1;i<=n;i++)
{
p[i].x=lower_bound(h+1,h+tot+1,p[i].x)-h;
p[i].y=lower_bound(h+1,h+tot+1,p[i].y)-h;
p[i].id=i,dp[i]=1;
}
CDQ(1,n);
sort(p+1,p+n+1,cmp);
int ans=0,pre=0;
for(int i=1;i<=n;i++)ans=max(ans,dp[i]);
printf("%d\n",ans);
for(int i=1;i<=n;i++)
if(dp[i]==ans&&(pre==0||(p[pre].x>=p[i].x&&p[pre].y<=p[i].y)))
printf("%d%c",i,(--ans)?' ':'\n'),pre=i;
}
return 0;
}