ural 1136. Parliament (通過後序遍歷構建二叉搜索樹)

對於一個後序遍歷序列,它可以分解爲   (左子樹部分)(右子樹部分)(根),由於是查找樹,可知(左子樹)<(根)  (右子樹)>(根)

所以對於當前一個區間內的序列[l,r],可以知道r下標對應的數值一定是當前子樹的根,然後找到一個下標k,  [l,k-1] 裏面的數值都小於根,他們是左子樹 , [k,r-1]裏面的數值都大於根,他們是右子樹,這樣就可以遞歸建樹下去了。建樹之後直接用新的規則後序遍歷二叉樹即可


#include <cstdio>
#include <cstring>
#define N 3010

int a[N],tot;
struct node
{
   int n;
   int lch,rch;
}t[N];
int n;

int build(int l ,int r)
{
   if(l > r)  return -1;

   int i,pos;

   t[tot].n = a[r]; //保存樹根
   pos = tot++;
   if(l == r) return pos;

   for(i=l; i<r; i++)
      if(a[i] > a[r])
         break;

      //左子樹區間[l,i-1]  ,  右子樹區間[i,r-1]
      int lch = build(l,i-1);
      int rch = build(i,r-1);
      t[pos].lch = lch;
      t[pos].rch = rch;
      return pos;
}

void Travel(int rt)
{
   if(rt == -1) return ;
   Travel(t[rt].rch);
   Travel(t[rt].lch);
   if(rt == 0)
      printf("%d\n",t[rt].n);
   else
      printf("%d ",t[rt].n);
}

int main()
{
   while(scanf("%d",&n)!=EOF)
   {
      memset(t,-1,sizeof(t));
      for(int i=0; i<n; i++)
         scanf("%d",&a[i]);
      tot=0;
      build(0,n-1); //遞歸建樹
      Travel(0);
   }
   return 0;
}


發佈了202 篇原創文章 · 獲贊 6 · 訪問量 23萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章