bzoj3531——树链剖分+动态开点线段树

3531: [Sdoi2014]旅行

Time Limit: 20 Sec  Memory Limit: 512 MB

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。

 =============================================================

开始日期：17.3.24

结束日期：17.3.29

树上的修改和路径上信息的统计很容易想到树剖，统计对于信仰有要求，每种信仰开一颗线段树来维护就好了。

为什么写了六天这么久呢？这题从一开始就TLE，写错的地方有点多。

算lca的时候少打了个f[]，每次多算了一个点，调出来了。

线段树数组又双叒叕开小了。

听说#define max(x,y) x>y?x:y 跑得比香港记者快，于是：

1 //...
2 #define max(x,y) x>y?x:y
3 //...
4 
5 void get(...){
6     //...
7     return max(get(ls),get(rs))
8 }

强行把常数扩大一倍。。。

时限20s，AC代码10.6s。所以这样T掉了。

一定要这样写的话可以改成：

#define max(x,y) (xxx=x>yyy=y)?xxx:yyy
int xxx,yyy;

 

AC code:

  1 #include<cstring>
  2 #include<cstdio>
  3 #include<algorithm>
  4 #include<cmath>
  5 #define clr(a) memset(a,0,sizeof(a));
  6 #define foru(i,x,y) for(int i=x;i<=y;i++)
  7 using namespace std;
  8 const int N=100100;
  9 struct edge{int to,nxt;}e[N*2];
 10 struct node{int s,m,lson,rson;}t[10000010];
 11 int f[N],d[N],w[N],c[N],id[N],top[N],siz[N],son[N],head[N],r[N];
 12 int ne,n,q,cnt,nt;
 13 
 14 void add(int a,int b){
 15     e[++ne]=(edge){b,head[a]};head[a]=ne;
 16 }
 17 
 18 void dfs(int k,int fa,int dep){
 19     f[k]=fa;d[k]=dep;siz[k]=1;son[k]=0;
 20     for(int i=head[k];i;i=e[i].nxt){
 21         int v=e[i].to;
 22         if(v==fa)continue; 
 23         dfs(v,k,dep+1);
 24         siz[k]+=siz[v];
 25         if(siz[v]>siz[son[k]])son[k]=v;
 26     }
 27 }
 28 
 29 void build(int k,int tp){
 30     id[k]=++cnt;
 31     top[k]=tp; 
 32     if(son[k])build(son[k],tp);
 33     for(int i=head[k];i;i=e[i].nxt)
 34         if(e[i].to!=son[k]&&e[i].to!=f[k])
 35             build(e[i].to,e[i].to);
 36 }
 37 
 38 #define mid ((L+R)>>1)
 39 #define ls t[k].lson
 40 #define rs t[k].rson
 41 
 42 void pu(int k){t[k].s=t[ls].s+t[rs].s;t[k].m=max(t[ls].m,t[rs].m);}
 43 
 44 void update(int &k,int L,int R,int p,int x){
 45     if(!k)k=++nt;
 46     if(p>R||p<L)return;
 47     if(L==R){t[k].s=t[k].m=x;return;}
 48     update(ls,L,mid,p,x); update(rs,mid+1,R,p,x);
 49     pu(k);
 50 }
 51 
 52 int getsum(int k,int L,int R,int l,int r){
 53     if(!k)return 0;
 54     if(r<L||l>R)return 0;
 55     if(l<=L&&R<=r)return t[k].s;
 56     return getsum(ls,L,mid,l,r)+getsum(rs,mid+1,R,l,r);
 57 }
 58 
 59 int getmax(int k,int L,int R,int l,int r){
 60     if(!k)return 0;
 61     if(r<L||l>R)return 0;
 62     if(l<=L&&R<=r)return t[k].m;
 63     return max(getmax(ls,L,mid,l,r),getmax(rs,mid+1,R,l,r));
 64 }
 65 
 66 int finds(int x,int y){
 67     int ans=0,cc=c[x];
 68     while(top[x]!=top[y]){
 69         if(d[top[x]]<d[top[y]])swap(x,y);//
 70         ans+=getsum(r[cc],1,cnt,id[top[x]],id[x]);
 71         x=f[top[x]];//
 72     }
 73     if(d[x]>d[y])swap(x,y);
 74     ans+=getsum(r[cc],1,cnt,id[x],id[y]);//
 75     return ans;
 76 }
 77 
 78 int findm(int x,int y){
 79     int ans=0,cc=c[x];
 80     while(top[x]!=top[y]){
 81         if(d[top[x]]<d[top[y]])swap(x,y);//
 82         ans=max(ans,getmax(r[cc],1,cnt,id[top[x]],id[x]));
 83         x=f[top[x]];//
 84     }
 85     if(d[x]>d[y])swap(x,y);
 86     ans=max(ans,getmax(r[cc],1,cnt,id[x],id[y]));//
 87     return ans;
 88 }
 89 
 90 char ch[10];
 91 int main(){
 92     int x,y;
 93     scanf("%d%d",&n,&q);
 94     foru(i,1,n)scanf("%d%d",w+i,c+i);
 95     foru(i,1,n-1){
 96         scanf("%d%d",&x,&y);
 97         add(x,y);add(y,x);
 98     }
 99     dfs(1,0,1);
100     build(1,1);
101     foru(i,1,n)update(r[c[i]],1,cnt,id[i],w[i]);
102     foru(i,1,q){
103         scanf("%s%d%d",ch,&x,&y);
104         if(ch[0]=='C'){
105             if(ch[1]=='C'){
106                 update(r[c[x]],1,cnt,id[x],0);
107                 c[x]=y;
108                 update(r[c[x]],1,cnt,id[x],w[x]);
109             }else{
110                 update(r[c[x]],1,cnt,id[x],y);
111                 w[x]=y;
112             }
113         }else{
114             if(ch[1]=='S'){
115                 printf("%d\n",finds(x,y));
116             }
117             else{
118                 printf("%d\n",findm(x,y));
119             }
120         }
121     }
122 }

 

posted @ 2017-03-30 13:34  羊毛羊  阅读(349)  评论(0)  编辑  收藏  举报