bzoj 4551(DFS序+线段树)

传送门
题解:跑一边DFS序,在序列上修改,将最大值重定义为深度最大的点。于是每一次修改就是在该点对应的区间内修改最大值(线段树维护),查询就是查该点在线段树上对应的最大值(其实就是那一个值)。
注意:pushdown()中lazy_tag也要取max!万年没写过区间修改的线段树,多快忘了。

#include
#include
#include
using namespace std;
#define root 1,1,n
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define pushup(rt) mx[rt]=max(mx[rt<<1],mx[rt<<1|1])
const int MAXN=1e5+5;
int n,q;
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt;
}e[MAXN];
int dep[MAXN],in[MAXN],out[MAXN],tim=0,rk[MAXN];
int mx[MAXN<<2],tag[MAXN<<2];
inline int max(int a,int b) {
    return dep[a]>dep[b]?a:b;
}
inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
inline void adde(int u,int v) {
    e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
}
void dfs(int p,int fa) {
    in[p]=++tim,dep[p]=dep[fa]+1;
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa) dfs(v,p);
    }
    out[p]=tim;
}
void pushdown(int rt) {
    if (tag[rt]) {
        tag[rt<<1]=max(tag[rt<<1],tag[rt]);
        tag[rt<<1|1]=max(tag[rt<<1|1],tag[rt]);
        mx[rt<<1]=max(mx[rt<<1],tag[rt]); 
        mx[rt<<1|1]=max(mx[rt<<1|1],tag[rt]);
    }
}
void build(int rt,int l,int r) {
    tag[rt]=0;
    if (l==r) {mx[rt]=1;return ;}
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void modify(int rt,int l,int r,int L,int R,int val) {
    if (L<=l&&r<=R) {
        tag[rt]=max(tag[rt],val);
        mx[rt]=max(mx[rt],val);
        return ;
    }
    pushdown(rt);
    int mid=(l+r)>>1;
    if (L<=mid) modify(lson,L,R,val);
    if (midint query(int rt,int l,int r,int pos) {
    if (l==r) return mx[rt];
    pushdown(rt);
    int mid=(l+r)>>1;
    if (pos<=mid) return query(lson,pos);
    else return query(rson,pos);
}
int main() {
//  freopen("bzoj 4551.in","r",stdin);
    memset(head,-1,sizeof(head));
    n=read(),q=read();
    for (register int i=1;iint u=read(),v=read();
        adde(u,v); 
    }
    dep[0]=0,dfs(1,0);
    build(root);
    for (register int r=1;r<=q;++r) {
        char ss;
        while (!isalpha(ss=getchar()));
        int pos=read();
        if (ss^'Q') modify(root,in[pos],out[pos],pos);
        else printf("%d\n",query(root,in[pos]));
    }
    return 0;
}

你可能感兴趣的:(数据结构)