• bzoj1036: [ZJOI2008]树的统计Count 树链剖分


    一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意:从点u到点v的路径上的节点包括u和v本身,

    树链剖分裸题

    树链剖分就是将树剖成重链和轻链,然后按重儿子优先的dfs序建线段树,然后遇到查询时,只需在树上暴跳即可,每次将深度大的点跳到这条链的顶端,直到顶端相同(本质还是dfs序线段树,只是建树方式不太一样便可维护链上的信息,)复杂度O(nlognlogn),树上暴跳复杂度是O(logn),因为当树退化时跳的越远,所以复杂度logn

    /**************************************************************
        Problem: 1036
        User: walfy
        Language: C++
        Result: Accepted
        Time:2524 ms
        Memory:5552 kb
    ****************************************************************/
     
    //#pragma comment(linker, "/stack:200000000")
    //#pragma GCC optimize("Ofast,no-stack-protector")
    //#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
    //#pragma GCC optimize("unroll-loops")
    #include<bits/stdc++.h>
    #define fi first
    #define se second
    #define mp make_pair
    #define pb push_back
    #define pi acos(-1.0)
    #define ll long long
    #define vi vector<int>
    #define mod 1000000007
    #define C 0.5772156649
    #define ls l,m,rt<<1
    #define rs m+1,r,rt<<1|1
    #define pil pair<int,ll>
    #define pli pair<ll,int>
    #define pii pair<int,int>
    #define cd complex<double>
    #define ull unsigned long long
    #define base 1000000000000000000
    #define fio ios::sync_with_stdio(false);cin.tie(0)
     
    using namespace std;
     
    const double g=10.0,eps=1e-12;
    const int N=30000+10,maxn=60000+10,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f;
     
    struct edge{
        int to,Next;
    }e[maxn];
    int cnt,head[N];
    int son[N],fa[N],top[N],sz[N],id[N];
    int res,w[N],re[N],dep[N];
    void add(int u,int v)
    {
        e[cnt].to=v;
        e[cnt].Next=head[u];
        head[u]=cnt++;
    }
    void init()
    {
        cnt=0;
        memset(head,-1,sizeof head);
        memset(son,-1,sizeof son);
    }
    void dfs1(int u,int f,int de)
    {
        fa[u]=f;sz[u]=1;
        dep[u]=de;
        for(int i=head[u];~i;i=e[i].Next)
        {
            int v=e[i].to;
            if(v!=f)
            {
                dfs1(v,u,de+1);
                sz[u]+=sz[v];
                if(son[u]==-1||sz[v]>sz[son[u]])son[u]=v;
            }
        }
    }
    void dfs2(int u,int f,int tp)
    {
        top[u]=tp;
        id[u]=++res;
        if(son[u]!=-1)dfs2(son[u],u,tp);
        for(int i=head[u];~i;i=e[i].Next)
        {
            int v=e[i].to;
            if(v!=f&&v!=son[u])dfs2(v,u,v);
        }
    }
    int sum[N<<2],ma[N<<2];
    void pushup(int rt)
    {
        sum[rt]=sum[rt<<1]+sum[rt<<1|1];
        ma[rt]=max(ma[rt<<1],ma[rt<<1|1]);
    }
    void build(int l,int r,int rt)
    {
        if(l==r){sum[rt]=ma[rt]=w[re[l]];return ;}
        int m=(l+r)>>1;
        build(ls);build(rs);
        pushup(rt);
    }
    void update(int x,int c,int l,int r,int rt)
    {
        if(l==r){sum[rt]=ma[rt]=c;return ;}
        int m=(l+r)>>1;
        if(x<=m)update(x,c,ls);
        else update(x,c,rs);
        pushup(rt);
    }
    int queryma(int L,int R,int l,int r,int rt)
    {
        if(L<=l&&r<=R)return ma[rt];
        int m=(l+r)>>1,ans=-1e9;
        if(L<=m)ans=max(ans,queryma(L,R,ls));
        if(m<R)ans=max(ans,queryma(L,R,rs));
        return ans;
    }
    int querysum(int L,int R,int l,int r,int rt)
    {
        if(L<=l&&r<=R)return sum[rt];
        int m=(l+r)>>1,ans=0;
        if(L<=m)ans+=querysum(L,R,ls);
        if(m<R)ans+=querysum(L,R,rs);
        return ans;
    }
    int query(int op,int a,int b)
    {
        int f1=top[a],f2=top[b],ans=0;
        if(op==0)ans=-1e9;
        while(f1!=f2)
        {
            if(dep[f1]<dep[f2])swap(f1,f2),swap(a,b);
            if(op==0)ans=max(ans,queryma(id[f1],id[a],1,res,1));
            else ans+=querysum(id[f1],id[a],1,res,1);
            a=fa[f1];f1=top[a];
        }
        if(dep[a]>dep[b])swap(a,b);
        if(op==0)ans=max(ans,queryma(id[a],id[b],1,res,1));
        else ans+=querysum(id[a],id[b],1,res,1);
        return ans;
    }
    int main()
    {
        int n;
        scanf("%d",&n);
        init();
        for(int i=1;i<n;i++)
        {
            int a,b;scanf("%d%d",&a,&b);
            add(a,b);add(b,a);
        }
        for(int i=1;i<=n;i++)scanf("%d",&w[i]);
        dfs1(1,-1,1);
        dfs2(1,-1,1);
        for(int i=1;i<=res;i++)re[id[i]]=i;
        build(1,res,1);
        int q;scanf("%d",&q);
        while(q--)
        {
            char op[10];int a,b;
            scanf("%s%d%d",op,&a,&b);
            if(op[1]=='M')printf("%d
    ",query(0,a,b));
            else if(op[1]=='S')printf("%d
    ",query(1,a,b));
            else update(id[a],b,1,res,1);
        }
        return 0;
    }
    /***********************
    8
    1 2
    1 7
    2 3
    2 4
    4 5
    4 6
    7 8
    ***********************/
    View Code
  • 相关阅读:
    argparse模块的使用
    tf.stack() /tf.unstack()
    什么是tensor
    tf.size()函数
    tf.nn.l2_loss()的用法
    CNN中的卷积
    tf.reverse()
    学习音视频编程技术 博客
    shell 批量计算MD5值
    线程池的实现
  • 原文地址:https://www.cnblogs.com/acjiumeng/p/9021585.html
Copyright © 2020-2023  润新知