• POJ 3237 树链剖分+线段树


    思路:树链剖分+线段树

    纯自己瞎想的 想到哪儿就写哪儿,,, 所以调了一下午+一晚上…..【尴尬】

    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    #define N 22222
    using namespace std;
    char ch[11];
    int cnt=0,rec[N],top[N],fa[N],deep[N];
    int cases,n,root,size[N],son[N],jyx,jyy;
    int first[N],next[N],v[N],w[N],tot;
    struct Edge{int from,to,weight;}edge[N];
    struct Tree{int maxx,minn,mark;void init(){maxx=minn=0;mark=0;}}tree[N*4];
    void add(int x,int y,int z){w[tot]=z,v[tot]=y,next[tot]=first[x],first[x]=tot++;}
    void dfs(int x){
        size[x]=1;son[x]=0;
        for(int i=first[x];~i;i=next[i])
            if(v[i]!=fa[x]){
                fa[v[i]]=x,deep[v[i]]=deep[x]+1;
                dfs(v[i]);
                size[x]+=size[v[i]];
                if(size[v[i]]>size[son[x]])son[x]=v[i];
            }
    }
    void build(int x,int tp){
        top[x]=tp;rec[x]=++cnt;
        if(son[x])build(son[x],tp);
        for(int i=first[x];~i;i=next[i])
            if(v[i]!=fa[x]&&v[i]!=son[x])
                build(v[i],v[i]);
    }
    void push_down(int lson,int rson){
        swap(tree[lson].maxx,tree[lson].minn);
        tree[lson].maxx*=-1,tree[lson].minn*=-1;
        swap(tree[rson].maxx,tree[rson].minn);
        tree[rson].maxx*=-1,tree[rson].minn*=-1;
        tree[lson].mark^=1;tree[rson].mark^=1;
    }
    void push_up(int pos){
        tree[pos].maxx=max(tree[pos<<1].maxx,tree[pos<<1|1].maxx);
        tree[pos].minn=min(tree[pos<<1].minn,tree[pos<<1|1].minn);
    }
    void insert(int l,int r,int pos,int location,int weight){
        if(l==r){tree[pos].maxx=tree[pos].minn=weight;tree[pos].mark=0;return;}
        int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
        if(tree[pos].mark)tree[pos].mark=0,push_down(lson,rson);
        if(mid>=location)insert(l,mid,lson,location,weight);
        else insert(mid+1,r,rson,location,weight);
        push_up(pos);
    }
    void update(int l,int r,int pos,int Left,int Right){
        if(l>=Left&&r<=Right){
                swap(tree[pos].maxx,tree[pos].minn);
                tree[pos].maxx*=-1;tree[pos].minn*=-1;
                tree[pos].mark^=1;
            return;
        }
        int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
        if(tree[pos].mark)tree[pos].mark=0,push_down(lson,rson);
        if(mid<Left)update(mid+1,r,rson,Left,Right);
        else if(mid>=Right)update(l,mid,lson,Left,Right);
        else update(l,mid,lson,Left,Right),update(mid+1,r,rson,Left,Right);
        push_up(pos);
    }
    int query(int l,int r,int pos,int Left,int Right){
        if(l>=Left&&r<=Right)return tree[pos].maxx;
        int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
        if(tree[pos].mark)tree[pos].mark=0,push_down(lson,rson);
        if(mid>=Right)return query(l,mid,lson,Left,Right);
        else if(mid<Left)return query(mid+1,r,rson,Left,Right);
        else return max(query(l,mid,lson,Left,Right),query(mid+1,r,rson,Left,Right));
    }
    int find(int x,int y){
        int fx=top[x],fy=top[y],temp=-0x3fffffff;
        while(fx!=fy){
            if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);
            temp=max(temp,query(1,cnt,1,rec[fx],rec[x]));
            x=fa[fx],fx=top[x];
        }
        if(x==y)return temp;
        if(deep[x]>deep[y])swap(x,y);
        return max(temp,query(1,cnt,1,rec[son[x]],rec[y]));
    }
    void negate(int x,int y){
        int fx=top[x],fy=top[y];
        while(fx!=fy){
            if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);
            update(1,cnt,1,rec[fx],rec[x]);
            x=fa[fx],fx=top[x];
        }
        if(x==y)return;
        if(deep[x]>deep[y])swap(x,y);
        update(1,cnt,1,rec[son[x]],rec[y]);
    }
    int main(){
        scanf("%d",&cases);
        while(cases--){
            memset(first,-1,sizeof(first));
            memset(size,0,sizeof(size));
            memset(fa,0,sizeof(fa));
            scanf("%d",&n);
            tot=cnt=deep[root]=fa[root]=0; 
            for(int i=1;i<n;i++){
                scanf("%d%d%d",&edge[i].from,&edge[i].to,&edge[i].weight);
                add(edge[i].from,edge[i].to,edge[i].weight);
                add(edge[i].to,edge[i].from,edge[i].weight);
            }
            root=(n+1)>>1;
            dfs(root);build(root,root);
            for(int i=1;i<n;i++){
                if(deep[edge[i].from]>deep[edge[i].to])swap(edge[i].from,edge[i].to);
                insert(1,cnt,1,rec[edge[i].to],edge[i].weight);
            }
            while(scanf("%s",ch)&&ch[0]!='D'){
                scanf("%d%d",&jyx,&jyy);
                if(ch[0]=='Q')printf("%d
    ",find(jyx,jyy));
                else if(ch[0]=='C')insert(1,cnt,1,rec[edge[jyx].to],jyy);
                else negate(jyx,jyy);
            }
        }
    }

    这里写图片描述

  • 相关阅读:
    《大道至简:软件工程实践者的思想》读后感
    周学习进度总结(2019.7.14)
    周学习进度总结(2019.7.7)
    周学习进度总结(2019.8.13)
    周学习进度总结(2019.8.4)
    石家庄铁道大学2019 年秋季 2018级JAVA课堂测试试卷(一)
    周学习进度总结(2019.7.27)
    周学习进度总结(2019.8.25)
    周学习进度总结(2019.7.20)
    C#中判断是否为数字
  • 原文地址:https://www.cnblogs.com/SiriusRen/p/6532315.html
Copyright © 2020-2023  润新知