洛谷 P4114 Qtree1

Qtree系列都跟树有着莫大的联系,这道题当然也不例外

我是题面

读完题,我们大概就知道了,这道题非常简单,可以说是模板题。树剖+线段树轻松解决

直接看代码吧

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cctype>
#define ll long long
#define gc() getchar()
#define maxn 100010
using namespace std; inline ll read(){ //很朴素的快读,不多解释
ll a=0;int f=0;char p=gc();
while(!isdigit(p)){f|=p=='-';p=gc();}
while(isdigit(p)){a=(a<<3)+(a<<1)+(p^48);p=gc();}
return f?-a:a;
}
void write(ll a){ //很朴素的快写,也不多解释
if(a>9)write(a/10);
putchar(a%10+'0');
}
int n,a[maxn]; //a数组表示点与父亲连边的长度 int tot,head[maxn];
struct ahaha1{ //前向星存边
int w,to,next;
}e[maxn<<1];
inline void add(int u,int v,int w){
e[tot].w=w;e[tot].to=v;e[tot].next=head[u];head[u]=tot++;
} int f[maxn],sz[maxn],dep[maxn],son[maxn]; //f表示点的父亲,sz表示点的子树上节点个数,dep表示节点的深度,son表示子树最大的子节点
void dfs(int u,int fa){
int maxa=0;sz[u]=1;
for(int i=head[u];~i;i=e[i].next){
int v=e[i].to;if(v==fa)continue;
f[v]=u;dep[v]=dep[u]+1;a[v]=e[i].w;
dfs(v,u);sz[u]+=sz[v];
if(sz[v]>maxa)maxa=sz[v],son[u]=v;
}
}
int b[maxn],in[maxn],top[maxn]; //b表示当前编号所指向的点,in表示点的编号,top表示点所在链的顶端
void dfs(int u,int fa,int topf){
in[u]=++tot;b[tot]=u;top[u]=topf;
if(!son[u])return;
dfs(son[u],u,topf);
for(int i=head[u];~i;i=e[i].next){
int v=e[i].to;if(v==fa||v==son[u])continue;
dfs(v,u,v);
}
} #define lc p<<1
#define rc p<<1|1
struct ahaha2{ //线段树太简单不解释
int v;
}t[maxn<<2];
inline void pushup(int p){
t[p].v=max(t[lc].v,t[rc].v);
}
void build(int p,int l,int r){
if(l==r){t[p].v=a[b[l]];return;}
int m=l+r>>1;
build(lc,l,m);build(rc,m+1,r);
pushup(p);
}
void update(int p,int l,int r,int L,int z){
if(l==r){t[p].v=z;return;}
int m=l+r>>1;
if(m>=L)update(lc,l,m,L,z);
else update(rc,m+1,r,L,z);
pushup(p);
}
int query(int p,int l,int r,int L,int R){
if(l>R||r<L)return 0;
if(L<=l&&r<=R)return t[p].v;
int m=l+r>>1;
return max(query(lc,l,m,L,R),query(rc,m+1,r,L,R));
} inline void solve_1(){ //从第0条边开始存,每条边存两次,所以输入的第i条边对应的是第2*i-2和第2*i-1条边,谁是儿子改谁
int x=read()*2-2,w=read(),u=e[x].to,v=e[x+1].to;
if(f[v]==u)update(1,1,n,in[v],w);
else update(1,1,n,in[u],w);
}
inline void solve_2(){ //链上查询
int x=read(),y=read(),ans=0;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
ans=max(ans,query(1,1,n,in[top[x]],in[x])); //由于存的是与父亲的距离,而top的父亲直接取所以此处无需加1
x=f[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
ans=max(ans,query(1,1,n,in[x]+1,in[y])); //但是最后一次存储必须加1否则会多询问一条边
write(ans);putchar('\n');
} int main(){memset(head,-1,sizeof head);
n=read();
for(int i=1;i<n;++i){
int x=read(),y=read(),w=read();
add(x,y,w);add(y,x,w);
}
tot=0;dfs(1,-1);dfs(1,-1,1);
build(1,1,n);
char z[10];
while(1){
scanf("%s",z);
if(z[0]=='D')break;
switch(z[0]){
case 'C':solve_1();break;
case 'Q':solve_2();break;
}
}
return 0;
}
上一篇:洛谷 P4114 Qtree1 树链剖分


下一篇:javascript 模拟日历