20200703 专题：线段树合并

总览：

inline int merge(int x,int y,int l,int r){
if(!x||!y)	return x|y;
if(l==r){
//将y合并到x
return x;
}
int mid=(l+r)>>1;
tr[x].ls=merge(tr[x].ls,tr[y].ls,l,mid);
tr[x].rs=merge(tr[x].rs,tr[y].rs,mid+1,r);
pushup(x);
return x;
}


T1 P4556 [Vani有约会]雨天的尾巴 /【模板】线段树合并

x,y 处加，lca(x,y),f(lca(x,y)) 处减

#include <bits/stdc++.h>
using namespace std;
#define re register
namespace IO {
inline char ch() {
static char buf[1 << 21], *p1 = buf, *p2 = buf;
return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2)
? EOF
: *p1++;
}
inline int in {
int s = 0, f = 1;
char x;
for (x = getchar(); x < '0' || x > '9'; x = getchar())
if (x == '-') f = -1;
for (; x >= '0' && x <= '9'; x = getchar())
s = (s << 1) + (s << 3) + (x & 15);
return f == 1 ? s : -s;
}
}  // namespace IO
using namespace IO;

const int A=4e5+5;
const int K=1e5;
int n,m;
int nex,to;
inline void edge(int x,int y){
}
int dep[A],f[A],top[A],son[A],sz[A];

inline void DFS1(int fa,int x){
f[x]=fa,dep[x]=dep[fa]+1,sz[x]=1;
if(z==fa)	continue;
DFS1(x,z);
sz[x]+=sz[z];
if(sz[z]>sz[son[x]])	son[x]=z;
}
return;
}

inline void DFS2(int x){
if(son[x]){
top[son[x]]=top[x];
DFS2(son[x]);
}
if(top[z])	continue;
top[z]=z;
DFS2(z);
}
return;
}

inline void tree_cut(){
DFS1(0,1);
top[1]=1;
DFS2(1);
return;
}

inline int lca(int x,int y){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])	swap(x,y);
x=f[top[x]];
}
return dep[x]<dep[y]?x:y;
}

int rt[A],tot;
struct Tree{
int ls,rs,num,ans;
}tr[20*A];
int res[A];

inline void pushup(int x){
if(tr[tr[x].ls].num<=0&&tr[tr[x].rs].num<=0)	return;
if(tr[tr[x].ls].num>=tr[tr[x].rs].num){
tr[x].num=tr[tr[x].ls].num;
tr[x].ans=tr[tr[x].ls].ans;
}
else{
tr[x].num=tr[tr[x].rs].num;
tr[x].ans=tr[tr[x].rs].ans;
}
return;
}

inline void add(int &x,int l,int r,int w,int val){
if(!x)	x=++tot;
if(l==r){
tr[x].num+=val;
tr[x].ans=tr[x].num>0?l:0;
return;
}
int mid=(l+r)>>1;
pushup(x);
return;
}

inline int merge(int x,int y,int l,int r){
if(!x||!y)	return x|y;
if(l==r){
tr[x].num+=tr[y].num;
tr[x].ans=tr[x].num>0?l:0;
return x;
}
int mid=(l+r)>>1;
tr[x].ls=merge(tr[x].ls,tr[y].ls,l,mid);
tr[x].rs=merge(tr[x].rs,tr[y].rs,mid+1,r);
pushup(x);
return x;
}

inline void work(int fa,int x){
if(z==fa)	continue;
work(x,z);
rt[x]=merge(rt[x],rt[z],1,K);
}
res[x]=tr[rt[x]].ans;
return;
}

signed main() {
n=in,m=in;
for(int i=1;i<n;i++){
int u=in,v=in;
edge(u,v),edge(v,u);
}
tree_cut();
for(int i=1;i<=m;i++){
int u=in,v=in,w=in;
}
work(0,1);
for(int i=1;i<=n;i++)
printf("%d\n",res[i]);
return 0;
}


T2 P1600 天天爱跑步

i 有贡献，当且仅当
d e p s − d e p i = w i d e p i + w i = d e p s dep_s-dep_i=w_i\\ dep_i+w_i=dep_s

i 有贡献，当且仅当
d e p s − d e p l c a + d e p i − d e p l c a = w i d e p s − 2 × d e p l c a = w i − d e p i dep_s-dep_{lca}+dep_i-dep_{lca}=w_i\\ dep_s-2\times dep_{lca}=w_i-dep_i

