poj-3237(树链剖分)

对树有三种操作:

Q~  a b  询问a~b路径的最大值;

N~  a b  对a~b路径上的数进行取反操作;(a=-a)

C~  a b  将第a条边的值改为b;

对于取反操作,记录区间的最大和最小值和标记k,更新线段即可。

//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include
#include
#include
#include 
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define eps 1e-8
#define pii pair
#define inf 0x3f3f3f3f
#define rd(x) scanf("%d",&x)
#define rd2(x,y) scanf("%d%d",&x,&y)
#define ll long long int
#define mod 1000000007
#define maxn 100005
#define maxm 1000005
struct edge{
    int to,next;
}edge[maxn*2];
int head[maxn],tot;
int top[maxn];//v所在重链的顶端节点
int fa[maxn];
int deep[maxn];
int num[maxn];//表示以v为根的子树的节点数
int p[maxn];//表示v与其父亲节点的连边在线段树中的位置
int fp[maxn];//该边的儿子
int son[maxn];//重儿子
int pos;
int maxx(int a,int b){return a>b?a:b;}
int mixx(int a,int b){return anum[son[u]]) son[u]=v;
        }
    }
}
//第二遍dfs求出top和p
void getpos(int u,int sp){
    top[u]=sp;
    p[u]=pos++;
    fp[p[u]]=u;
    if(son[u]==-1) return;
    getpos(son[u],sp);
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v!=son[u]&&v!=fa[u]) getpos(v,v);
    }
}
struct node{
    int l,r,ma,mi,k;//k=1表示其左右儿子区间需要取反
}tree[maxn*3];
void build(int i,int l,int r){
    tree[i].l=l;tree[i].r=r;
    tree[i].k=0;
   // tree[i].ma=-inf;
   // tree[i].mi=inf;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build((i<<1)|1,mid+1,r);
}
void push_down(int i){//向下更新取反
    if(tree[i].l==tree[i].r) return;
    if(tree[i].k){
        tree[i].k^=1;
        tree[i<<1].k^=1;tree[(i<<1)|1].k^=1;
        tree[i<<1].mi=-tree[i<<1].mi;tree[i<<1].ma=-tree[i<<1].ma;
        tree[(i<<1)|1].mi=-tree[(i<<1)|1].mi;tree[(i<<1)|1].ma=-tree[(i<<1)|1].ma;
        swap(tree[i<<1].ma,tree[i<<1].mi);
        swap(tree[(i<<1)|1].ma,tree[(i<<1)|1].mi);
    }
}
void push_up(int i){
    tree[i].ma=maxx(tree[i<<1].ma,tree[(i<<1)|1].ma);
    tree[i].mi=mixx(tree[i<<1].mi,tree[(i<<1)|1].mi);
}
int query(int i,int l,int r){
    if(tree[i].l==l&&tree[i].r==r) return tree[i].ma;
    push_down(i);
    int mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid) return query(i<<1,l,r);
    else if(l>mid) return query((i<<1)|1,l,r);
    else return maxx(query(i<<1,l,mid),query((i<<1)|1,mid+1,r));
}
void change(int i,int l,int r){
    if(tree[i].l==l&&tree[i].r==r){
        tree[i].k^=1;
        tree[i].ma=-tree[i].ma;tree[i].mi=-tree[i].mi;
        swap(tree[i].ma,tree[i].mi);
        return;
    }
    push_down(i);
    int mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid) change(i<<1,l,r);
    else if(l>mid) change((i<<1)|1,l,r);
    else{
        change(i<<1,l,mid);
        change((i<<1)|1,mid+1,r);
    }
    push_up(i);
}
void update(int i,int k,int val){
    if(tree[i].l==tree[i].r) {
        tree[i].ma=tree[i].mi=val;
        return;
    }
    push_down(i);
    int mid=(tree[i].l+tree[i].r)>>1;
    if(k<=mid) update(i<<1,k,val);
    else update((i<<1)|1,k,val);
    push_up(i);
}
int findma(int u,int v){//最大值
    int f1=top[u],f2=top[v];
    int tmp=-1000000000;
    while(f1!=f2){
        if(deep[f1]deep[v]) swap(u,v);
    return maxx(tmp,query(1,p[son[u]],p[v]));
}
void change(int u,int v){//取反
    int f1=top[u],f2=top[v];
   // int tmp=-1000000000;
    while(f1!=f2){
        if(deep[f1]deep[v]) swap(u,v);
    //return maxx(tmp,query(1,p[son[u]],p[v]));
    change(1,p[son[u]],p[v]);
}
int t,n,a,b,c;
int e[maxn][3];
int main()
{
    rd(t);
    while(t--){
        rd(n);
        init();
        for(int i=1;ideep[e[i][1]]) swap(e[i][0],e[i][1]);
            update(1,p[e[i][1]],e[i][2]);
        }
        char op[10];
        while(scanf("%s",op)){
            if(op[0]=='D') break;
            scanf("%d%d",&a,&b);
            if(op[0]=='Q') printf("%d\n",findma(a,b));
            else if(op[0]=='N') change(a,b);
            else update(1,p[e[a][1]],b);
        }
    }
    return 0;
}


你可能感兴趣的:(树链剖分)