【BZOJ4448】[Scoi2015]情报传递 主席树+LCA

【BZOJ4448】[Scoi2015]情报传递

Description

奈特公司是一个巨大的情报公司，它有着庞大的情报网络。情报网络中共有n名情报员。每名情报员能有若干名（可能没有）下线，除1名大头日外其余n-1名情报员有且仅有1名上线。奈特公司纪律森严，每名情报员只能与自己的上、下线联系，同时，情报网络中仟意两名情报员一定能够通过情报网络传递情报。

奈特公司每天会派发以下两种任务中的一个任务：

1．搜集情报：指派T号情报员搜集情报

2．传递情报：将一条情报从X号情报员传递给Y号情报员

情报员最初处于潜伏阶段，他们是相对安全的，我们认为此时所有情报员的危险值为0；-旦某个情报员开始搜集情报，他的危险值就会持续增加，每天增加1点危险值（开始搜集情报的当天危险值仍为0，第2天危险值为1，第3天危险值为2，以此类推）。传递情报并不会使情报员的危险值增加。为了保证传递情报的过程相对安全，每条情报都有一个风险控制值C。余特公司认为，参与传递这条情报的所有情报员中，危险值大于C的情报员将对该条情报构成威胁。现在，奈特公司希望知道，对于每个传递情报任务，参与传递的情报员有多少个，其中对该条情报构成威胁的情报员有多少个。

Input

第1行包含1个正整数n，表示情报员个数。

笫2行包含n个非负整数，其中第i个整数Pi表示i号情报员上线的编号。特别地，若Pi=0，表示i号情报员是大头目。

第3行包含1个正整数q，表示奈特公司将派发q个任务（每天一个）。

随后q行，依次描述q个任务。每行首先有1个正整数k。若k=1，表示任务是传递情报，随后有3个正整数Xi、Yi、Ci，依次表示传递情报的起点、终点和风险控制值；若k=2，表示任务是搜集情报，随后有1个正整数Ti，表示搜集情报的情报员编号。

Output

对于每个传递情报任务输出一行，应包含两个整数，分别是参与传递情报的情报员个数和对该条情报构成威胁的情报员个数。输出的行数应等于传递情报任务的个数，每行仅包含两个整数，用一个空格隔开。输出不应包含多余的空行和空格。

Sample Input

7
0 1 1 2 2 3 3
6
1 4 7 0
2 1
2 4
2
7
1 4 7 1
1 4 7 3

Sample Output

5 0 5 2 5 1

HINT

对于3个传递情报任务，都是经过5名情报员，分别是4号、2号、1号、3号和7号。其中，对于第1个任务，所有情报员（危险值为0）都不对情报构成威胁；对于第2个任务，有2名情报员对情报构成威胁，分别是1号情报员（危险值为3）和4号情报员（危险值为2），7号情报员（危险值为1）并不构成威胁；对于第3个任务，只有1名情报员对情报构成威胁。

n< = 2×10^5，Q< = 2×105，0< Pi，C!< = N, 1< = Ti，Xi，Yi< = n

题解：离线处理+主席树+倍增LCA，没有别的了~

#include <cstdio> 
#include <cstring> 
#include <iostream> 
using namespace std; 
const int maxn=200010; 
struct sag 
{ 
    int ls,rs,siz; 
}s[maxn*30]; 
int n,m,cnt,root,tot; 
int rt[maxn],fa[maxn][20],to[maxn],next[maxn],head[maxn],dep[maxn]; 
int v[maxn],qa[maxn],qb[maxn],qc[maxn]; 
int rd() 
{ 
    int ret=0;  char gc=getchar(); 
    while(gc<'0'||gc>'9') gc=getchar(); 
    while(gc>='0'&&gc<='9')   ret=ret*10+gc-'0',gc=getchar(); 
    return ret; 
} 
void add(int a,int b) 
{ 
    to[cnt]=b,next[cnt]=head[a],head[a]=cnt++; 
} 
void insert(int x,int &y,int l,int r,int pos) 
{ 
    if(!pos) 
    { 
        y=x; 
        return ; 
    } 
    y=++tot; 
    if(l==r) 
    { 
        s[y].siz++; 
        return; 
    } 
    int mid=l+r>>1; 
    if(pos<=mid) s[y].rs=s[x].rs,insert(s[x].ls,s[y].ls,l,mid,pos); 
    else    s[y].ls=s[x].ls,insert(s[x].rs,s[y].rs,mid+1,r,pos); 
    s[y].siz=s[x].siz+1; 
} 
void dfs(int x) 
{ 
    insert(rt[fa[x][0]],rt[x],1,m,v[x]); 
    for(int i=head[x];i!=-1;i=next[i])  dep[to[i]]=dep[x]+1,dfs(to[i]); 
} 
int query(int a,int b,int c,int d,int l,int r,int pos) 
{ 
    if(pos<=0)   return 0; 
    if(r<=pos)   return s[a].siz+s[b].siz-s[c].siz-s[d].siz; 
    int mid=l+r>>1; 
    if(pos<=mid) return query(s[a].ls,s[b].ls,s[c].ls,s[d].ls,l,mid,pos); 
    return s[s[a].ls].siz+s[s[b].ls].siz-s[s[c].ls].siz-s[s[d].ls].siz+query(s[a].rs,s[b].rs,s[c].rs,s[d].rs,mid+1,r,pos); 
} 
int main() 
{ 
    n=rd(); 
    int i,j,a,b,c; 
    memset(head,-1,sizeof(head)); 
    for(i=1;i<=n;i++) 
    { 
        fa[i][0]=rd(); 
        if(!fa[i][0])   root=i; 
        else    add(fa[i][0],i); 
    } 
    m=rd(); 
    for(i=1;i<=m;i++) 
    { 
        b=rd(); 
        if(b==1)    qa[i]=rd(),qb[i]=rd(),qc[i]=i-rd()-1; 
        else    a=rd(),v[a]=i; 
    } 
    for(j=1;(1<<j)<n;j++) 
        for(i=1;i<=n;i++) 
            fa[i][j]=fa[fa[i][j-1]][j-1]; 
    dep[1]=1,dfs(1); 
    for(i=1;i<=m;i++) 
    { 
        if(qa[i]) 
        { 
            a=qa[i],b=qb[i]; 
            if(dep[a]<dep[b])    swap(a,b); 
            for(j=19;j>=0;j--)   if(dep[fa[a][j]]>=dep[b])    a=fa[a][j]; 
            if(a!=b) 
            { 
                for(j=19;j>=0;j--)   if(fa[a][j]!=fa[b][j])  a=fa[a][j],b=fa[b][j]; 
                a=fa[a][0]; 
            } 
            c=dep[qa[i]]+dep[qb[i]]-2*dep[a]+1; 
            printf("%d %d\n",c,query(rt[qa[i]],rt[qb[i]],rt[a],rt[fa[a][0]],1,m,qc[i])); 
        } 
    } 
    return 0; 
}