听说后天会考x

省选居然还考模板题的么(好吧好像NOI也有考而且也是树剖…)

题意:一棵树,每个点有权值,三种操作:单点修改、求链上最大值、求链上权值和。

直接上模板。

我可能不会写单点修改的线段树了就直接写了个区间修改的用…

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=30005;
const int T=233333;
inline int read()
{
int s=0,f=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){s=s*10+c-'0';c=getchar();}
return s*f;
}
struct edge
{
int to,nxt;
}edges[N<<1];
int n,q,cnt,tot;
int head[N<<1],trs[N<<2],trm[N<<2],lazy[N<<2],top[N],w[N],rk[N],fa[N],size[N],dep[N],son[N],idx[N];
char s[20]; inline void addEdge(int u,int v)
{
edges[++cnt]=(edge){v,head[u]};
head[u]=cnt;
} #define lson (node<<1)
#define rson (node<<1|1) inline void push_up(int node)
{
trs[node]=trs[lson]+trs[rson];
trm[node]=max(trm[lson],trm[rson]);
}
inline void push_down(int node,int l,int r)
{
if(lazy[node]==T)return;
int mid=(l+r)>>1;
trs[lson]=(mid-l+1)*lazy[node];trs[rson]=(r-mid)*lazy[node];trm[lson]=trm[rson]=lazy[node];
lazy[lson]=lazy[rson]=lazy[node];
lazy[node]=T;
}
inline void build(int node,int l,int r)
{
lazy[node]=T;
if(l==r)
{
trs[node]=trm[node]=w[rk[l]];
return ;
}
int mid=(l+r)>>1;
build(lson,l,mid);build(rson,mid+1,r);
push_up(node);
}
inline void modify(int node,int l,int r,int ql,int qr,int val)
{
if(ql<=l&&r<=qr)
{
lazy[node]=val;
trs[node]=(r-l+1)*val;
trm[node]=val;
return;
}
push_down(node,l,r);int mid=(l+r)>>1;
if(mid>=ql)modify(lson,l,mid,ql,qr,val);
if(mid+1<=qr)modify(rson,mid+1,r,ql,qr,val);
push_up(node);
}
inline int query_max(int node,int l,int r,int ql,int qr)
{
if(ql<=l&&r<=qr)return trm[node];
push_down(node,l,r);int mid=(l+r)>>1,res=-T;
if(mid>=ql)res=max(res,query_max(lson,l,mid,ql,qr));
if(mid+1<=qr)res=max(res,query_max(rson,mid+1,r,ql,qr));
return res;
}
inline int query_sum(int node,int l,int r,int ql,int qr)
{
if(ql<=l&&r<=qr)return trs[node];
push_down(node,l,r);int mid=(l+r)>>1,res=0;
if(mid>=ql)res+=query_sum(lson,l,mid,ql,qr);
if(mid+1<=qr)res+=query_sum(rson,mid+1,r,ql,qr);
return res;
} #undef lson
#undef rson #define cur edges[i].to
inline void dfs1(int x)
{
size[x]=1;
for(register int i=head[x];i;i=edges[i].nxt)
if(cur!=fa[x])
{
fa[cur]=x;dep[cur]=dep[x]+1;
dfs1(cur);size[x]+=size[cur];
if(size[son[x]]<size[cur])son[x]=cur;
}
}
inline void dfs2(int x,int t)
{
idx[x]=++tot;top[x]=t;rk[tot]=x;
if(son[x])dfs2(son[x],t);
for(register int i=head[x];i;i=edges[i].nxt)
if(cur!=fa[x]&&cur!=son[x])dfs2(cur,cur);
}
#undef cur inline int link_max(int a,int b)
{
int res=-T;
while(top[a]!=top[b])
{
if(dep[top[a]]<dep[top[b]])swap(a,b);
res=max(res,query_max(1,1,n,idx[top[a]],idx[a]));
a=fa[top[a]];
}
if(dep[a]>dep[b])swap(a,b);
res=max(res,query_max(1,1,n,idx[a],idx[b]));
return res;
}
inline int link_sum(int a,int b)
{
int res=0;
while(top[a]!=top[b])
{
if(dep[top[a]]<dep[top[b]])swap(a,b);
res+=query_sum(1,1,n,idx[top[a]],idx[a]);
a=fa[top[a]];
}
if(dep[a]>dep[b])swap(a,b);
res+=query_sum(1,1,n,idx[a],idx[b]);
return res;
}
int main()
{
n=read();
for(register int i=1;i<n;i++)
{
int u,v;u=read();v=read();
addEdge(u,v);addEdge(v,u);
}
for(register int i=1;i<=n;i++)w[i]=read();
dfs1(1);dfs2(1,1);build(1,1,n);
q=read();
for(register int i=1;i<=q;i++)
{
scanf("%s",s+1);
if(s[1]=='C')
{
int u,t;u=read();t=read();
modify(1,1,n,idx[u],idx[u],t);
}else
{
int u,v;u=read();v=read();
if(s[2]=='M')
printf("%d\n",link_max(u,v));
else
printf("%d\n",link_sum(u,v));
}
}
}

最新文章

  1. [2014.01.27]wfPng 水印贴图组件 2.1
  2. Software Testing hw1
  3. hasLayout与BFC的触发条件
  4. Fresco 源码分析(二) Fresco客户端与服务端交互(2) Fresco.initializeDrawee()分析 续
  5. windows phone 7 中怎样定义和使用资源(Resource)
  6. [转]Android WebView播放视频(包括全屏播放),androidwebview
  7. js标签放在html的什么位置比较好
  8. AdbWinApi编译详解(本人亲历)
  9. 开源纯C#工控网关+组态软件(七)数据采集与归档
  10. Linuxc - Makefile完成项目的管理。
  11. 转: JavaScript 获取对象属性和方法
  12. 什么 是JavaScript中的字符串类型之间的转换问题详解? 部分4
  13. 基于word2vec训练词向量(二)
  14. linux进程管理之作业控制
  15. Unable to locate Spring NamespaceHandler for XML schema namespace [http://www.springframework.org/schema/tx]
  16. java的几种对象(PO,VO,DAO,BO,POJO)解释 【转】
  17. Mac - Hexo+GitHub轻松搭建自己的博客
  18. h5直播
  19. 弗洛伊德算法(Floyd算法)
  20. 练习:自己写一个容器ArrayList集合 一一数组综合练习2

热门文章

  1. unctf2020 部分简单题题解
  2. 仅一年工作经验成功跳槽字节跳动,腾讯并拿到字节的offer,全靠这份面经!
  3. 如何卸载MathType 7?
  4. 详解FL Studio压缩器——Fruity Limiter(下)
  5. 【PYTEST】第三章参数化
  6. 从执行上下文角度重新理解.NET(Core)的多线程编程[2]:同步上下文
  7. So Easy! HDU - 4565
  8. 程序员说:为什么喜欢大量使用 if……else if替代switch?
  9. Feign 超时设置
  10. 老哥们,请问我做的对么?(记一次失败的st表乱搞)