LG4556 [Vani有约会]雨天的尾巴 动态开点线段树+线段树合并

问题描述

LG4556


题解

对于每一个结点,建立一棵动态开点线段树。

然后自低向上合并线段树。

同时维护整个值域的最大值和最大值位置。


\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

template <typename Tp>
void read(Tp &x){
    x=0;char ch=1;int fh;
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
    if(ch=='-') ch=getchar(),fh=-1;
    else fh=1;
    while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    x*=fh;
}

const int maxn=1000007;
const int maxm=2000007;

int n,m;
int Head[maxn],Next[maxm],to[maxm],tot;

int size[maxn],fa[maxn],son[maxn];
int dep[maxn],top[maxn],du[maxn];

void add(int x,int y){
    to[++tot]=y,Next[tot]=Head[x],Head[x]=tot;
}

void dfs1(int x,int f,int dp){
    size[x]=1,fa[x]=f,dep[x]=dp;
    int mx=-1;
    for(int i=Head[x];i;i=Next[i]){
        int y=to[i];
        if(y==f) continue;
        dfs1(y,x,dp+1);size[x]+=size[y];
        if(size[y]>mx) mx=size[y],son[x]=y;
        ++du[x];
    }
}

void dfs2(int x,int tp){
    top[x]=tp;
    if(!son[x]) return;
    dfs2(son[x],tp);
    for(int i=Head[x];i;i=Next[i]){
        int y=to[i];
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}

int lca(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    return x;
}

int L[maxn*22],R[maxn*22];
int rt[maxn],val[maxn*22],pos[maxn*22];
int cnt;

void pushup(int x){
    val[x]=max(val[L[x]],val[R[x]]);
    if(val[x]<=0) return;
    if(val[x]==val[L[x]]) pos[x]=pos[L[x]];
    else pos[x]=pos[R[x]];
}

int insert(int x,int l,int r,int p,int v){
    if(!x) x=++cnt;
    if(l==r){
        val[x]+=v,pos[x]=l;
        return x;
    }
    int mid=(l+r)>>1;
    if(p<=mid) L[x]=insert(L[x],l,mid,p,v);
    else R[x]=insert(R[x],mid+1,r,p,v);
    pushup(x);
    return x;
}

int ans[maxn];

int merge(int p,int q,int l,int r){
    if(!p||!q) return p+q;
    if(l==r){
        val[q]+=val[p],pos[q]=l;
        return q;
    }
    int mid=(l+r)>>1;
    L[q]=merge(L[p],L[q],l,mid);
    R[q]=merge(R[p],R[q],mid+1,r);
    pushup(q);
    return q;
}

int main(){
    read(n);read(m);
    for(int i=1,x,y;i<n;i++){
        read(x);read(y);
        add(x,y);add(y,x);
    }
    dfs1(1,0,1);dfs2(1,1);
    for(int i=1;i<=n;i++) rt[i]=++cnt;
    for(int i=1,x,y,z;i<=m;i++){
        read(x);read(y);read(z);
        int LCA=lca(x,y);
        insert(rt[x],1,100000,z,1);
        insert(rt[y],1,100000,z,1);
        insert(rt[LCA],1,100000,z,-1);
        insert(rt[fa[LCA]],1,100000,z,-1);
    }
    queue<int>q;
    for(int i=1;i<=n;i++){
        if(!du[i]) q.push(i);
    }
    while(!q.empty()){
        int x=q.front();q.pop();
        ans[x]=pos[rt[x]];
        merge(rt[x],rt[fa[x]],1,100000);
        --du[fa[x]];
        if(!du[fa[x]]) q.push(fa[x]);
    }
    for(int i=1;i<=n;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}

原文地址:https://www.cnblogs.com/liubainian/p/11828818.html

时间: 2024-11-09 01:42:47

LG4556 [Vani有约会]雨天的尾巴 动态开点线段树+线段树合并的相关文章

P4556 [Vani有约会]雨天的尾巴 树链剖分 线段树合并

P4556 [Vani有约会]雨天的尾巴 提交2.75k 通过789 时间限制1.00s 内存限制125.00MB 提交代码加入收藏 题目提供者yyy2015c01 难度省选/NOI- 历史分数100 提交记录查看题解 标签 查看算法标签 相关讨论 进入讨论版 查看讨论 推荐题目 查看推荐 展开 题目背景 深绘里一直很讨厌雨天.灼热的天气穿透了前半个夏天,后来一场大雨和随之而来的洪水,浇灭了一切.虽然深绘里家乡的小村落对洪水有着顽固的抵抗力,但也倒了几座老房子,几棵老树被连根拔起,以及田地里的粮

P4556 [Vani有约会]雨天的尾巴(线段树合并)

传送门 一道线段树合并 首先不难看出树上差分 我们把每一次修改拆成四个,在\(u,v\)分别放上一个,在\(lca\)和\(fa[lca]\)各减去一个,那么只要统计一下子树里的总数即可 然而问题就在于怎么统计.直接暴力肯定是要咕咕的,那么线段树合并就派上用场了 总之就是每个点开一个动态开点线段树,然后一遍dfs,让它的所有儿子的线段树合并到它这里 我按以前的写法不知为什么写挂了--然后换抄了种写法还是挂--后来发现是写抄的时候没有注意合并的顺序-- //minamoto #include<bi

[Vani有约会]雨天的尾巴

我之前考试是遇到过这题,但是数据范围k<=20,状压就能过. 结果原题范围k<=100000-- 果断线段树合并. 普及线段树合并: 比如两个相同大小的线段树,将b树各个区间上的值合并到a树上,从树根开始合并,然后递归合并左右儿子,有三种情况: (假设现在a树遍历到x点,b树遍历到y点) 1.x,y至少其一未被修改过(语文不好勿喷),则将x变为遍历过的那个. 2.x,y位于叶节点(l==r),则sum[x]+=sum[y]. 3.一般情况,递归处理左右儿子,最后更新当前点. 本题中合并如下:

[CSP-S模拟测试]:表格(动态开点二维线段树+离散化)

题目传送门(内部题112) 输入格式 一个数$N$,表示矩形的个数. 接下来$N$行,每行四个整数$X_a,Y_a,X_b,Y_b$.分别表示每个矩形左下角和右上角的坐标. 保证$(X_a<X_b,Y_a<Y_b)$. 输出格式 一行,表示能看到的颜色数量. 样例 样例输入: 30 -1 1 12 1 3 5-4 0 5 4 样例输出: 4 数据范围与提示 样例解释: 数据范围: 对于$10\%$的数据,保证$N\leqslant 100,|X_a,X_b,Y_a,Y_b|\leqslant

P2617 Dynamic Rankings (动态开点权值线段树 + 树状数组)

题意:带修求区间k小 题解:回忆在使用主席树求区间k小时 利用前缀和的思想 既然是前缀和 那么我们可以使用更擅长维护前缀和的树状数组 但是这里每一颗权值线段树就不是带版本的 而是维护数组里i号点的权值信息 所以实际上并不是主席树 每一棵和前面一棵并没有共用结点 对于一次修改操作 我们先删去这个点的原信息 再更新进去 树状数组上的点一起跳 可能看代码比较好理解一点 这个方法限制性也很强 必须离线 #include <bits/stdc++.h> using namespace std; cons

bzoj3307: 雨天的尾巴

3307: 雨天的尾巴 Time Limit: 10 Sec  Memory Limit: 128 MBSubmit: 479  Solved: 214[Submit][Status][Discuss] Description N个点,形成一个树状结构.有M次发放,每次选择两个点x,y对于x到y的路径上(含x,y)每个点发一袋Z类型的物品.完成所有发放后,每个点存放最多的是哪种物品. Input 第一行数字N,M接下来N-1行,每行两个数字a,b,表示a与b间有一条边再接下来M行,每行三个数字x

GDOI模拟雨天的尾巴【树套树】

雨天的尾巴 深绘里一直很讨厌雨天.灼热的天气穿透了前半个夏天,后来一场大雨和随之而来的洪水,浇灭了一切.虽然深绘里家乡的小村落对洪水有着顽固的抵抗力,但也倒了几座老房子,几棵老树被连根拔起,以及田地里的粮食被弄得一片狼藉.无奈的深绘里和村民们只好等待救济粮来维生.不过救济粮的发放方式很特别.首先村落里的一共有n 座房屋,并形成一个树状结构.然后救济粮分m 次发放,每次选择两个房屋(x,y),然后对于x 到y 的路径上(含x 和y) 每座房子里发放一袋z 类型的救济粮.然后深绘里想知道,当所有的救

[bzoj3307]雨天的尾巴_线段树合并

雨天的尾巴 bzoj-3307 题目大意:N个点,形成一个树状结构.有M次发放,每次选择两个点x,y对于x到y的路径上(含x,y)每个点发一袋Z类型的物品.完成所有发放后,每个点存放最多的是哪种物品. 想法:我们像主席树一样,维护桶然后对应节点合并即可. 最后,附上丑陋的代码... ... #include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #define

[luogu4556]雨天的尾巴

[luogu4556]雨天的尾巴 luogu 发现是一顿子修改然后再询问,那么把修改树上差分一下再线段树合并 但是... 如果你只有35分... https://www.luogu.org/discuss/show/88259 #include<bits/stdc++.h> using namespace std; const int _=1e5+5; int re(){ int x=0,w=1;char ch=getchar(); while(ch<'0'||ch>'9'){if