【DFS序+线段树】BZOJ1103 [POI2007]大都市meg

来源：互联网发布：ff14猫男数据编辑：程序博客网时间：2024/05/17 07:47

题面在这里

很显然可以用DFS序把树上问题转化为区间问题

修改一条边只影响了下面那棵子树中所有点的答案

线段树维护区间加，单点询问就好了

示例程序：

#include<cstdio>#include<algorithm>#define Fst first#define Sec second#define mp make_pairusing namespace std;inline char nc(){    static char buf[100000],*p1=buf,*p2=buf;    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;}inline int red(){    int res=0,f=1;char ch=nc();    while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=nc();}    while ('0'<=ch&&ch<='9') res=res*10+ch-48,ch=nc();    return res*f;}inline char fstchar(){    char ch=nc();while (ch<'A'||'Z'<ch) ch=nc();    return ch;}const int maxn=250005,maxe=500005;pair<int,int> E[maxn];int n,q,times,in[maxn],out[maxn],dep[maxn],id[maxn];int tot,nxt[maxe],lnk[maxn],son[maxe];inline void add(int x,int y){    son[++tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;}void dfs(int x){    in[x]=++times;id[times]=x;    for (int j=lnk[x];j;j=nxt[j])     if (son[j]>x){        dep[son[j]]=dep[x]+1;        dfs(son[j]);     }    out[x]=times;}struct node{    node *l,*r;    int L,R,s,tag;    node () {}    node (int _L,int _R):L(_L),R(_R),s(0),tag(0) {}    void pushup() {s=l->s+r->s;}    void pushdown(){        if (L==R||!tag) return;        l->tag+=tag;l->s+=tag*(l->R - l->L +1);        r->tag+=tag;r->s+=tag*(r->R - r->L +1);        tag=0;    }}nil,base[maxe];typedef node* P_node;P_node null,len,Rot;void init(){    nil=node(0,0);null=&nil;    null->l=null->r=null;len=base;}P_node newnode(int L,int R){    *len=node(L,R);    len->l=len->r=null;    return len++;}P_node build(int L,int R){    P_node x=newnode(L,R);    if (L==R) {x->s=dep[id[L]];return x;}    int mid=L+R>>1;    x->l=build(L,mid); x->r=build(mid+1,R);    x->pushup(); return x;}void insert(P_node x,int L,int R){    if (R<x->L||x->R<L) return;    if (L<=x->L&&x->R<=R) {x->s-=x->R - x->L +1;x->tag+=-1;return;}    x->pushdown();    insert(x->l,L,R);insert(x->r,L,R);    x->pushup();}int query(P_node x,int k){    x->pushdown();    if (x->L==x->R) return x->s;    int mid=x->L+x->R>>1;    if (k<=mid) return query(x->l,k);else return query(x->r,k);}int main(){    n=red();    for (int i=1,x,y;i<n;i++){        x=red(),y=red();        if (x>y) swap(x,y);E[i]=mp(x,y);        add(x,y);add(y,x);    }    sort(E+1,E+n);    dfs(1);    init();Rot=build(1,n);    q=red()+n-1;    while (q--)     if (fstchar()=='A'){        int x=red(),y=red();        int t=lower_bound(E+1,E+n,mp(x,y))->Sec;        insert(Rot,in[t],out[t]);     }else printf("%d\n",query(Rot,in[red()]));    return 0;}

阅读全文

1 0