题目: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;
}