bzoj4448: [Scoi2015]情報傳遞（主席樹+Lca）

題目傳送門
嚴重吐槽這題資料！！
花了一下午一直在調一個AC程式碼。
提交莫名WA。因為我建的是單向邊。
題目說明規定了父親。那不就是單向邊嗎。。
然後本機棧不夠大RE了半天。
然後肉老師告訴我一個本地擴棧的方法然後我過了。。
然後又WA了。
然後%題解。發現建的都是雙向邊。
這題不是規定了父親嗎？？？
然後建了雙向才過了。。。
資料跟題意不符啊。

解法：
當前第i天有個詢問。問到了某個節點，這個節點是從第j天開始收集情報的。
那麼當i-j>c才能構成威脅。
轉化一下i-c>j。
那麼每個節點的權值就為這個點什麼時候開始收集情報。
那麼每次求比i-c小的節點個數就行了。
每個點的主席樹維護根到這個點的權值。
然後lca搞一下即可。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct node {int c,lc,rc;}t[11000000];int cnt,rt[2100000];
void build(int &u,int l,int r,int p) {
    if(u==0)u=++cnt;t[u].c++;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p);
    else build(t[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2) {
    if(u1==0) {u1=u2;return ;}if(u2==0)return ;
    t[u1].c+=t[u2].c;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
struct trnode {int x,y,next;}a[2100000];int len,last[2100000];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int z,dep[2100000],mx[21][210000];
void dfs(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]) {mx[0][y]=x;dep[y]=dep[x]+1;dfs(y);}
    }
}
int bin[21],n;
void work() {
    bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;
    for(int j=1;j<=20;j++)for(int i=1;i<=n;i++)if(dep[i]>=bin[j])
        mx[j][i]=mx[j-1][mx[j-1][i]];
}
int lca(int x,int y) {
    if(dep[x]>dep[y])swap(x,y);
    for(int i=20;i>=0;i--)if(dep[y]-dep[x]>=bin[i])y=mx[i][y];
    if(x==y)return x;
    for(int i=20;i>=0;i--)if(dep[x]>=bin[i]&&mx[i][x]!=mx[i][y])
        x=mx[i][x],y=mx[i][y];
    return mx[0][x];
}
int find(int u1,int u2,int u3,int u4,int l,int r,int p) {
    if(l==r)return t[u1].c+t[u2].c-t[u3].c-t[u4].c;
    int mid=(l+r)/2;
    if(p<=mid)return find(t[u1].lc,t[u2].lc,t[u3].lc,t[u4].lc,l,mid,p);
    else return find(t[u1].rc,t[u2].rc,t[u3].rc,t[u4].rc,mid+1,r,p)+
        t[t[u1].lc].c+t[t[u2].lc].c-t[t[u3].lc].c-t[t[u4].lc].c;
}
struct edge {int t,x,y,c;}s[210000];
void merge(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]){Merge(rt[y],rt[x]);merge(y);}
    }
}
int main() {
    //freopen("4448.in","r",stdin);freopen("4448.out","w",stdout);
    scanf("%d",&n);int root;len=0;memset(last,0,sizeof(last));cnt=0;
    for(int i=1;i<=n;i++) {int x;scanf("%d",&x);if(x==0)root=i;else ins(x,i);}
    int Q;dep[root]=0;mx[0][root]=0;z=0;dfs(root);work();scanf("%d",&Q);
    for(int i=1;i<=Q;i++) {
        scanf("%d%d",&s[i].t,&s[i].x);
        if(s[i].t==2)build(rt[s[i].x],1,Q,i);
        else scanf("%d%d",&s[i].y,&s[i].c);
    }merge(root);
    for(int i=1;i<=Q;i++)if(s[i].t==1) {
        int Lca=lca(s[i].x,s[i].y);
        printf("%d %d\n",dep[s[i].x]+dep[s[i].y]-dep[Lca]*2+1,
            find(rt[s[i].x],rt[s[i].y],rt[Lca],rt[mx[0][Lca]],1,Q,i-s[i].c-1));
    }
    return 0;
}