#include <bits/stdc++.h>
using namespace std;
#define re register
namespace IO {
inline char ch() {
static char buf[1 << 21], *p1 = buf, *p2 = buf;
return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2)
? EOF
: *p1++;
}
inline int in {
int s = 0, f = 1;
char x;
for (x = getchar(); x < '0' || x > '9'; x = getchar())
if (x == '-') f = -1;
for (; x >= '0' && x <= '9'; x = getchar())
s = (s << 1) + (s << 3) + (x & 15);
return f == 1 ? s : -s;
}
}  // namespace IO
using namespace IO;

const int A=3e5+5;
const int K=3e5;
int n,m;
int nex,to;
inline void edge(int x,int y){
}
int f[A],dep[A],sz[A],son[A],top[A];

inline void DFS1(int fa,int x){
f[x]=fa,dep[x]=dep[fa]+1,sz[x]=1;
if(z==fa)	continue;
DFS1(x,z);
sz[x]+=sz[z];
if(sz[z]>sz[son[x]])	son[x]=z;
}
return;
}

inline void DFS2(int x){
if(son[x]){
top[son[x]]=top[x];
DFS2(son[x]);
}
if(top[z])	continue;
top[z]=z;
DFS2(z);
}
return;
}

inline void tree_cut(){
DFS1(0,1);
top[1]=1;
DFS2(1);
return;
}

inline int lca(int x,int y){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])	swap(x,y);
x=f[top[x]];
}
return dep[x]<dep[y]?x:y;
}

int tim[A];
int rt[A],tot;
struct Tree{
int ls,rs,s[2];
}tr[20*A];
int res[A];

inline void pushup(int x){
if(!tr[x].ls&&!tr[x].rs)	return;
tr[x].s[0]=tr[tr[x].ls].s[0]+tr[tr[x].rs].s[0];
tr[x].s[1]=tr[tr[x].ls].s[1]+tr[tr[x].rs].s[1];
return;
}

inline void add(int &x,int l,int r,int w,int val,int pos){
if(!x)	x=++tot;
if(l==r){
tr[x].s[pos]+=val;
return;
}
int mid=(l+r)>>1;
pushup(x);
return;
}

inline int merge(int x,int y,int l,int r){
if(!x||!y)	return x|y;
if(l==r){
tr[x].s[0]+=tr[y].s[0];
tr[x].s[1]+=tr[y].s[1];
return x;
}
int mid=(l+r)>>1;
tr[x].ls=merge(tr[x].ls,tr[y].ls,l,mid);
tr[x].rs=merge(tr[x].rs,tr[y].rs,mid+1,r);
pushup(x);
return x;
}

inline int qurey(int x,int l,int r,int w,int pos){
if(!x)	return 0;
if(l==r)	return tr[x].s[pos];
int mid=(l+r)>>1;
if(w<=mid)	return qurey(tr[x].ls,l,mid,w,pos);
else	return qurey(tr[x].rs,mid+1,r,w,pos);
}

inline void work(int fa,int x){
if(z==fa)	continue;
work(x,z);
rt[x]=merge(rt[x],rt[z],-K,K);
}
res[x]+=qurey(rt[x],-K,K,dep[x]+tim[x],0);
res[x]+=qurey(rt[x],-K,K,tim[x]-dep[x],1);
return;
}

signed main() {
n=in,m=in;
for(int i=1;i<n;i++){
int u=in,v=in;
edge(u,v),edge(v,u);
}
tree_cut();
for(int i=1;i<=n;i++)	tim[i]=in;
for(int i=1;i<=m;i++){
int u=in,v=in;
int k=lca(u,v);
}
work(0,1);
for(int i=1;i<=n;i++)	printf("%d ",res[i]);
return 0;
}

• 点赞 1
• 评论
• 分享
x

海报分享

扫一扫，分享海报

• 收藏
• 手机看

分享到微信朋友圈

x

扫一扫，手机阅读

• 打赏

打赏

ハルカナソラヘ

你的鼓励将是我创作的最大动力

C币 余额
2C币 4C币 6C币 10C币 20C币 50C币
• 一键三连

点赞Mark关注该博主, 随时了解TA的最新博文
10-25
08-04 2582

08-02 48
08-15 1323
04-09 1397
05-11 147
07-16 660
10-27 38
09-28 1157
08-10 23