【Bzoj4448】情報傳遞

題意

給一棵樹和m個操作(持續m秒),操作如下
1.給一個點打上標記,被標記的點每秒加1危險度,(被標記時仍爲0,後一秒爲1)。
2.查詢x,y的路徑上有多少危險度大於k的點。


解析

這種每秒增加的標記顯然是不好維護的。
考慮轉化。
設某點在t時刻被標記,i時刻提出詢問,那麼此時它的危險度爲i-t。
則有i-t>k,t

#include <cstdio>
#include <algorithm>

#define Rep( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i<=(i##_END);i++)
#define For( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i++)
#define Lop( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i>=(i##_END);i--)
#define Dnt( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i--)

using std :: max;
using std :: min;

const int maxx = 200000 + 25;

typedef int Array[maxx];

int n,m,x,y,z,num,cnt,tot,f,rt,t;
int head[maxx],to[maxx<<1],nxt[maxx<<1];
int T[maxx];

Array son,ftr,dpt,top,rnk,size,ans,ans1;

class Que{

    public:
        int l,r,id;
        int v,pos,f;

}A[maxx];

bool cmp(Que a,Que b){
    return a.v == b.v? a.f < b.f : a.v < b.v;
}

bool cmp1(Que a,Que b){
    return a.id < b.id;
}

namespace Bit{

    void Add(int x,int k){
        for(int i=x;i<=n;i+=i&(-i))
            T[i] += k;
    }

    int Query(int l,int r){
        int ans = 0;
        for(int i=r;i;i-=i&(-i))
            ans += T[i];
        for(int i=l-1;i;i-=i&(-i))
            ans -= T[i];
        return ans;
    }

}

namespace Cute{

    using namespace Bit; 

    void Ins(int x,int y){
        if((!x) || (!y)) return;
        to[++num] = y;nxt[num] = head[x];head[x] = num;
    }

    void Dfs1(int x){
        size[x] = 1;
        for(int i=head[x];i;i=nxt[i]){
            if(to[i] == ftr[x]) continue;
            dpt[to[i]] = dpt[x] + 1;ftr[to[i]] = x;
            Dfs1(to[i]);size[x] += size[to[i]];
            if(size[to[i]] > size[son[x]]) son[x] = to[i];
        }
    }

    void Dfs2(int x,int brn){
        rnk[x] = ++cnt;top[x] = brn;
        if(son[x]) Dfs2(son[x],brn);
        for(int i=head[x];i;i=nxt[i])
            if(to[i] != ftr[x] && to[i] != son[x])
                Dfs2(to[i],to[i]);
    }

    int Get(int x,int y,int &t){
        int ans = 0;t = dpt[x] + dpt[y];
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) std :: swap(x,y);
            ans += Query(rnk[top[y]],rnk[y]);
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) std :: swap(x,y);
        ans += Query(rnk[x],rnk[y]);
        t -= 2*dpt[x];t ++;
        return ans;
    }

}

using namespace Cute;

int main(){
    scanf("%d",&n);
    Rep( i , 1 , n ){
        scanf("%d",&x);
        if(x == 0) rt = i;
        Ins(i,x),Ins(x,i);
    }
    Dfs1(rt),Dfs2(rt,rt);
    scanf("%d",&m);
    Rep( i , 1 , m ){
        scanf("%d",&f);
        if(f == 2) scanf("%d",&x),A[i].pos = x,A[i].v = i;
        if(f == 1){
            scanf("%d%d%d",&x,&y,&z);
            A[i].l = x;A[i].r = y;
            A[i].v = i-z-1;
            A[i].f = 1;A[i].id = i;
        }
    }
    std :: sort(A+1,A+m+1,cmp);
    Rep( i , 1 , m ){
        if(A[i].f == 0) Add(rnk[A[i].pos],1);
        if(A[i].f == 1){
            Get(A[i].l,A[i].r,t);
            ans[A[i].id] = t;
            ans1[A[i].id] = Get(A[i].l,A[i].r,t);
            //printf("%d %d\n",t,Get(A[i].l,A[i].r,t));
        }
    }
    std :: sort(A+1,A+m+1,cmp1);
    Rep( i , 1 , m ) if(A[i].f) printf("%d %d\n",ans[A[i].id],ans1[A[i].id]);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章