zoukankan      html  css  js  c++  java
  • P4556 [Vani有约会]雨天的尾巴 (线段树合并)

    P4556 [Vani有约会]雨天的尾巴

    题意:

    首先村落里的一共有n座房屋,并形成一个树状结构。然后救济粮分m次发放,每次选择两个房屋(x,y),然后对于x到y的路径上(含x和y)每座房子里发放一袋z类型的救济粮。
    然后深绘里想知道,当所有的救济粮发放完毕后,每座房子里存放的最多的是哪种救济粮。

    题解:

    树链剖分的写法很明显了,维护一个max即可

    讲一下线段树合并的写法

    区间更新用单点更新和差分来代替,求一个LCA,x->y的更新即可用在点x+1,点y+1,点lca(x,y)-1,点fa(lca(x,y))-1 后,线段树合并来取代, 线段树维护最多的救济粮编号val,最多救济粮的数量sum,然后在合并的时候就可以统计出u节点的答案了

    代码

    /**
     *        ┏┓    ┏┓
     *        ┏┛┗━━━━━━━┛┗━━━┓
     *        ┃       ┃  
     *        ┃   ━    ┃
     *        ┃ >   < ┃
     *        ┃       ┃
     *        ┃... ⌒ ...  ┃
     *        ┃       ┃
     *        ┗━┓   ┏━┛
     *          ┃   ┃ Code is far away from bug with the animal protecting          
     *          ┃   ┃   神兽保佑,代码无bug
     *          ┃   ┃           
     *          ┃   ┃        
     *          ┃   ┃
     *          ┃   ┃           
     *          ┃   ┗━━━┓
     *          ┃       ┣┓
     *          ┃       ┏┛
     *          ┗┓┓┏━┳┓┏┛
     *           ┃┫┫ ┃┫┫
     *           ┗┻┛ ┗┻┛
     */
    // warm heart, wagging tail,and a smile just for you!
    //
    //                            _ooOoo_
    //                           o8888888o
    //                           88" . "88
    //                           (| -_- |)
    //                           O  =  /O
    //                        ____/`---'\____
    //                      .'  |     |//  `.
    //                     /  |||  :  |||//  
    //                    /  _||||| -:- |||||-  
    //                    |   |   -  /// |   |
    //                    | \_|  ''---/''  |   |
    //                      .-\__  `-`  ___/-. /
    //                  ___`. .'  /--.--  `. . __
    //               ."" '<  `.___\_<|>_/___.'  >'"".
    //              | | :  `- \`.;` _ /`;.`/ - ` : | |
    //                 `-.   \_ __ /__ _/   .-` /  /
    //         ======`-.____`-.___\_____/___.-`____.-'======
    //                            `=---='
    //        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
    //                     佛祖保佑      永无BUG
    #include <set>
    #include <map>
    #include <stack>
    #include <cmath>
    #include <queue>
    #include <cstdio>
    #include <string>
    #include <bitset>
    #include <vector>
    #include <cstring>
    #include <iostream>
    #include <algorithm>
    using namespace std;
    typedef long long LL;
    typedef pair<int, int> pii;
    typedef unsigned long long uLL;
    
    #define lson l,mid,rt<<1
    #define rson mid+1,r,rt<<1|1
    #define bug printf("*********
    ")
    #define FIN freopen("input.txt","r",stdin);
    #define FON freopen("output.txt","w+",stdout);
    #define IO ios::sync_with_stdio(false),cin.tie(0)
    #define debug1(x) cout<<"["<<#x<<" "<<(x)<<"]
    "
    #define debug2(x,y) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<"]
    "
    #define debug3(x,y,z) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<" "<<#z<<" "<<z<<"]
    "
    const int maxn = 3e5 + 5;
    const int INF = 0x3f3f3f3f;
    const int mod = 1e9 + 7;
    const double Pi = acos(-1);
    LL gcd(LL a, LL b) {
        return b ? gcd(b, a % b) : a;
    }
    LL lcm(LL a, LL b) {
        return a / gcd(a, b) * b;
    }
    double dpow(double a, LL b) {
        double ans = 1.0;
        while(b) {
            if(b % 2)ans = ans * a;
            a = a * a;
            b /= 2;
        } return ans;
    }
    LL quick_pow(LL x, LL y) {
        LL ans = 1;
        while(y) {
            if(y & 1) {
                ans = ans * x % mod;
            } x = x * x % mod;
            y >>= 1;
        } return ans;
    }
    struct EDGE {
        int v, nxt;
    } edge[maxn << 1];
    int head[maxn], tot;
    void add_edge(int u, int v) {
        edge[tot].v = v;
        edge[tot].nxt = head[u];
        head[u] = tot++;
    }
    struct node {
        int l, r, sum, val;
    } tree[maxn * 40];
    int root[maxn];
    int tree_cnt;
    
    int sz[maxn], dep[maxn], fa[maxn], top[maxn], w[maxn], son[maxn], W[maxn], cnt;
    void init() {
        dep[1] = 1; fa[1] = 0;
        memset(head, -1, sizeof(head));
        tree_cnt = 0;
        tot = 0;
        cnt = 0;
    }
    #define ls tree[rt].l
    #define rs tree[rt].r
    
    void push_up(int rt) {
        if(tree[ls].sum > tree[rs].sum) {
            tree[rt].sum = tree[ls].sum;
            tree[rt].val = tree[ls].val;
        } else if(tree[ls].sum == tree[rs].sum) {
            tree[rt].sum = tree[ls].sum;
            tree[rt].val = min(tree[ls].val, tree[rs].val);
        } else {
            tree[rt].sum = tree[rs].sum;
            tree[rt].val = tree[rs].val;
        }
    }
    void update(int &x,  int l, int r, int pos, int val) {
        if(!x) x = ++tree_cnt;
        if(l == r) {
            tree[x].sum += val;
            if(tree[x].sum) {
                tree[x].val = l;
            } else {
                tree[x].val = 0;
            }
            return;
        }
        int mid = (l + r) >> 1;
        if(pos <= mid) update(tree[x].l,  l, mid, pos, val);
        else update(tree[x].r,  mid + 1, r, pos, val);
        push_up(x);
    }
    void merge(int &x, int y, int l, int r) {
        if(!x || !y) {
            x = x + y;
            return;
        }
        if(l == r) {
            tree[x].sum += tree[y].sum;
            if(tree[x].sum) {
                tree[x].val = l;
            } else {
                tree[x].val = 0;
            }
            return;
        }
        int mid = (l + r) >> 1;
        merge(tree[x].l, tree[y].l, l, mid);
        merge(tree[x].r, tree[y].r, mid + 1, r);
        push_up(x);
    }
    
    
    void dfs1(int u) {
        sz[u] = 1; son[u] = 0;
        for (int i = head[u]; ~i; i = edge[i].nxt) {
            int v = edge[i].v;
            if (v != fa[u]) {
                fa[v] = u;
                dep[v] = dep[u] + 1;
                dfs1(v);
                sz[u] += sz[v];
                if (sz[v] > sz[son[u]]) son[u] = v;
            }
        }
    }
    
    void dfs2(int u, int tp, int x) {
        top[u] = tp; w[u] = ++cnt; W[cnt] = u;
        if (son[u]) dfs2(son[u], tp, 1);
        for (int i = head[u]; ~i; i = edge[i].nxt) {
            int v = edge[i].v;
            if (v == son[u] || v == fa[u]) continue;
            dfs2(v, v, 2);
        }
    }
    
    
    int LCA(int x, int y) {
        while (top[x] != top[y]) {
            if (dep[top[x]] < dep[top[y]]) std::swap(x, y);
            x = fa[top[x]];
        }
        if (dep[x] > dep[y]) std::swap(x, y);
        return x;
    }
    int ans[maxn];
    void dfs(int u, int fa) {
        for(int i = head[u]; i != -1; i = edge[i].nxt) {
            int v = edge[i].v;
            if(v == fa) continue;
            dfs(v, u);
            merge(root[u], root[v], 1, 100000);
        }
        ans[u] = tree[root[u]].val;
    }
    int main() {
    #ifndef ONLINE_JUDGE
        FIN
    #endif
    
        int n, m;
        init();
        scanf("%d%d", &n, &m);
        for(int i = 1; i < n; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            add_edge(u, v);
            add_edge(v, u);
        }
        dfs1(1);
        dfs2(1, 1, 1);
        for(int i = 1; i <= m; i++) {
            int x, y, z;
            scanf("%d%d%d", &x, &y, &z);
            int lca = LCA(x, y);
            update(root[x], 1, 100000, z, 1);
            // bug;
            update(root[y], 1, 100000, z, 1);
            update(root[lca], 1, 100000, z, -1);
            if(fa[lca]) update(root[fa[lca]], 1, 100000, z, -1);
        }
        dfs(1, 0);
        for(int i = 1; i <= n; i++) {
            printf("%d
    ", ans[i]);
        }
        return 0;
    }
    
  • 相关阅读:
    Mysql体系结构(第4天)
    常用的课程网站
    Mysql 升级、用户与授权,
    Bootstrap
    项目中用到的一些特殊字符和图标 Content:"2715",特殊字符和图标
    CentOS 7 安装与卸载MySQL
    Iterm2使用
    Linux定时任务
    Ext.create细节分析
    Ext.define细节分析
  • 原文地址:https://www.cnblogs.com/buerdepepeqi/p/11656719.html
Copyright © 2011-2022 走看看