bzoj 3531(动态加点线段树，树链剖分)

3531: [Sdoi2014]旅行

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 1331  Solved: 629

[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<cstdio>

#include<iostream>

#include<algorithm>

#include<cstring>

using namespace std;

int n,q,len,cnt,ans,sug;

int to[210000],next[210000],h[210000];

int size[110000],deep[110000],fa[110000],son[110000],dui[110000],top[110000];

int ma[10100000],sum[10100000],w[210000],root[210000],c[210000];

int l[10100000],r[10100000];

char ch[3];

inline int read()

{
char y; int x=0,f=1; y=getchar();
while (y<'0' || y>'9') {if (y=='-')f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;

}

void insert(int x,int y)

 {

  ++len; to[len]=y; next[len]=h[x]; h[x]=len;

 }

void dfs(int now,int from,int dep)

 {

  size[now]=1; deep[now]=dep; fa[now]=from; son[now]=0;

  int u=h[now];

  while (u!=0)

  {

   if (to[u]!=from)

    {

    dfs(to[u],now,dep+1);

    size[now]+=size[to[u]]; if (size[to[u]]>size[son[now]]) son[now]=to[u];
}
 u=next[u];
}

 }

void dfs1(int now,int tg)

 {

  ++cnt;

  if (tg==-1) tg=now; top[now]=tg; dui[now]=cnt;

  if (son[now]==0) return;

  dfs1(son[now],tg);

  int u=h[now];

  while (u!=0)

  {

  if (to[u]!=fa[now] && to[u]!=son[now])

  {

  dfs1(to[u],-1);
  }
u=next[u];
 }

 }

void pushup(int now)

 {

  ma[now]=max(ma[l[now]],ma[r[now]]);

 }

void add(int &now,int lg,int rg,int zhi,int opp)

 {

  if (now==0) now=++sug; sum[now]+=zhi;

  if (lg==rg)

  {

  ma[now]+=zhi; return;
 }

  int mid=(lg+rg)/2;

  if (opp>mid) add(r[now],mid+1,rg,zhi,opp);else
 add(l[now],lg,mid,zhi,opp);
pushup(now); 

 }

void qusum(int now,int lg,int rg,int lo,int ro)

 {

  if (lo<=lg && rg<=ro) {ans+=sum[now];return;} 

  int mid=(lg+rg)/2;

  if (mid>=lo) qusum(l[now],lg,mid,lo,ro);

  if (mid+1<=ro) qusum(r[now],mid+1,rg,lo,ro);

 }

void qumax(int now,int lg,int rg,int lo,int ro)

 {

  if (lo<=lg && rg<=ro) {ans=max(ans,ma[now]);return;} 

  int mid=(lg+rg)/2;

  if (mid>=lo) qumax(l[now],lg,mid,lo,ro);

  if (mid+1<=ro) qumax(r[now],mid+1,rg,lo,ro);

 }

void getsum(int x,int y)

 {

  int opp=w[x];

  if (deep[x]<deep[y]) swap(x,y);

  ans=0;

  while (top[x]!=top[y])

  {

    if (deep[top[x]]>deep[top[y]]) qusum(root[opp],1,n,dui[top[x]],dui[x]),x=fa[top[x]];else

    qusum(root[opp],1,n,dui[top[y]],dui[y]),y=fa[top[y]];
}
qusum(root[opp],1,n,dui[y],dui[x]);
printf("%d\n",ans);

 }

void getmax(int x,int y)

 {

  int opp=w[x];

  if (deep[x]<deep[y]) swap(x,y);

    ans=0;

  while (top[x]!=top[y])

  {

    if (deep[top[x]]>deep[top[y]]) qumax(root[opp],1,n,dui[top[x]],dui[x]),x=fa[top[x]];else

    qumax(root[opp],1,n,dui[top[y]],dui[y]),y=fa[top[y]];
}
qumax(root[opp],1,n,dui[y],dui[x]);
printf("%d\n",ans);

 }

int main()

{
cnt=0;
n=read(); q=read();
for (int i=1;i<=n;++i)
{
c[i]=read(); w[i]=read();
}
for (int i=1;i<=n-1;++i)
{
int x,y; x=read(); y=read();
insert(x,y); insert(y,x);
}
sug=0;
dfs(1,0,0);
dfs1(1,-1);
for (int i=1;i<=n;++i)
{
add(root[w[i]],1,n,c[i],dui[i]);
}
for (int i=1;i<=q;++i)
{
scanf("%s",ch); int x,y; x=read(); y=read();
if (ch[0]=='C')
{
if (ch[1]=='C')
{
  add(root[w[x]],1,n,-c[x],dui[x]);
  w[x]=y;
  add(root[w[x]],1,n,c[x],dui[x]);

}else
 {
   add(root[w[x]],1,n,-c[x],dui[x]);
c[x]=y; 
add(root[w[x]],1,n,c[x],dui[x]); 
 }
 }else
  {
  if (ch[1]=='S')
   {
   
getsum(x,y);
   }else
    {
   
getmax(x,y);

  }
  }
}

}