7-3 二叉搜索树的最近公共祖先

7-3 二叉搜索树的最近公共祖先

分数 30

 作者 陈越

单位 浙江大学

给定一棵二叉搜索树的先序遍历序列，要求你找出任意两结点的最近公共祖先结点（简称 LCA）。

输入格式：

输入的第一行给出两个正整数：待查询的结点对数 M（≤ 1 000）和二叉搜索树中结点个数 N（≤ 10 000）。随后一行给出 N 个不同的整数，为二叉搜索树的先序遍历序列。最后 M 行，每行给出一对整数键值 U 和 V。所有键值都在整型int范围内。

输出格式：

对每一对给定的 U 和 V，如果找到 A 是它们的最近公共祖先结点的键值，则在一行中输出 LCA of U and V is A.。但如果 U 和 V 中的一个结点是另一个结点的祖先，则在一行中输出 X is an ancestor of Y.，其中 X 是那个祖先结点的键值，Y 是另一个键值。如果 二叉搜索树中找不到以 U 或 V 为键值的结点，则输出 ERROR: U is not found. 或者 ERROR: V is not found.，或者 ERROR: U and V are not found.。

输入样例：

6 8
6 3 1 2 5 4 8 7
2 5
8 7
1 9
12 -3
0 8
99 99

输出样例：

LCA of 2 and 5 is 3.
8 is an ancestor of 7.
ERROR: 9 is not found.
ERROR: 12 and -3 are not found.
ERROR: 0 is not found.
ERROR: 99 and 99 are not found.

#include <bits/stdc++.h>
using namespace std;

#define NMAX 10005
struct PTnode
{
    int data;
    int parent;
    int depth;
};

struct PTree
{
    struct PTnode node[NMAX];
    int n;
};

struct PTree t;

void shuru(int l,int r,int dep)
{
	if(l >= r) return ;
	if(t.node[r].data < t.node[l].data || t.node[l+1].data > t.node[l].data)
	{
		t.node[l+1].parent = l;
		t.node[l+1].depth = dep;
		shuru(l+1,r,dep+1);
	}
	else
	{
		int mid = l+1;
		t.node[l].depth = dep-1;
		while(t.node[mid].data < t.node[l].data && mid < r)
		{
			mid ++;
		}
		t.node[l+1].parent = l;
		t.node[l+1].depth = dep;
		t.node[mid].parent = l;
		t.node[mid].depth = dep;
		shuru(l+1,mid-1,dep+1);//左子树 
		shuru(mid,r,dep+1);//右子树 
	}
	return ;
}

int Ffind(int a)
{
    for(int i = 0;i < t.n;i ++)
    {
        if(t.node[i].data == a)
        {
            return i+1;
        }
    }
    return 0;
}

void panduan(int a,int b)
{
    int n = Ffind(a)-1,m = Ffind(b)-1;
    if(n == -1 && m == -1)
    {
        cout << "ERROR: " << a  << " and " << b << " are not found." << endl;
        return ;
    }
    else if(n == -1)
    {
        cout << "ERROR: " << a << " is not found." << endl;
        return ;
    }
    else if(m == -1)
    {
        cout << "ERROR: " << b << " is not found." << endl;
        return ;
    }
//    else if(a == b)
//    {
//        cout << "LCA of " << a << " and " << b << " is " << t.node[t.node[n].parent].data << "."<<endl;
//        return ;
//    }
    else
    {
        int i = n,j = m;
        int k;
         while(t.node[i].depth != t.node[j].depth)
         {
             if(t.node[i].depth > t.node[j].depth)
             {
                i = t.node[i].parent;
                k = 1;
             }
             else
             {
                j = t.node[j].parent;
                k = 0;
             }
         }
        if(t.node[i].data == t.node[j].data && k == 1)
        {
            cout << t.node[m].data << " is an ancestor of " << t.node[n].data << "." << endl;
            return ;
        }
        else if(t.node[i].data == t.node[j].data && k == 0)
        {
            cout << t.node[n].data << " is an ancestor of " << t.node[m].data << "." << endl;
            return ;
        }
        else
        {
            while(t.node[i].data != t.node[j].data)
            {
                i = t.node[i].parent;
                j = t.node[j].parent;
            }
            cout << "LCA of " << t.node[n].data << " and " << t.node[m].data << " is " << t.node[i].data << "." << endl;
            return ;
        }
    }
}

int main()
{
    int m,n;
    cin >> m >> n;
    t.n = n;
    for(int i = 0;i < n;i ++)cin >> t.node[i].data;
    shuru(0,n-1,1);
//    for(int i = 0;i < n;i++)
//        cout << t.node[i].data << " " << t.node[t.node[i].parent].data << endl;
    for(int i = 0;i < m;i ++)
    {
        int a,b;
        cin >> a >> b;
        panduan(a,b);
    }
    return 0;
}

全屏浏览题目

切换布局