7-11 树最近公共祖先
已知树结点为互不相等且不等于0的整数。请编写程序找出非空树中两个结点的最近公共祖先。例如对于图1(a)所示的树t,结点1和2的最近公共祖先是5;结点2和4的最近公共祖先是8。
输入格式:
输入为2行,第1行为一组用空格间隔的整数,个数不超过100个,表示带空指针信息的二叉树先根序列。其中空指针信息用0表示。第2行为空格间隔的两个互不相等的整数A和B,表示给定的两个结点值,保证A和B肯定在输入的树中。
注:我们已知二叉树与其自然对应的树相比,二叉树中结点的左孩子对应树中结点的左孩子,二叉树中结点的右孩子对应树中结点的右兄弟。进而我们可以利用“带空指针信息的先根序列构建二叉树”的方法来构建其对应的树的左孩子-右兄弟存储结构。如8 5 1 0 6 0 2 0 0 3 4 0 0 7 0 0 0对应图1(a)所示的树,1 2 0 3 0 4 0 0 0对应如图1(b)所示的树。
输出格式:
输出为一个整数,表示A和B的最近公共祖先结点的值。
输入样例1:
8 5 1 0 6 0 2 0 0 3 4 0 0 7 0 0 0
1 2
输出样例1:
5
输入样例2:
8 5 1 0 6 0 2 0 0 3 4 0 0 7 0 0 0
2 4
输出样例2:
8
不知道是不是c语言的速度还是比较慢,可能自己造的堆栈还是没有系统的快吧,我寻思这是最优解却超时了,强行加入系统的堆栈反而过了?迷惑。
超时代码,但是这个思路清晰
#include <stdio.h>
#include <stdlib.h>
int main(){
int a,b,n=1,pre[1001],stock[1001],stk[1001],t;
//scanf("%d",&stock[1]);
while(n){
scanf("%d",&a);
if(a){
pre[a]=stock[n];
stk[a]=n;
stock[++n]=a;
}
else{
n--;
}
}
scanf("%d%d",&a,&b);
if(stk[a]>stk[b]){
t=stk[a]-stk[b];
while(t--){
a=pre[a];
}
}
else{
t=stk[b]-stk[a];
while(t--){
b=pre[b];
}
}
//printf("%d %d",stk[a],stk[b]);
while(pre[a]!=pre[b]){
a=pre[a];
b=pre[b];
}
printf("%d",pre[a]);
}
完美代码
被迫c++,妥协,明明画蛇添足。
#include <stdio.h>
#include <stdlib.h>
#include <queue>
#include <stack>
using namespace std;
int main(){
int a,b,n=1,pre[1001],stock[1001],stk[1001],t;
stack<int> temp;
scanf("%d",&a);
stk[a]=n++;
temp.push(a);
while(!temp.empty()){
scanf("%d",&a);
if(a){
pre[a]=temp.top();
//printf("a:%d ceng:%d\n",a,n);
stk[a]=n++;
temp.push(a);
}
else{
temp.pop();
n--;
}
}
scanf("%d",&a);
scanf("%d%d",&a,&b);
//printf("a:%d b:%d\n",a,b);
//printf("%d %d\n",stk[a],stk[b]);
if(stk[a]>stk[b]){
t=stk[a]-stk[b];
while(t--){
a=pre[a];
}
}
else{
t=stk[b]-stk[a];
while(t--){
b=pre[b];
}
}
//printf("%d %d\n",stk[a],stk[b]);
while(pre[a]!=pre[b]){
a=pre[a];
b=pre[b];
}
printf("%d",pre[a]);
}