【BJOI2017】树的难题【点分治】【线段树】

传送门
传送门

题意：给一棵树，树上有颜色，每种颜色有权值，定义一条路径的权值为所有颜色相同段的权值之和，求长度在[L,R][L,R][L,R]中的路径的最大权值。

数据范围：暴力过不了

显然是个点分治

对于分治中心考虑过中心的路径贡献的答案

（以下的“子树”指根直接与分治中心相连的子树）

把分治中心作为根，定义一棵子树的颜色为与根（即分治中心）相连的边的颜色

显然可以算出两边的贡献，如果两边颜色一样，再减掉这个权值

有个单调队列的神仙做法，看不懂

所以自己yyyyyy了线段树做法（题解也有）

把所有子树按颜色排序，把相同颜色放一起

维护两棵线段树，表示长度为xxx的路径的最大权值，一棵维护相同颜色，一棵维护不同颜色

遍历所有子树的所有节点，设当前节点的深度为disdisdis，区间查询两棵线段树中[L−dis,R−dis][L-dis,R-dis][L−dis,R−dis]的答案。

遍历完一棵子树后，将它们加入同色线段树。

当进入不同颜色时，把前一个颜色的点插入异色线段树，并给同色线段树打清除标记

复杂度大概O(NlogN2)O(Nlog_N^2)O(NlogN2)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cctype>
#include <algorithm>
#define MAXN 200005
#define MAXM 400005
#define re register
#define INF 0x3f3f3f3f
using namespace std;
class fast_input {private:static const int SIZE = 1 << 15 | 1;char buf[SIZE], *front, *back;void Next(char &c) {if(front == back) back = (front = buf) + fread(buf, 1, SIZE, stdin);c = front == back ? (char)EOF : *front++;}public :template<class T>void operator () (T &x) {char c, f = 1;for(Next(c); !isdigit(c); Next(c)) if(c == '-') f = -1;for(x = 0; isdigit(c); Next(c)) x = x * 10 + c - '0';x *= f;}void operator () (char &c, char l = 'a', char r = 'z') {for(Next(c); c > r || c < l; Next(c)) ;}
}input;struct edge{int u,v,w;}e[MAXM];
int head[MAXN],nxt[MAXM],cnt;
void addnode(int u,int v,int w)
{e[++cnt]=(edge){u,v,w};nxt[cnt]=head[u];head[u]=cnt;
}
int cost[MAXN],n,m,L,R;
bool cut[MAXN];
int siz[MAXN],sum[MAXN],maxp[MAXN],root;
void findroot(int u,int f,int sum)
{siz[u]=1,maxp[u]=0;for (re int i=head[u];i;i=nxt[i])if (e[i].v!=f&&!cut[e[i].v]){findroot(e[i].v,u,sum);siz[u]+=siz[e[i].v];maxp[u]=max(maxp[u],siz[e[i].v]);}if (sum-siz[u]>maxp[u]) maxp[u]=sum-siz[u];if (maxp[u]<maxp[root]) root=u;
}
struct SGT
{#define lc p<<1#define rc p<<1|1int mx[MAXN<<2],cl[MAXN<<2];inline void pushcl(int p){mx[p]=-INF;cl[p]=1;}inline void pushdown(int p){if (cl[p]) pushcl(lc),pushcl(rc),cl[p]=0;}void build(int p,int l,int r){mx[p]=-INF;if (l==r) return;int mid=(l+r)>>1;build(lc,l,mid),build(rc,mid+1,r);}void modify(int p,int l,int r,int k,int v){if (l==r) return (void)(mx[p]=max(mx[p],v));pushdown(p);int mid=(l+r)>>1;if (k<=mid) modify(lc,l,mid,k,v);else modify(rc,mid+1,r,k,v);mx[p]=max(mx[lc],mx[rc]);}int query(int p,int l,int r,int ql,int qr){pushdown(p);if (ql<=l&&r<=qr) return mx[p];if (r<ql||qr<l) return -INF;int mid=(l+r)>>1;return max(query(lc,l,mid,ql,qr),query(rc,mid+1,r,ql,qr));}inline void clear(){pushcl(1);}
}sam,dif;
int ans=-INF,maxlen;
int nod[MAXN],col[MAXN];
inline bool cmp(const int& a,const int& b){return col[a]<col[b];}
int tot;
int len[MAXN],val[MAXN],num,_len[MAXN],_val[MAXN],_num;
int dfs(int u,int f,int l,int v,int las_col)
{int s=1;if (l>=L&&l<=R) ans=max(ans,v);maxlen=max(maxlen,l);len[++num]=l,val[num]=v;_len[++_num]=l,_val[_num]=v;for (int i=head[u];i;i=nxt[i])if (e[i].v!=f&&!cut[e[i].v])s+=dfs(e[i].v,u,l+1,e[i].w==las_col? v:v+cost[e[i].w],e[i].w);return s;
}
void calc()
{maxlen=0;for (re int i=head[root];i;i=nxt[i])if (!cut[e[i].v])nod[++tot]=e[i].v,col[e[i].v]=e[i].w;sort(nod+1,nod+tot+1,cmp);for (re int i=1;i<=tot;i++){if (col[nod[i]]!=col[nod[i-1]]){for (int u=1;u<=_num;u++) dif.modify(1,1,n,_len[u],_val[u]);sam.clear(),_num=0;}sum[nod[i]]=dfs(nod[i],0,1,cost[col[nod[i]]],col[nod[i]]);for (re int u=1;u<=num;u++){ans=max(ans,val[u]+sam.query(1,1,n,L-len[u],R-len[u])-cost[col[nod[i]]]);ans=max(ans,val[u]+dif.query(1,1,n,L-len[u],R-len[u]));}for (re int u=1;u<=num;u++) sam.modify(1,1,n,len[u],val[u]);num=0;}sam.clear(),dif.clear();tot=num=_num=0;
}
void solve()
{cut[root]=true;calc();for (re int i=head[root];i;i=nxt[i])if (!cut[e[i].v]){maxp[root=0]=INF;findroot(e[i].v,0,sum[e[i].v]);sum[e[i].v]=0,solve();}
}
int main()
{input(n),input(m),input(L),input(R);for (re int i=1;i<=m;i++) input(cost[i]);for (re int i=1;i<n;i++){int u,v,w;input(u),input(v),input(w);addnode(u,v,w),addnode(v,u,w);}sam.build(1,1,n),dif.build(1,1,n);maxp[root=0]=INF;findroot(1,0,n);solve();printf("%d\n",ans);return 0;
}

