zoukankan      html  css  js  c++  java
  • [bzoj5457]城市

    传送门

    这个题有多种解法,如虚树,分治,线段树合并,平衡树启发式合并,dsu on tree
    我采用了dsu on tree和线段树合并来解决这个题目
    很显然能看出这个题和维护子树信息有关
    dsu on tree相信大家都知道是一种处理子树信息的优秀算法,利用轻重链剖分的优秀性质,
    我们可以爆搜轻链,但是每次保留重链的信息,这样时间复杂度就是优秀的(O(nlogn))
    代码(dsu on tree):

    #include<cstdio>
    #include<iostream>
    #include<algorithm>
    using namespace std;
    void read(int &x) {
        char ch; bool ok;
        for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
        for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
    }
    #define rg register
    const int maxn=4e5+1;
    int heavy[maxn],dep[maxn],size[maxn],n,m,cnt,pre[maxn*2],nxt[maxn*2],h[maxn],ans[maxn],ans1[maxn],a[maxn],b[maxn],t[maxn],now,nnow;
    inline void add(int x,int y)
    {
        pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt;
        pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt;
    }
    inline void dfs(int x,int fa)
    {
        size[x]=1;
        for(rg int i=h[x];i;i=nxt[i])
            if(pre[i]!=fa)dep[pre[i]]=dep[x]+1,dfs(pre[i],x),size[x]+=size[pre[i]];
    }
    inline void dfs1(int x)
    {
        for(rg int i=h[x];i;i=nxt[i])
            if(dep[pre[i]]>dep[x]&&size[pre[i]]>size[heavy[x]])heavy[x]=pre[i];
        if(heavy[x])dfs1(heavy[x]);
        for(rg int i=h[x];i;i=nxt[i])
            if(dep[pre[i]]>dep[x]&&pre[i]!=heavy[x])dfs1(pre[i]);
    }
    inline void clear(int x,int fa)
    {
        t[a[x]]-=b[x];
        for(rg int i=h[x];i;i=nxt[i])if(pre[i]!=fa)clear(pre[i],x);
    }
    inline void get(int x)
    {
        t[a[x]]+=b[x];
        if(t[a[x]]>t[now]||(t[a[x]]==t[now]&&a[x]<now))now=a[x],nnow=t[a[x]];
    }
    inline void ins(int x,int fa)
    {
        get(x);
        for(rg int i=h[x];i;i=nxt[i])if(pre[i]!=fa)ins(pre[i],x);
    }
    inline void dsu(int x,int fa)
    {
        for(rg int i=h[x];i;i=nxt[i])
            if(pre[i]!=fa&&heavy[x]!=pre[i])dsu(pre[i],x),clear(pre[i],x);
        if(heavy[x])dsu(heavy[x],x);get(x);
        for(rg int i=h[x];i;i=nxt[i])
            if(pre[i]!=fa&&heavy[x]!=pre[i])ins(pre[i],x);
        ans[x]=now,ans1[x]=t[now];
    }
    int main()
    {
        read(n),read(m);
        for(rg int i=1,x,y;i<n;i++)read(x),read(y),add(x,y);
        for(rg int i=1;i<=n;i++)read(a[i]),read(b[i]);
        dfs(1,0),dfs1(1),dsu(1,0);
        for(rg int i=1;i<=n;i++)printf("%d %d
    ",ans[i],ans1[i]);
    }
    

    线段树合并也能解决这个问题,对于每个节点建一颗线段树,从叶子节点开始合并,每次合并复杂度为(O(logn)),然后更新答案,总时间复杂度为(O(nlog_2n)),空间复杂度为(O(nlogn))(动态开点)
    代码(线段树合并):

    #include<cstdio>
    #include<algorithm>
    int f[400001],n,m,pre[800001],cnt,rt[400001],ls[10000001],rs[10000001],nxt[800001],h[400001],a[400001],b[400001],id,sum[10000001],pw[10000001],ans[400001],ans1[400001];
    void add(int x,int y)
    {
        pre[++cnt]=y;nxt[cnt]=h[x];h[x]=cnt;
        pre[++cnt]=x;nxt[cnt]=h[y];h[y]=cnt;
    }
    void update(int x)
    {
        if(sum[ls[x]]>sum[rs[x]])sum[x]=sum[ls[x]],pw[x]=pw[ls[x]];
        else if(sum[ls[x]]==sum[rs[x]])sum[x]=sum[ls[x]],pw[x]=std::min(pw[ls[x]],pw[rs[x]]);
        else sum[x]=sum[rs[x]],pw[x]=pw[rs[x]];
    }
    void build(int &k,int l,int r,int v)
    {
        if(!k)k=++id;
        if(l==r){sum[k]+=b[v];pw[k]=a[v];return ;}
        int mid=(l+r)>>1;
        if(a[v]<=mid)build(ls[k],l,mid,v);
        else build(rs[k],mid+1,r,v);
        update(k);
    }
    int merge(int x,int y,int l,int r)
    {
        if(!x||!y)return x+y;
        if(l==r){sum[x]=sum[x]+sum[y];return x;}
        ls[x]=merge(ls[x],ls[y],l,(l+r)>>1);
        rs[x]=merge(rs[x],rs[y],((l+r)>>1)+1,r);
        update(x);return x;
    }
    void dfs(int x,int fa)
    {
        for(int i=h[x];i;i=nxt[i])if(pre[i]!=fa)dfs(pre[i],x);
        build(rt[x],1,m,x);
        for(int i=h[x];i;i=nxt[i])if(pre[i]!=fa)rt[x]=merge(rt[x],rt[pre[i]],1,m);
        ans[x]=sum[rt[x]];ans1[x]=pw[rt[x]];
    }
    int main()
    {
        scanf("%d%d",&n,&m);
        for(int i=1,x,y;i<n;i++)scanf("%d%d",&x,&y),add(x,y);
        for(int i=1;i<=n;i++)scanf("%d%d",&a[i],&b[i]);
        dfs(1,0);for(int i=1;i<=n;i++)printf("%d %d
    ",ans1[i],ans[i]);
    }
    
  • 相关阅读:
    IOS中UIImage用代码压缩
    自定义UILabel设置垂直方向的居上,居中,居下
    IOS开发之数据加密与解密:AES256+Base64转码
    IOS开发之网络数据Json处理
    Unity3d中CharacterController的移动和碰撞分析
    关于Unity3d中物体之间碰撞检测的方法
    WPF 调用Win32控件
    C# 扩展方法
    颜色空间转换 RGB到HSV
    加速计算技术-- Intel 的IPP库
  • 原文地址:https://www.cnblogs.com/lcxer/p/10280375.html
Copyright © 2011-2022 走看看