bzoj3307雨天的尾巴（可持久化线段树合并）

  1 #include<queue>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #define lll tr[spc].ls
  6 #define rrr tr[spc].rs
  7 const int maxin=1000000000;
  8 struct pnt{
  9     int fa[20];
 10     int hd;
 11     int gd;
 12     int dp;
 13     int wgt;
 14     int ans;
 15     int prp;
 16     int root;
 17 }p[110000];
 18 struct ent{
 19     int twd;
 20     int lst;
 21 }e[220000];
 22 struct ljt{
 23     int twd;
 24     int lst;
 25     int vls;
 26 }el[420000];
 27 struct trnt{
 28     int ls;
 29     int rs;
 30     int val;
 31     int mpl;
 32 }tr[2000000],org;
 33 int bin[2000000];
 34 int top;
 35 int siz;
 36 int cnt1;
 37 int cnt2;
 38 int n,m;
 39 void adde(int f,int t)
 40 {
 41     cnt1++;
 42     e[cnt1].twd=t;
 43     e[cnt1].lst=p[f].hd;
 44     p[f].hd=cnt1;
 45     return ;
 46 }
 47 void addc(int pt,int c,int v)
 48 {
 49     cnt2++;
 50     el[cnt2].twd=c;
 51     el[cnt2].lst=p[pt].gd;
 52     el[cnt2].vls=v;
 53     p[pt].gd=cnt2;
 54     return ;
 55 }
 56 namespace Sgt{
 57     void P_delete(int spc)
 58     {
 59         bin[++top]=spc;
 60         tr[spc]=org;
 61         return ;
 62     }
 63     void P_destory(int spc1,int spc2)
 64     {
 65         P_delete(spc1);
 66         P_delete(spc2);
 67         return ;
 68     }
 69     int P_new(void)
 70     {
 71         if(top)
 72             return bin[top--];
 73         return ++siz;
 74     }
 75     void pushup(int spc)
 76     {
 77         if(tr[lll].val>=tr[rrr].val)
 78             tr[spc].val=tr[lll].val,tr[spc].mpl=tr[lll].mpl;
 79         else
 80             tr[spc].val=tr[rrr].val,tr[spc].mpl=tr[rrr].mpl;
 81         return ;
 82     }
 83     int P_merge(int spc1,int spc2,int l,int r)
 84     {
 85         if(!spc1||!spc2)
 86             return spc1+spc2;
 87         int spc=P_new();
 88         if(l==r)
 89         {
 90             tr[spc].val=tr[spc1].val+tr[spc2].val;
 91             tr[spc].mpl=l;
 92             P_destory(spc1,spc2);
 93             return spc;
 94         }
 95         int mid=(l+r)>>1;
 96         tr[spc].ls=P_merge(tr[spc1].ls,tr[spc2].ls,l,mid);
 97         tr[spc].rs=P_merge(tr[spc1].rs,tr[spc2].rs,mid+1,r);
 98         pushup(spc);
 99         P_destory(spc1,spc2);
100         return spc;
101     }
102     void Update(int l,int r,int &spc,int pos,int v)
103     {
104         if(!spc)
105             spc=P_new();
106         if(l==r)
107         {
108             tr[spc].val+=v;
109             tr[spc].mpl=(tr[spc].val)?l:0;
110             return ;
111         }
112         int mid=(l+r)>>1;
113         if(pos<=mid)
114             Update(l,mid,tr[spc].ls,pos,v);
115         else
116             Update(mid+1,r,tr[spc].rs,pos,v);
117         pushup(spc);
118         if(!tr[spc].val)
119             tr[spc].mpl=0;
120         return ;
121     }
122 }
123 void Basic_dfs(int x,int f)
124 {
125     p[x].dp=p[f].dp+1;
126     p[x].fa[0]=f;
127     for(int i=1;i<=19;i++)
128         p[x].fa[i]=p[p[x].fa[i-1]].fa[i-1];
129     p[x].wgt=1;
130     for(int i=p[x].hd;i;i=e[i].lst)
131     {
132         int to=e[i].twd;
133         if(to==f)
134             continue;
135         Basic_dfs(to,x);
136         p[x].wgt+=p[to].wgt;
137     }
138     return ;
139 }
140 void Ans_dfs(int x,int f)
141 {
142     for(int i=p[x].hd;i;i=e[i].lst)
143     {
144         int to=e[i].twd;
145         if(to==f)
146             continue;
147         Ans_dfs(to,x);
148     }
149     for(int i=p[x].gd;i;i=el[i].lst)
150     {
151         int clr=el[i].twd;
152         Sgt::Update(1,maxin,p[x].root,clr,el[i].vls);
153     }
154     p[x].ans=tr[p[x].root].mpl;
155     p[f].root=Sgt::P_merge(p[f].root,p[x].root,1,maxin);
156 }
157 int Lca(int x,int y) 
158 {
159     if(p[x].dp<p[y].dp)
160         std::swap(x,y);
161     for(int i=19;i>=0;i--)
162         if(p[p[x].fa[i]].dp>=p[y].dp)
163             x=p[x].fa[i];
164     if(x==y)
165         return x;
166     for(int i=19;i>=0;i--)
167         if(p[x].fa[i]!=p[y].fa[i])
168             x=p[x].fa[i],y=p[y].fa[i];
169     return p[x].fa[0];
170 }
171 int main()
172 {
173     scanf("%d%d",&n,&m);
174     for(int i=1;i<n;i++)
175     {
176         int a,b;
177         scanf("%d%d",&a,&b);
178         adde(a,b);
179         adde(b,a);
180     }
181     Basic_dfs(1,1);
182     for(int i=1;i<=m;i++)
183     {
184         int x,y,z;
185         scanf("%d%d%d",&x,&y,&z);
186         addc(x,z,1);
187         addc(y,z,1);
188         int lca=Lca(x,y);
189         addc(lca,z,-1);
190         if(lca!=1)
191             addc(p[lca].fa[0],z,-1);
192     }
193     Ans_dfs(1,1);
194     for(int i=1;i<=n;i++)
195         printf("%d\n",p[i].ans);
196     return 0;
197 }