BZOJ3307: 雨天的尾巴

  1 #include<cstdio>
  2 
  3 #include<cstdlib>
  4 
  5 #include<cmath>
  6 
  7 #include<cstring>
  8 
  9 #include<algorithm>
 10 
 11 #include<iostream>
 12 
 13 #include<vector>
 14 
 15 #include<map>
 16 
 17 #include<set>
 18 
 19 #include<queue>
 20 
 21 #include<string>
 22 
 23 #define inf 1000000000
 24 
 25 #define maxn 250000+5
 26 
 27 #define maxm 2500000+5
 28 
 29 #define eps 1e-10
 30 
 31 #define ll long long
 32 
 33 #define pa pair<int,int>
 34 
 35 #define for0(i,n) for(int i=0;i<=(n);i++)
 36 
 37 #define for1(i,n) for(int i=1;i<=(n);i++)
 38 
 39 #define for2(i,x,y) for(int i=(x);i<=(y);i++)
 40 
 41 #define for3(i,x,y) for(int i=(x);i>=(y);i--)
 42 
 43 #define mod 1000000007
 44 
 45 using namespace std;
 46 
 47 inline int read()
 48 
 49 {
 50 
 51     int x=0,f=1;char ch=getchar();
 52 
 53     while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
 54 
 55     while(ch>=‘0‘&&ch<=‘9‘){x=10*x+ch-‘0‘;ch=getchar();}
 56 
 57     return x*f;
 58 
 59 }
 60 int n,m,q,tot[2],head[maxn][2],top[maxn],id[maxn],son[maxn],dep[maxn],s[maxn],fa[maxn],ans[maxn];
 61 int a[maxn];
 62 struct edge{int go,next,ch;}e[maxm][2];
 63 struct seg{int l,r,mx[2];}t[4*maxn];
 64 map<int,int>mp;
 65 inline void ins(int k,int x,int y,int z)
 66 {
 67     e[++tot[k]][k]=(edge){y,head[x][k],z};head[x][k]=tot[k];
 68 }
 69 inline void insert(int k,int x,int y,int z)
 70 {
 71     ins(k,x,y,z);ins(k,y,x,z);
 72 }
 73 inline void dfs(int x)
 74 {
 75     s[x]=1;
 76     for(int i=head[x][0],y;i;i=e[i][0].next)
 77         if(!dep[y=e[i][0].go])
 78         {
 79             dep[y]=dep[x]+1;fa[y]=x;
 80             dfs(y);
 81             s[x]+=s[y];
 82             if(s[y]>s[son[x]])son[x]=y;
 83         }
 84 }
 85 inline void dfs2(int x,int chain)
 86 {
 87     id[x]=++m;top[x]=chain;
 88     if(son[x])dfs2(son[x],chain);
 89     for(int i=head[x][0];i;i=e[i][0].next)if(e[i][0].go!=son[x]&&e[i][0].go!=fa[x])dfs2(e[i][0].go,e[i][0].go);
 90 }
 91 inline void pushup(int k)
 92 {
 93     int l=k<<1,r=k<<1|1;
 94     t[k].mx[0]=-1;
 95     if(t[l].mx[0]>t[k].mx[0]||(t[l].mx[0]==t[k].mx[0]&&a[t[l].mx[1]]<a[t[k].mx[1]]))t[k].mx[0]=t[l].mx[0],t[k].mx[1]=t[l].mx[1];
 96     if(t[r].mx[0]>t[k].mx[0]||(t[r].mx[0]==t[k].mx[0]&&a[t[r].mx[1]]<a[t[k].mx[1]]))t[k].mx[0]=t[r].mx[0],t[k].mx[1]=t[r].mx[1];
 97 }
 98 inline void build(int k,int l,int r)
 99 {
100     t[k].l=l;t[k].r=r;int mid=(l+r)>>1;
101     if(l==r){t[k].mx[0]=0;t[k].mx[1]=l;return;}
102     build(k<<1,l,mid);build(k<<1|1,mid+1,r);
103 }
104 inline void add(int k,int x,int y)
105 {
106     int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
107     if(l==r){t[k].mx[0]+=y;return;}
108     if(x<=mid)add(k<<1,x,y);else add(k<<1|1,x,y);
109     pushup(k);
110 }
111 inline void dfs3(int x)
112 {
113     for(int i=head[x][1];i;i=e[i][1].next)
114         if(e[i][1].ch==1)add(1,e[i][1].go,1);
115     ans[x]=t[1].mx[0]?t[1].mx[1]:0;
116     for(int i=head[x][1];i;i=e[i][1].next)
117         if(e[i][1].ch==-1)add(1,e[i][1].go,-1);
118     if(son[x])dfs3(son[x]);
119 }
120 
121 int main()
122 
123 {
124 
125     freopen("input.txt","r",stdin);
126 
127     freopen("output.txt","w",stdout);
128 
129     n=read();q=read();
130     for1(i,n-1)insert(0,read(),read(),0);
131     dep[1]=1;dfs(1);dfs2(1,1);
132     int cnt=0;
133     for1(i,q)
134     {
135         int x=read(),y=read(),z=read(),t=mp[z];
136         if(t)z=t;else mp[z]=++cnt,a[cnt]=z,z=cnt;
137         while(top[x]!=top[y])
138         {
139             if(dep[top[x]]<dep[top[y]])swap(x,y);
140             ins(1,top[x],z,1);
141             ins(1,x,z,-1);
142             x=fa[top[x]];
143         }
144         if(dep[x]>dep[y])swap(x,y);
145         ins(1,x,z,1);
146         ins(1,y,z,-1);
147     }
148     build(1,1,cnt);
149     for1(i,n)if(top[i]==i)dfs3(i);
150     for1(i,n)printf("%d\n",a[ans[i]]);
151 
152     return 0;
153 
154 }