【BJOI2017】树的难题【点分治】【线段树】相关推荐

[BJOI2017]树的难题点分治，线段树合并
[BJOI2017]树的难题 LG传送门点分治+线段树合并. 我不会写单调队列,所以就写了好写的线段树. 考虑对于每一个分治中心,把出边按颜色排序,这样就能把颜色相同的子树放在一起处理.用一棵动态开 ...
【bzoj4372】烁烁的游戏动态点分治+线段树
题目描述给一颗n个节点的树,边权均为1,初始点权均为0,m次操作: Q x:询问x的点权. M x d w:将树上与节点x距离不超过d的节点的点权均加上w. 输入第一行两个正整数:n,m 接下来的 ...
洛谷T44252 线索_分治线段树_思维题
分治线段树,其实就是将标记永久化,到最后再统一下传所有标记. 至于先后顺序,可以给每个节点开一个时间戳. 一般地,分治线段树用于离线,只查询一次答案的题目. 本题中,标记要被下传 222 次. Cod ...
BZOJ4012[HNOI2015]开店——树链剖分+可持久化线段树/动态点分治+vector
题目描述风见幽香有一个好朋友叫八云紫,她们经常一起看星星看月亮从诗词歌赋谈到人生哲学.最近她们灵机一动,打算在幻想乡开一家小店来做生意赚点钱.这样的想法当然非常好啦,但是她们也发现她们面临着一个 ...
[BZOJ4372][烁烁的游戏][动态树分治+线段树+LCA]
[BZOJ4372][烁烁的游戏][动态树分治+线段树+LCA] 题目大意: 给定一颗nn个节点的树,边权均为11,初始每个点权值为00 . 其中操作QQ xx询问x点的点权,操作 MM xx dd ...
【BZOJ4372】烁烁的游戏动态树分治+线段树
[BZOJ4372]烁烁的游戏 Description 背景:烁烁很喜欢爬树,这吓坏了树上的皮皮鼠. 题意: 给定一颗n个节点的树,边权均为1,初始树上没有皮皮鼠. 烁烁他每次会跳到一个节点u,把周围 ...
2019ICPC上海网络赛A 边分治+线段树
题目: 给定一棵树, 带边权. 现在有2种操作: 1.修改第i条边的权值. 2.询问u到其他一个任意点的最大距离是多少. 解法:边分治+线段树首先我们将所有的点修改和边修改都存在对应的边里面. 然后 ...
P6242-[模板]线段树3【吉司机线段树】
正题题目链接:https://www.luogu.com.cn/problem/P6242 题目大意给出一个长度为nnn的序列aaa,mmm次要求支持操作区间加上一个值kkk 区间所有aia_i ...
【BZOJ4515】游戏，树链剖分+永久化标记线段树维护线段信息（李超线段树）
Time:2016.05.10 Author:xiaoyimi 转载注明出处谢谢传送门思路: 李超线段树一开始听faebdc讲,并没有听的很懂ww 后来找到良心博文啊有木有折越首先可以把修改 ...
线段树简单入门 (含普通线段树, zkw线段树, 主席树)
线段树简单入门递归版线段树线段树的定义线段树, 顾名思义, 就是每个节点表示一个区间. 线段树通常维护一些区间的值, 例如区间和. 比如, 上图 \([2, 5]\) 区间的和, 为以下区间的和 ...

【BJOI2017】树的难题【点分治】【线段树】

【BJOI2017】树的难题【点分治】【线段树】相关推荐

最新文章

热门文章