BZOJ4009 [HNOI2015]接水果

//It is made by ljh2000
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <string>
#include <complex>
using namespace std;
typedef long long LL;
const int MAXN = 40011;
const int MAXM = 80011;
int n,m,Q,ecnt,first[MAXN],to[MAXM],next[MAXM],f[MAXN][17],dfn[MAXN],last[MAXN],deep[MAXN];
int Mcnt,ans[MAXN],c[MAXN<<1];
struct Matrix{ int lx,rx,ly,ry,val;}b[MAXN<<2];//矩形的左上角和右下角
struct ask{ int x,y,k,id; }q[MAXN],tem[MAXN<<1],tem2[MAXN<<1];
struct Thing{ int x,y,val; }wyh[MAXN<<3];
inline bool cmp(Matrix pp,Matrix pq){ return pp.val<pq.val; }
inline bool cmp2(ask pp,ask pq){ return pp.x<pq.x; }
inline bool cmp3(Thing pp,Thing pq){ return pp.x<pq.x; }

inline void link(int x,int y){ next[++ecnt]=first[x]; first[x]=ecnt; to[ecnt]=y; }
inline void add(int x,int val){ while(x<=n) { c[x]+=val; x+=x&(-x); } }
inline int query(int x){ int tot=0; while(x>0) { tot+=c[x]; x-=x&(-x); } return tot; }

inline int getint(){
    int w=0,q=0; char c=getchar(); while((c<‘0‘||c>‘9‘) && c!=‘-‘) c=getchar();
    if(c==‘-‘) q=1,c=getchar(); while (c>=‘0‘&&c<=‘9‘) w=w*10+c-‘0‘,c=getchar(); return q?-w:w;
}

inline void dfs(int x,int fa){//预处理dfs序和倍增
	dfn[x]=++ecnt;
	for(int i=first[x];i;i=next[i]) {
		int v=to[i]; if(v==fa) continue;
		deep[v]=deep[x]+1; f[v][0]=x; dfs(v,x);
	}
	last[x]=ecnt;
}

inline int lca(int x,int y){
	if(deep[x]<deep[y]) swap(x,y);/*!!!*/ int t=0; while((1<<t)<=deep[x]) t++; t--;
	for(int i=t;i>=0;i--) if(deep[x]-(1<<i)>=deep[y]) x=f[x][i]; if(x==y) return x;
	for(int i=t;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; return f[x][0];
}

inline void solve(int l,int r,int L,int R){//询问处理区间为[l,r]，矩阵区间为[L,R]
	if(l>r) return ;
	if(L==R) {
		for(int i=l;i<=r;i++) ans[q[i].id]=b[L].val;
		return ;
	}
	int cnt=0,now=0,nowk,nowl=0,nowr=0,mid=(L+R)>>1;
	//把每个矩形拆分成四个点事件，从上往下执行扫描线，树状数组维护区间sum
	//在图上画一画能够发现，这样从上往下做可以统计经过这个点的矩阵个数
	//并且只要从头做到尾就相当于是执行了撤销操作，因为-1操作全都被执行了
	for(int i=L;i<=mid;i++) {
		wyh[++cnt]=(Thing){b[i].lx,b[i].ly,1};
		wyh[++cnt]=(Thing){b[i].lx,b[i].ry+1,-1};//不能因为越界就不处理！
		wyh[++cnt]=(Thing){b[i].rx+1,b[i].ly,-1};
		wyh[++cnt]=(Thing){b[i].rx+1,b[i].ry+1,1};
	}
	sort(wyh+1,wyh+cnt+1,cmp3);
	for(int i=l;i<=r;i++) {
		while(now<cnt && wyh[now+1].x<=q[i].x) now++,add(wyh[now].y,wyh[now].val);
		nowk=query(q[i].y);
		if(q[i].k<=nowk) tem[nowl++]=q[i];
		else { q[i].k-=nowk; tem2[nowr++]=q[i]; }//不能从r放起！x顺序会失去！！！
	}
	for(;now<cnt;) now++,add(wyh[now].y,wyh[now].val);
	for(int i=0;i<nowl;i++) q[i+l]=tem[i];
	for(int i=0;i<nowr;i++) q[l+nowl+i]=tem2[i];
	solve(l,l+nowl-1,L,mid);
	solve(l+nowl,r,mid+1,R);
}

inline void work(){
	n=getint(); m=getint(); Q=getint(); int x,y,z,LCA,yfa;
	for(int i=1;i<n;i++) { x=getint(); y=getint(); link(x,y); link(y,x); }
	ecnt=0; dfs(1,0); for(int j=1;j<=16;j++) for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1];
	for(int i=1;i<=m;i++) {//将每个盘子的影响范围考虑成一个或者两个矩形
		x=getint(); y=getint(); z=getint(); LCA=lca(x,y);
		if(dfn[x]>dfn[y]) swap(x,y);
		if(LCA==x) {//考虑拆成两个矩形
			yfa=y; for(int j=16;j>=0;j--) if(deep[yfa]-(1<<j)>deep[x]) yfa=f[yfa][j];
			//b[++Mcnt]=(Matrix){1,dfn[yfa]-1,dfn[y],last[y],z};//直接拆分成两个矩形，显然不会有交集，所以可以分开计算贡献
			//b[++Mcnt]=(Matrix){dfn[y],last[y],last[yfa]+1,n,z};//注意只统计x<y的情况！
			if(dfn[yfa]>1) b[++Mcnt]=(Matrix){1,dfn[yfa]-1,dfn[y],last[y],z};//直接拆分成两个矩形，显然不会有交集，所以可以分开计算贡献
			if(last[yfa]<n) b[++Mcnt]=(Matrix){dfn[y],last[y],last[yfa]+1,n,z};//注意只统计x<y的情况！
		}
		else b[++Mcnt]=(Matrix){dfn[x],last[x],dfn[y],last[y],z};
	}

	for(int i=1;i<=Q;i++) {
		q[i].x=dfn[getint()]; q[i].y=dfn[getint()]; q[i].k=getint(); q[i].id=i;
		if(q[i].x>q[i].y) swap(q[i].x,q[i].y);
	}

	sort(b+1,b+Mcnt+1,cmp);
	sort(q+1,q+Q+1,cmp2);

	solve(1,Q,1,Mcnt);
	for(int i=1;i<=Q;i++) printf("%d\n",ans[i]);
}

int main()
{
    work();
    return 0;
}