Codeforces 588E 树上主席树+Lca

Codeforces 588E 树上主席树+Lca

Codeforces 588E Duff in the Army

西安邀请赛网络赛J的升级版本，要求输出具体的方案，这题就没办法离线水过去了。对dfs序建一个主席树，那么对于每个询问答案就是
\(sum[u]+sum[v]-sum[lca(u,v)]-sum[fa[lca(u,v)]]\)
之后就不难了，实现起来比较复杂，不过写的蛮快的，wa了之后扫了一下主席树感觉没什么问题，就扫了一眼lca 发现最后求lca的部分写错了。

#include <bits/stdc++.h>
using namespace std;
#define dd(x) cout<<#x<<"="<<x<<" "
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define de(x) cout<<#x<<"="<<x<<"\n"
#define mes(p) memset(p,0,sizeof(p))
#define fi first
#define se second
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define sz(x) (int)x.size()
#define pb push_back
#define ls (rt<<1)
#define rs (ls|1)
#define all(x) x.begin(),x.end()
const int maxn=100005;
typedef long long ll;
typedef vector <int > vi;
typedef pair <int, int> pi;
int dep[maxn], f[maxn][32],cnt, T[maxn];
int n, m, q;
vector <int > num[maxn], v[maxn];
vector <int > ans;
struct  node{
	int sum,l,r;
	node() {sum=l=r=0;}
}t[maxn<<6];
void update(int pre,int &now,int l,int r,int pos){
	now = ++cnt;
	t[now].sum = t[pre].sum+1;
	if(l==r) return ;
	t[now]. l = t[pre].l;
	t[now].r = t[pre].r;
	int mid = l+r >> 1;
	if(mid>=pos) update(t[pre].l,t[now].l,l,mid,pos);
	else update(t[pre].r,t[now].r,mid+1,r,pos);
}
void dfs(int x,int p,int d){
	dep[x] = d;
	f[x][0] = p;
	rep(i,1,30)
		f[x][i] = f[f[x][i-1]][i-1];
	T[x] = T[p];
	for(auto u:num[x])
	 	update(T[x],T[x],1,m,u);
	for(auto u:v[x])
		if(u!=p)
			dfs(u,x,d+1);
}
int lca(int u,int v){
	if(dep[u] < dep[v]) swap(u,v);
	for(int i=29,d=dep[u]-dep[v];i>=0;i--)
		if(d>>i&1) u=f[u][i];
	if(u==v) return v;
	for(int i=29;i>=0;i--)
	if(f[u][i]!=f[v][i])
		u=f[u][i], v=f[v][i];
	return f[u][0];
}
void qr(int x,int y,int z,int d,int l,int r,int k){
	int sl = t[t[x].l].sum + t[t[y].l].sum -t[t[z].l].sum - t[t[d].l].sum;
	if(l==r){
		if(k)ans.pb(l);
		return ;
	}
	int mid = l+r >> 1;
	if(sl>=k) qr(t[x].l,t[y].l,t[z].l,t[d].l,l,mid,k);
	else {
		if(sl)  qr(t[x].l,t[y].l,t[z].l,t[d].l,l,mid,sl);
		qr(t[x].r,t[y].r,t[z].r,t[d].r,mid+1,r,k-sl);
	}
}
int main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
 	cin >> n >> m >> q;
	rep(i,0,n-1){
		int x,y;
		cin >> x >> y;
		v[x].pb(y);
		v[y].pb(x);
	}
	rep(i,0,m){
		int x;
		cin >> x;
		num[x].pb(i+1); 
	} 
	dfs(1,1,0);	
	rep(i,0,q){
		int x,y,a;
		ans.clear();
		cin >> x>>y>> a;
		int l=lca(x,y), d=f[l][0];
		if(l==1) d=0;
		int k = min(a,t[T[x]].sum+t[T[y]].sum-t[T[l]].sum-t[T[d]].sum);
		qr(T[x],T[y],T[l],T[d],1,m,k);
		cout << sz(ans);
		for(auto i:ans )cout << " " <<i;
		cout << endl;
	}
	return 0;
}