BNUOJ 13047 Greatest Parent(要说LCA有点勉强(。・∀・)ノ゙)

/*************************************
**题意是给你一棵树,树的每个结点都有一个 val 值
**然后给你询问,每个询问两个数:u,v
**让你找出所有结点 u 的祖先中 val 值大于或等于 v 的最大的祖先
**
**思路就是 dfs 这棵树
**维护一个栈,栈中保存着当前遍历结点的所有祖宗结点
**具体维护手段就是每次遍历一个结点,把这个结点加入栈中
**然后二分关于这个点的所有查询
**遍历这个结点的所有儿子结点
**最后出栈。
**************************************/

#pragma warning(disable:4996)
#include <cstdio>
#include <vector>
#include <stack>
#define N 100010
using namespace std;

vector<int>to[N], query[N], id[N];
//栈中元素是一个pair,,first表示结点的 val ,second表示结点的编号
pair<int, int> q[4 * N];
int v[N], ans[50005];
int head = 0, tail = 0;

void init(int n, int m){
	for (int i = 0; i <= n; i++){
		to[i].clear();
		query[i].clear();
		id[i].clear();
	}
	head = tail = 0;
}

int binary_search(int l, int r, int target){
	//从左数第一个大于等于target的数
	int low = l, high = r, mid;
	while (low + 1 < high){
		mid = (low + high) / 2;
		if (q[mid].first < target)low = mid + 1;
		else high = mid;
	}

	if (q[low].first >= target)return low;
	return high;
}

void dfs(int u){
	//该节点入栈
	q[tail++] = pair<int, int>(v[u], u);

	//计算所有关于这个结点的询问
	for (int i = 0; i < query[u].size(); i++){
		int val = query[u][i];
		ans[id[u][i]] = q[binary_search(head, tail - 1, val)].second;
	}

	//遍历儿子结点
	for (int i = 0; i < to[u].size(); i++){
		int v = to[u][i];
		dfs(v);
	}
	//该节点出栈
	tail--;
}

int main(){
	int t;
	scanf("%d", &t);
	for (int kase = 1; kase <= t; kase++){
		int n, m;
		scanf("%d %d", &n, &m);
		//cin >> n >> m;
		init(n, m);
		int x, y;
		v[0] = 1;
		for (int i = 1; i < n; i++){
			scanf("%d %d", &x, &y);
			//cin >> x >> y;
			to[x].push_back(i);
			v[i] = y;
		}

		for (int i = 1; i <= m; i++){
			scanf("%d %d", &x, &y);
			query[x].push_back(y);
			id[x].push_back(i);
		}
		dfs(0);
		printf("Case %d:\n", kase);
		for (int i = 1; i <= m; i++)printf("%d\n", ans[i]);// cout << ans[i] << endl;
	}
	return 0;
}

你可能感兴趣的:(BNUOJ 13047 Greatest Parent(要说LCA有点勉强(。・∀・)ノ゙))