洛谷-3313 [SDOI2014]旅行

mac2026-09-27  10

题目描述 S国有N个城市,编号从1到N。城市间用N-1条双向道路连接,满足从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教,如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。 为了方便,我们用不同的正整数代表各种宗教, S国的居民常常旅行。旅行时他们总会走最短路,并且为了避免麻烦,只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级,旅行者们常会记下途中(包括起点和终点)留宿过的城市的评级总和或最大值。 在S国的历史上常会发生以下几种事件: “CC x c“:城市x的居民全体改信了c教; “CW x w“:城市x的评级调整为w; “QS x y“:一位旅行者从城市x出发,到城市y,并记下了途中留宿过的城市的评级总和; “QM x y“:一位旅行者从城市x出发,到城市y,并记下了途中留宿过的城市的评级最大值。 由于年代久远,旅行者记下的数字已经遗失了,但记录开始之前每座城市的信仰与评级,还有事件记录本身是完好的。请根据这些信息,还原旅行者记下的数字。 为了方便,我们认为事件之间的间隔足够长,以致在任意一次旅行中,所有城市的评级和信仰保持不变。 输入格式 输入的第一行包含整数N,Q依次表示城市数和事件数。 接下来N行,第i+l行两个整数Wi,Ci依次表示记录开始之前,城市i的评级和信仰。 接下来N-1行每行两个整数x,y表示一条双向道路。 接下来Q行,每行一个操作,格式如上所述。 输出格式 对每个QS和QM事件,输出一行,表示旅行者记下的数字。

输入输出样例 输入 #1 5 6 3 1 2 3 1 2 3 3 5 1 1 2 1 3 3 4 3 5 QS 1 5 CC 3 1 QS 1 5 CW 3 3 QS 1 5 QM 2 4

输出 #1 8 9 11 3

说明/提示 N , Q < = 1 0 5 , C < = 1 0 5 N,Q < =10^5 , C < =10^5 N,Q<=105,C<=105

解释:树链剖分直接上模板,不同宗教不同线段树来维护,同时我们进行动态开点,就可以保证通过

#include <cstdio> #include <iostream> #define il inline const int maxm=1e6+1; struct tree{ int ls,rs,maxx,sum; }; int root[maxm],rt; tree st[maxm*4]; int deep[maxm],son[maxm],fa[maxm],size[maxm],top[maxm]; int head[maxm],to[maxm*2],net[maxm*2],cnt; int yval[maxm],val[maxm],yc[maxm],c[maxm]; int xz[maxm]; int id[maxm],who[maxm],tot; void add_edge(int x,int y){ to[++cnt]=y,net[cnt]=head[x],head[x]=cnt; } int dfs1(int x,int fax,int dep){ fa[x]=fax,deep[x]=dep,size[x]=1; int maxson=-1; for(int i=head[x];i;i=net[i]) if(to[i]!=fax){ size[x]+=dfs1(to[i],x,dep+1); if(maxson<size[to[i]]) maxson=size[to[i]],son[x]=to[i]; } return size[x]; } void dfs2(int x,int topx){ id[x]=++tot; who[tot]=x; val[tot]=yval[x],c[tot]=yc[x]; top[x]=topx; if(!son[x]) return; dfs2(son[x],topx); for(int i=head[x];i;i=net[i]) if(!id[to[i]]) dfs2(to[i],to[i]); } void update(int o){ st[o].maxx=std::max(st[st[o].ls].maxx,st[st[o].rs].maxx); st[o].sum=st[st[o].ls].sum+st[st[o].rs].sum; } void mofidy(int &o,int l,int r,int ind,int num){ if(!o) o=++rt; if(l>=r){ st[o].maxx=st[o].sum=num; return; } int mid=(l+r)>>1; if(ind<=mid) mofidy(st[o].ls,l,mid,ind,num); else mofidy(st[o].rs,mid+1,r,ind,num); update(o); } int askmax(int o,int l,int r,int ql,int qr){ if(!o) return -1; if(r<ql||l>qr) return -1; if(ql<=l&&r<=qr) return st[o].maxx; int mid=(l+r)>>1; return std::max(askmax(st[o].ls,l,mid,ql,qr),askmax(st[o].rs,mid+1,r,ql,qr)); } int asksum(int o,int l,int r,int ql,int qr){ if(!o) return 0; if(r<ql||l>qr) return 0; if(ql<=l&&r<=qr) return st[o].sum; int mid=(l+r)>>1; return asksum(st[o].ls,l,mid,ql,qr)+asksum(st[o].rs,mid+1,r,ql,qr); } int tree_max(int u,int v){ int ans=-1; int xz=yc[v]; while(top[u]!=top[v]){ if(deep[top[u]]<deep[top[v]]) std::swap(u,v); ans=std::max(ans,askmax(root[xz],1,tot,id[top[u]],id[u])); u=fa[top[u]]; } if(deep[u]<deep[v]) std::swap(u,v); ans=std::max(ans,askmax(root[xz],1,tot,id[v],id[u])); return ans; } int tree_sum(int u,int v){ int ans=0; int xz=yc[v]; while(top[u]!=top[v]){ if(deep[top[u]]<deep[top[v]]) std::swap(u,v); ans+=asksum(root[xz],1,tot,id[top[u]],id[u]); u=fa[top[u]]; } if(deep[u]<deep[v]) std::swap(u,v); ans+=asksum(root[xz],1,tot,id[v],id[u]); return ans; } int read(){ int x=0,w=1;char ch=0; while(ch<'0'||ch>'9'){ if(ch=='-') w=-1; ch=getchar(); } while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar(); return x*w; } int main(){ int n,q; n=read(),q=read(); for(int i=1;i<=n;i++) yval[i]=read(),yc[i]=read(); for(int i=1;i<n;i++){ int u=read(),v=read(); add_edge(u,v); add_edge(v,u); } dfs1(1,0,1),dfs2(1,1); for(int i=1;i<=n;i++) mofidy(root[yc[who[i]]],1,tot,i,yval[who[i]]); for(int i=1;i<=q;i++){ char s[20]; int x,y; scanf("%s",s); x=read(),y=read(); if(s[0]=='C'&&s[1]=='C'){ mofidy(root[yc[x]],1,tot,id[x],0); yc[x]=y; mofidy(root[yc[x]],1,tot,id[x],yval[x]); } if(s[0]=='C'&&s[1]=='W') yval[x]=y,mofidy(root[yc[x]],1,tot,id[x],y); if(s[0]=='Q'&&s[1]=='M') printf("%d\n",tree_max(x,y)); if(s[0]=='Q'&&s[1]=='S') printf("%d\n",tree_sum(x,y)); } return 0; }
最新回复(0)