PAT A1151

题目:https://pintia.cn/problem-sets/994805342720868352/problems/1038430130011897856

没想到还是没有得满分。。。明明已经有过A1143的经验。当时是一个BST,满足左子树都小于根结点的值,右子树都大于根结点的值。这个题我又去暴力了,依旧超时。笨啊。

同样不需要建树,中序遍历的时候,给每个节点赋值,表示位置,就又转化成BST了。然后找到需要查找的u,v的位置,所以这里用map处理,每个值对应一个位置,然后从先序遍历中找位置在两个位置中间的即是LCA。

#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;
const int maxn = 10010;
int m, n, pre[maxn], in[maxn];
map<int, int> pos;
int u, v;

int main(){
	scanf("%d%d", &m, &n);
	for(int i = 0; i < n; i++){
		scanf("%d", in+i);
		pos[in[i]] = i;
	}
	for(int i = 0; i < n; i++)
		scanf("%d", pre+i);
	for(int i = 0; i < m; i++){
		scanf("%d%d", &u, &v);
		int flag1 = (pos.find(u) == pos.end()), flag2 = (pos.find(v) == pos.end());
		if(flag1 || flag2){
			if(flag1 && flag2)
				printf("ERROR: %d and %d are not found.\n", u, v);
			else if(flag1)
				printf("ERROR: %d is not found.\n", u);
			else
				printf("ERROR: %d is not found.\n", v);
		}else{
			int posu = pos[u], posv = pos[v];
			if(posu > posv){
				swap(posu, posv);
			}
			int posLCA, LCA;
			for(int i = 0; i < n; i++){
				posLCA = pos[pre[i]];
				LCA = pre[i];
				if(posLCA <= posv && posLCA >= posu){
					break;
				}
			}
			if(posLCA == pos[u])
				printf("%d is an ancestor of %d.\n", u, v);
			else if(posLCA == pos[v])
				printf("%d is an ancestor of %d.\n", v, u);
			else
				printf("LCA of %d and %d is %d.\n", u, v, LCA);
		}
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值