原题链接:http://codeforces.com/contest/798/problem/E
题解:
大力补了一发E,看了半天的官方题解,结合了它的代码才看懂。。。还是菜啊?
首先很容易考虑到,可以通过拓扑序的方式来构造出答案,那么如何建图就成了问题的关键。
如果ai==-1,则让ai=n+1,如果某个点没有比它小的节点,就把它连到n+1
然后可以通过遍历一个有向图,每次遍历到某个边,就将该边删除。
考虑边(i,j)表示pi>pj,那么,在区间【1,a[i]-1】(i为当前节点)中的任意点j,如果存在边(j,k)且k>i,则必有pi>pj,如果不存在,那么表示已经不存在比点pi小的节点。
可以通过反证法给予证明:
考虑区间【1,a[i]-1】存边(j,k)其中,1<=j<=a[i]-1,1<=k<now,且pi>pj,因为我们是从1遍历到i,如果该边存在,那么一定有j>i,因为从i连出去的边我们会第一个遍历,否则这条边我们一定已经遍历过,那么考虑a[k],因为a[k]!=i,那么一定有pm>pk,而且m<k,在这种情况下,a[k]=m<i,所以不存在。
所以问题就转换成了,求一个区间中的最大值,同时支持修改区间中的某个点的模型。可以通过一个线段树来维护,每个点表示这个区间中的点连接的点集中,编号最大的点是那一个,并且维护该边的编号。删除操作只需要把该点连接到0就好。
时间复杂度O(nlogn)
#include<bits/stdc++.h>
#define xx first
#define yy second
#define mp make_pair
using namespace std;
const int MAXN=5e5+5;
pair<int,int> tree[MAXN<<2];
int a[MAXN],b[MAXN],ans[MAXN];
bool book[MAXN];
int n;
vector<int> sv;
void push_up(int rt)
{
tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
void build(int l,int r,int rt)
{
if(l==r)
{
tree[rt]=mp(b[l],l);
return ;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int pos,int val,int l,int r,int rt)
{
if(l==r)
{
tree[rt]=mp(val,tree[rt].yy);
return ;
}
int mid=(l+r)>>1;
if(pos<=mid)
{
update(pos,val,l,mid,rt<<1);
}
else
{
update(pos,val,mid+1,r,rt<<1|1);
}
push_up(rt);
}
pair<int,int> query(int L,int R,int l,int r,int rt)
{
pair<int,int> ret;
if(L>R)
{
return mp(0,0);
}
if(L<=l&&r<=R)
{
return tree[rt];
}
int mid=(l+r)>>1;
if(L<=mid)
{
ret=max(ret,query(L,R,l,mid,rt<<1));
}
if(R>mid)
{
ret=max(ret,query(L,R,mid+1,r,rt<<1|1));
}
return ret;
}
void dfs(int now)
{
book[now]=true;
update(now,0,1,n,1);
if(b[now]!=n+1&&!book[b[now]])
{
dfs(b[now]);
}
while(1)
{
pair<int,int> mx=query(1,a[now]-1,1,n,1);
if(mx.xx>now)
{
dfs(mx.yy);
}
else
{
break;
}
}
sv.push_back(now);
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
{
b[i]=n+1;
}
for(int i=1;i<=n;i++)
{
if(a[i]!=-1)
b[a[i]]=i;
}
for(int i=1;i<=n;i++)
{
if(a[i]==-1)
a[i]=n+1;
}
build(1,n,1);
for(int i=1;i<=n;i++)
{
if(!book[i])
dfs(i);
}
for(int i=0;i<sv.size();i++)
{
ans[sv[i]]=i+1;
}
for(int i=1;i<=n;i++)
{
printf("%d ",ans[i]);
}
printf("\n");
return 0;
}