PAT甲级1143 Lowest Common Ancestor BST+LCA

PAT甲级1143 Lowest Common Ancestor BST+LCA_第1张图片

本想建树然后通过递归左右子树的形式去做,但是发现题中说明最多有一万个结点,写完以后果然超时....

判别LCA的方法

判断当前的两个点的位置情况,如果一个在左子树,一个在右子树,那么根便是最小公共祖先;

如果都在左子树,递归左子树判断;如果都在右子树,递归右子树判断;

如果一个就是根,那么它便是最小公共祖先

解决超时方法:

由于题目告知是bst树,其实只需要比较结点值便可以确定左右子树的关系。但是递归时根结点的判断比较麻烦。主要有如下两种情况:

1.都在左子树,那么左子树的根下标就是当前的根下标+1(先序遍历特点)

2.如果在右子树,这时就需要循环遍历所有结点,找到第一个比根大的,它就是右子树的根

几大坑点:

1.题目背景描述中说右子树是大于等于根,所以我判断的时候就加了等号,下面的输入输出中又说了结点值都不相同,按理应该没毛病,结果两测试点不过,改回来就好。。。

2.找根的时候我一开始是从右往左找,最后一个测试点超时,改成从左往右就好了,应该是测试数据的问题,很蛇皮

#include
#include
#include
using namespace std;
vector pre;
int m,n;
map h;//结点是否出现
int findright(int value,int index)
{
    int i;
    for(i=index+1;ivalue)break;
    }
    return i;
}
void lca(int root,int a,int b)//root为当前根结点的在先序的下标位置
{
    //if(root<0||root>n)return;
    if((apre[root])||(a>pre[root]&&bpre[root]&&b>pre[root])lca(findright(pre[root],root),a,b);
    else if(pre[root]==a)printf("%d is an ancestor of %d.\n",a,b);
    else printf("%d is an ancestor of %d.\n",b,a);
}
int main()
{
    //int m,n;
    cin>>m>>n;
    pre.resize(n);
    for(int i=0;i>pre[i];h[pre[i]]=true;}
    int a,b;
    for(int i=0;i>a>>b;
        if(!h[a]&&!h[b])printf("ERROR: %d and %d are not found.\n",a,b);
        else if(!h[a])printf("ERROR: %d is not found.\n",a);
        else if(!h[b])printf("ERROR: %d is not found.\n",b);
        else lca(0,a,b);
    }
    return 0;
}

 

 

 

你可能感兴趣的:(PAT甲级)