3531: [Sdoi2014]旅行
Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 850 Solved: 433
[Submit][Status][Discuss]
Description
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。并记下了途中留宿过
的城市的评级最大值。
因为年代久远,旅行者记下的数字已经遗失了。但记录開始之前每座城市的信仰与评级,还有事件记录本身是完善的。请依据这些信息,还原旅行者记下的数字。 为了方便,我们觉得事件之间的间隔足够长,以致在随意一次旅行中,全部城市的评级和信仰保持不变。
Input
输入的第一行包括整数N,Q依次表示城市数和事件数。
接下来N行,第i+l行两个整数Wi。Ci依次表示记录開始之前,城市i的
评级和信仰。
接下来N-1行每行两个整数x,y表示一条双向道路。
接下来Q行,每行一个操作,格式如上所述。
Output
对每一个QS和QM事件。输出一行,表示旅行者记下的数字。
Sample Input
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
Sample Output
8
9
11
3
HINT
N,Q < =10^5 , C < =10^5
数据保证对全部QS和QM事件,起点和终点城市的信仰同样;在随意时
刻,城市的评级总是不大于10^4的正整数,且宗教值不大于C。
Source
树链剖分,对于每一种信仰建一棵线段树,动态开点。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 100005
#define maxm 10000005
using namespace std;
int n,m,cnt,tot,x,y;
int p[maxn],sz[maxn],head[maxn],belong[maxn],d[maxn],f[maxn][17];
int w[maxn],c[maxn],rt[maxn],ls[maxm],rs[maxm],mx[maxm],sum[maxm];
bool vst[maxn];
char ch[10];
struct edge_type
{
int next,to;
}e[maxn*2];
inline int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
inline void add_edge(int x,int y)
{
e[++cnt]=(edge_type){head[x],y};head[x]=cnt;
e[++cnt]=(edge_type){head[y],x};head[y]=cnt;
}
inline void dfs1(int x)
{
for(int i=1;i<=16;i++)
{
if ((1<<i)<=d[x]) f[x][i]=f[f[x][i-1]][i-1];
else break;
}
vst[x]=true;sz[x]=1;
for(int i=head[x];i;i=e[i].next)
{
int y=e[i].to;
if (vst[y]) continue;
d[y]=d[x]+1;
f[y][0]=x;
dfs1(y);
sz[x]+=sz[y];
}
}
inline void dfs2(int x,int chain)
{
p[x]=++tot;belong[x]=chain;
int k=0;
for(int i=head[x];i;i=e[i].next)
if (d[e[i].to]>d[x]&&sz[k]<sz[e[i].to]) k=e[i].to;
if (k) dfs2(k,chain);
for(int i=head[x];i;i=e[i].next)
if (d[e[i].to]>d[x]&&e[i].to!=k) dfs2(e[i].to,e[i].to);
}
inline int lca(int x,int y)
{
if (d[x]<d[y]) swap(x,y);
int t=int(log2(d[x]-d[y]));
D(i,t,0) if (d[x]-(1<<i)>=d[y]) x=f[x][i];
if (x==y) return x;
t=int(log2(d[x]));
D(i,t,0) if (f[x][i]!=f[y][i]){x=f[x][i];y=f[y][i];}
return f[x][0];
}
inline void pushup(int k)
{
mx[k]=max(mx[ls[k]],mx[rs[k]]);
sum[k]=sum[ls[k]]+sum[rs[k]];
}
inline void change(int &k,int l,int r,int x,int num)
{
if (!k) k=++tot;
if (l==r){mx[k]=sum[k]=num;return;}
int mid=(l+r)>>1;
if (x<=mid) change(ls[k],l,mid,x,num);
else change(rs[k],mid+1,r,x,num);
pushup(k);
}
inline int getmx(int k,int l,int r,int x,int y)
{
if (!k) return 0;
if (l==x&&r==y) return mx[k];
int mid=(l+r)>>1;
if (y<=mid) return getmx(ls[k],l,mid,x,y);
else if (x>mid) return getmx(rs[k],mid+1,r,x,y);
else return max(getmx(ls[k],l,mid,x,mid),getmx(rs[k],mid+1,r,mid+1,y));
}
inline int getsum(int k,int l,int r,int x,int y)
{
if (!k) return 0;
if (l==x&&r==y) return sum[k];
int mid=(l+r)>>1;
if (y<=mid) return getsum(ls[k],l,mid,x,y);
else if (x>mid) return getsum(rs[k],mid+1,r,x,y);
else return getsum(ls[k],l,mid,x,mid)+getsum(rs[k],mid+1,r,mid+1,y);
}
inline int solvemx(int c,int x,int tmp)
{
int mx=0;
while (belong[x]!=belong[tmp])
{
mx=max(mx,getmx(rt[c],1,n,p[belong[x]],p[x]));
x=f[belong[x]][0];
}
mx=max(mx,getmx(rt[c],1,n,p[tmp],p[x]));
return mx;
}
inline int solvesum(int c,int x,int tmp)
{
int sum=0;
while (belong[x]!=belong[tmp])
{
sum+=getsum(rt[c],1,n,p[belong[x]],p[x]);
x=f[belong[x]][0];
}
sum+=getsum(rt[c],1,n,p[tmp],p[x]);
return sum;
}
int main()
{
n=read();m=read();
F(i,1,n){w[i]=read();c[i]=read();}
F(i,1,n-1){x=read();y=read();add_edge(x,y);}
dfs1(1);dfs2(1,1);
tot=0;
F(i,1,n) change(rt[c[i]],1,n,p[i],w[i]);
F(i,1,m)
{
scanf("%s",ch);x=read();y=read();
if (ch[0]=='C')
{
if (ch[1]=='C')
{
change(rt[c[x]],1,n,p[x],0);change(rt[y],1,n,p[x],w[x]);
c[x]=y;
}
else{change(rt[c[x]],1,n,p[x],y);w[x]=y;}
}
else
{
int tmp=lca(x,y);
if (ch[1]=='S')
{
int ans=solvesum(c[x],x,tmp)+solvesum(c[x],y,tmp);
if (c[x]==c[tmp]) ans-=w[tmp];
printf("%d\n",ans);
}
else printf("%d\n",max(solvemx(c[x],x,tmp),solvemx(c[x],y,tmp)));
}
}
}