7-11 树最近公共祖先

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 
#include 
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 
#include 
#include 
#include 
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]);
}