社区讨论
树剖31pts求调教
P3038[USACO11DEC] Grass Planting G参与者 1已保存回复 0
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 0 条
- 当前快照
- 1 份
- 快照标识符
- @lr7jk4fv
- 此快照首次捕获于
- 2024/01/10 16:51 2 年前
- 此快照最后确认于
- 2024/01/10 16:53 2 年前
CPP
#include<bits/stdc++.h>
#define int long long
#define double long double
#define INF INT_MAX
using namespace std;
const int maxn=1e5+5;
int n,q,cnt;
int a[maxn];
int f[maxn],deap[maxn],siz[maxn],son[maxn];
int id[maxn],p[maxn],top[maxn],dfn[maxn];
vector<int> g[maxn];
set<int> ans[maxn];
void dfs1(int dq,int fa,int sum){
f[dq]=fa,deap[dq]=sum,siz[dq]=1;
int maxson=-1;
for(int i=0;i<g[dq].size();i++){
if(g[dq][i]==fa) continue;
dfs1(g[dq][i],dq,sum+1);
siz[dq]+=siz[g[dq][i]];
if(maxson==-1 || siz[g[dq][i]]>siz[maxson]) maxson=g[dq][i],son[dq]=g[dq][i];
}
return;
}
void dfs2(int dq,int fa){
id[dq]=++cnt,dfn[cnt]=dq,p[cnt]=a[dq],top[dq]=fa;
if(!son[dq]) return;
dfs2(son[dq],fa);
for(int i=0;i<g[dq].size();i++){
if(g[dq][i]==f[dq] || g[dq][i]==son[dq]) continue;
dfs2(g[dq][i],g[dq][i]);
}
return;
}
struct Tree{
int val,add;
#define sum(x) t[x].val
#define add(x) t[x].add
}t[maxn*4];
void up_date(int dq){sum(dq)=sum(dq*2)+sum(dq*2+1);}
void build(int dq,int l,int r){
if(l==r) return;
int mid=(l+r)/2;
build(dq*2,l,mid);
build(dq*2+1,mid+1,r);
up_date(dq);
}
void spread(int dq,int l,int r){
if(add(dq)){
int mid=(l+r)/2;
sum(dq*2)=(mid-l+1)*add(dq);
sum(dq*2+1)=(r-mid)*add(dq);
add(dq*2)+=add(dq);
add(dq*2+1)+=add(dq);
add(dq)=0;
}
}
void change(int dq,int ll,int rr,int l,int r,int d){
if(l<=ll && rr<=r){
sum(dq)+=d*(rr-ll+1);
add(dq)+=d;
return;
}
spread(dq,ll,rr);
int mid=(ll+rr)/2;
if(l<=mid) change(dq*2,ll,mid,l,r,d);
if(r>mid) change(dq*2+1,mid+1,rr,l,r,d);
up_date(dq);
}
int ask(int dq,int ll,int rr,int l,int r){
if(l<=ll && rr<=r) return sum(dq);
spread(dq,ll,rr);
int res=0,mid=(ll+rr)/2;
if(l<=mid) res+=ask(dq*2,ll,mid,l,r);
if(r>mid) res+=ask(dq*2+1,mid+1,rr,l,r);
return res;
}
void changee(int x,int y,int d){
while(top[x]!=top[y]){
if(deap[top[x]]<deap[top[y]]){
change(1,1,n,id[top[y]],id[y],d);
y=f[top[y]];
}
else{
change(1,1,n,id[top[x]],id[x],d);
x=f[top[x]];
}
}
if(deap[x]<deap[y]) change(1,1,n,id[x]+1,id[y],d);
else change(1,1,n,id[y]+1,id[x],d);
}
int askk(int x,int y){
int res=0;
while(top[x]!=top[y]){
if(deap[top[x]]<deap[top[y]]){
res+=ask(1,1,n,id[top[y]],id[y]);
y=f[top[y]];
}
else{
res+=ask(1,1,n,id[top[x]],id[x]);
x=f[top[x]];
}
}
if(deap[x]<deap[y]) res+=ask(1,1,n,id[x]+1,id[y]);
else res+=ask(1,1,n,id[y]+1,id[x]);
return res;
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin>>n>>q;
for(int i=1;i<n;i++){
int u,v;
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
dfs1(1,0,1);
dfs2(1,1);
build(1,1,n);
while(q--){
char op;
int x,y;
cin>>op>>x>>y;
if(op=='P') changee(x,y,1);
if(op=='Q') cout<<askk(x,y)<<endl;
}
return 0;
}
回复
共 0 条回复,欢迎继续交流。
正在加载回复...