bzoj 3531 [SDOI2014]旅行

3531: [Sdoi2014]旅行

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 1034   Solved: 502
[ 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

Round 1 Day 1

题解：树链剖分+动态开点

就是针对每一个宗教都分别建立线段树，记录最大值和总和

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 100003
#define M 10000005
using namespace std;
int m,n,w[N],c[N],next[N*2],point[N],v[N*2],tot;
int deep[N],size[N],belong[N],vl[N],pos[N],sz,tree;
int fa[N][20],mi[20];
int ls[M],rs[M],root[M],tmax[M],ts[M];
void add(int x,int y)
{
  tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y;
  tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void build(int now,int f,int depth)
{
 deep[now]=depth; size[now]=1;
 for (int i=1;i<=17;i++)
  {
  	if (deep[now]-mi[i]<0) break;
  	fa[now][i]=fa[fa[now][i-1]][i-1];
  }	
 for (int i=point[now];i;i=next[i])
  if (v[i]!=f)
   {
   	 fa[v[i]][0]=now;
   	 build(v[i],now,depth+1);
   	 size[now]+=size[v[i]];
   } 
}
void dfs(int k,int chain)
{
  belong[k]=chain;  pos[k]=++sz;
  int l=0;
  for (int i=point[k];i;i=next[i])
   if (size[v[i]]>size[l]&&v[i]!=fa[k][0])  l=v[i];
  if (!l) return;
  dfs(l,chain);
  for (int i=point[k];i;i=next[i] )
   if (v[i]!=l&&v[i]!=fa[k][0]) dfs(v[i],v[i]); 
}
void update(int x)
{
 tmax[x]=max(tmax[ls[x]],tmax[rs[x]]);
 ts[x]=ts[ls[x]]+ts[rs[x]];
}
void pointchange(int &k,int l,int r,int x,int v)
{
  if (!k) k=++tree;
  if (l==r)
   {
   	 tmax[k]=ts[k]=v; return ;
   }
  int mid=(l+r)/2;
  if (x<=mid)
   pointchange(ls[k],l,mid,x,v);
  else
   pointchange(rs[k],mid+1,r,x,v);
  update(k);
}
int qjsum(int k,int l,int r,int ll,int rr)
{
  int ans=0;
  if (!k) return 0;
  if (ll<=l&&rr>=r)
   return ts[k];
  int mid=(l+r)/2;
  if (ll<=mid)
   ans+=qjsum(ls[k],l,mid,ll,rr);
  if (rr>mid)
   ans+=qjsum(rs[k],mid+1,r,ll,rr);
  return ans;
}
int qjmax(int k,int l,int r,int ll,int rr)
{
 int ans=0;
 if (!k) return 0;
 if (ll<=l&&rr>=r)
  {
  return tmax[k];
  }
 int mid=(l+r)/2;
 if (ll<=mid)
  ans=max(ans,qjmax(ls[k],l,mid,ll,rr));
 if (rr>mid)
  ans=max(ans,qjmax(rs[k],mid+1,r,ll,rr));
 return ans;
}
int lca(int x,int y)
{
 if (deep[x]<deep[y]) swap(x,y);
 int k=deep[x]-deep[y];
 for (int i=0;i<=17;i++)
  if (k>>i&1)  x=fa[x][i];
 if (x==y) return x;
 for (int i=17;i>=0;i++)
  if (fa[x][i]!=fa[y][i])
   x=fa[x][i],y=fa[y][i];
  return fa[x][0];
}
int solvesum(int k,int x,int y)
{
  int ans=0;
  while (belong[x]!=belong[y])
   {
   	 if(deep[belong[x]]<deep[belong[y]]) swap(x,y);
   	 ans+=qjsum(k,1,n,pos[belong[x]],pos[x]);
   	 x=fa[belong[x]][0];
   }
  if (deep[x]>deep[y]) swap(x,y);
  ans+=qjsum(k,1,n,pos[x],pos[y]);
  return ans;
}
int solvemax(int k,int x,int y)
{
  int ans=0;
  while(belong[x]!=belong[y])
   {
   	if (deep[belong[x]]<deep[belong[y]]) swap(x,y);
	ans=max(ans,qjmax(k,1,n,pos[belong[x]],pos[x]));
	x=fa[belong[x]][0]; 
   }
  if(deep[x]>deep[y]) swap(x,y);
  ans=max(ans,qjmax(k,1,n,pos[x],pos[y]));
  return ans;
}
int main()
{
  //freopen("a.in","r",stdin);
  //freopen("a.out","w",stdout);
  scanf("%d%d",&n,&m);
  for (int i=1;i<=n;i++) scanf("%d%d",&w[i],&c[i]);
  for (int i=1;i<n;i++)
   {
   	 int x,y; scanf("%d%d",&x,&y);
   	 add(x,y);
   }
  mi[0]=1; for (int i=1;i<18;i++) mi[i]=mi[i-1]*2;
  build(1,0,1);
  dfs(1,1);
  for (int i=1;i<=n;i++)
   pointchange(root[c[i]],1,n,pos[i],w[i]);
  for (int i=1;i<=m;i++)
   {
   	 char s[10]; int x,y; scanf("%s%d%d",s,&x,&y);
   	 if (s[1]=='C')
   	  { pointchange(root[c[x]],1,n,pos[x],0);
		c[x]=y;
		pointchange(root[c[x]],1,n,pos[x],w[x]);
	  }
   	 else
   	  if (s[1]=='W')
   	   {	pointchange(root[c[x]],1,n,pos[x],y); w[x]=y; }
   	  else
   	   if (s[1]=='S')
   	   {
   	   	 printf("%d\n",solvesum(root[c[x]],x,y));
   	   }
   	   else
   	    {
   	      printf("%d\n",solvemax(root[c[x]],x,y));
   	    }
   }
}