[BZOJ1103][POI2007]大都市meg

Time Limit: 10 Sec
Memory Limit: 162 MB

Description

在经济全球化浪潮的影响下,习惯于漫步在清晨的乡间小路的邮递员Blue Mary也开始骑着摩托车传递邮件了。不过，她经常回忆起以前在乡间漫步的情景。昔日，乡下有依次编号为1…n的n个小村庄，某些村庄之间有一些双向的土路。从每个村庄都恰好有一条路径到达村庄1（即比特堡）。并且，对于每个村庄，它到比特堡的路径恰好只经过编号比它的编号小的村庄。另外，对于所有道路而言，它们都不在除村庄以外的其他地点相遇。在这个未开化的地方，从来没有过高架桥和地下铁道。随着时间的推移，越来越多的土路被改造成了公路。至今，Blue Mary还清晰地记得最后一条土路被改造为公路的情景。现在，这里已经没有土路了——所有的路都成为了公路，而昔日的村庄已经变成了一个大都市。 Blue Mary想起了在改造期间她送信的经历。她从比特堡出发，需要去某个村庄，并且在两次送信经历的间隔期间,有某些土路被改造成了公路.现在Blue Mary需要你的帮助：计算出每次送信她需要走过的土路数目。（对于公路，她可以骑摩托车；而对于土路，她就只好推车了。）

Input

第一行是一个数 $n(1 < = n < = 2 50000)$ .以下 $n-1$ 行，每行两个整 $a，b（1 < = a<=b)$ 以下一行包含一个整数 $m(1 < = m < = 2 50000)$ ，表示Blue Mary曾经在改造期间送过 $m$ 次信。以下 $n+m-1$ 行，每行有两种格式的若干信息，表示按时间先后发生过的 $n+m-1$ 次事件:若这行为 A a b $(a<=b)$ 表示将从a到b的土路改成公路。
若这行为 W a, 则表示Blue Mary曾经从比特堡送信到村庄a。
Output

有m行，每行包含一个整数，表示对应的某次送信时经过的土路数目。
　　
Sample Input

Sample Output

HINT
在这里插入图片描述

题解：
dfs序+线段树区间修改。
修路操作当成子树全部结点权值 $-1$
查询的话就是单点查询了。

#include<bits/stdc++.h>
#define LiangJiaJun main
using namespace std;

struct seg{
    int l,r,w,tag;
}tr[1000004];
struct edge{
    int to,nt;
}e[500004];
int cnt;
int n,m,ne,h[250004];
int dfn[250004],kcf[250004],lp[250004],opg[250004];
void add(int u,int v){
     e[++ne].to=v;e[ne].nt=h[u];
     h[u]=ne;
}
void dfs(int x){
     dfn[x]=++cnt;
     kcf[cnt]=x;
     for(int i=h[x];i;i=e[i].nt){
         if(!dfn[e[i].to]){
            opg[e[i].to]=opg[x]+1;
            dfs(e[i].to);
         }
     }
     lp[x]=cnt;
}
void push(int k){
     int x=tr[k].tag;
     tr[k<<1].tag+=x;
     tr[k<<1|1].tag+=x;
     tr[k].tag=0;
     tr[k<<1].w+=x*(tr[k<<1].r-tr[k<<1].l+1);
     tr[k<<1|1].w+=x*(tr[k<<1|1].r-tr[k<<1|1].l+1);
     return ;
}
void build(int k,int l,int r){
     tr[k].l=l;tr[k].r=r;
     tr[k].tag=0;
     if(l==r){
        tr[k].w=opg[kcf[l]];
        return ;
     }
     int mid=(l+r)>>1;
     build(k<<1,l,mid);
     build(k<<1|1,mid+1,r);
     tr[k].w=tr[k<<1].w+tr[k<<1|1].w;
}
void change(int k,int a,int b,int x){
     int l=tr[k].l,r=tr[k].r;
     if(l==a&&r==b){
        tr[k].w+=x*(r-l+1);
        tr[k].tag+=x;
        return ;
     }
     if(tr[k].tag)push(k);
     int mid=(l+r)>>1;
     if(b<=mid)change(k<<1,a,b,x);
     else if(a>mid)change(k<<1|1,a,b,x);
     else{
        change(k<<1,a,mid,x);
        change(k<<1|1,mid+1,b,x);
     }
     tr[k].w=tr[k<<1].w+tr[k<<1|1].w;
}
int ask(int k,int a){
    int l=tr[k].l,r=tr[k].r;
    if(l==a&&r==a){
        return tr[k].w;
    }
    if(tr[k].tag)push(k);
    int mid=(l+r)>>1;
    if(a<=mid)return ask(k<<1,a);
    else return ask(k<<1|1,a);
}
int w33ha(){
    ne=0;
    memset(h,0,sizeof(h));
    for(int i=1;i<=n;i++){
        dfn[i]=0;
        lp[i]=0;
        kcf[i]=0;
    }
    cnt=0;
    opg[1]=0;
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1);
    build(1,1,cnt);
    scanf("%d",&m);
    m=n+m-1;
    while(m--){
        char ch[4];
        int x,y;
        scanf("%s",ch+1);
        if(ch[1]=='A'){
            scanf("%d%d",&x,&y);
            change(1,dfn[max(x,y)],lp[max(x,y)],-1);
        }
        if(ch[1]=='W'){
            scanf("%d",&x);
            printf("%d\n",ask(1,dfn[x]));
        }
    }
    return 0;
}

int LiangJiaJun(){
    while(scanf("%d",&n)!=EOF)w33ha();
    return 0;
}

[BZOJ1103][POI2007]大都市meg

猜你喜欢