uva 548 - Tree

这题我是仿照LRJ书上做的,,书上是给先序和中序,求后序, 那么本题是给后序和中序,然后先序遍历求最小路径,可以同时操作。

由于不甚熟练,自己码了接近2小时,大都是递归的时候细节处理不当。。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <map>
#include <vector>
#include <set>
#include <stack>
#include <queue>
using namespace std;
#define N 100000
#define Dbag printf("haha\n");
int minpath, minleaf;
void dfs(int n, int sum, int *a1, int *a2)//a1后序,a2中序,n表示当前的树的节点数
{
    sum+=a1[0];
    if(n==1)                           //节点数为1的时候,自然是根节点
    {
        if(minpath > sum)
            {minpath = sum; minleaf = a1[0];}
        else if(minpath == sum)
            minleaf = min(minleaf, a1[0]);
        return ;
    }
    if(n<=0)
        return ;
    int i;
    for(i = 0; i < n; i++)          //在中序a2中找出根的位置
        if(a1[0]==a2[i])
            break;
    int len = i;                  //下面递归一定要注意传的参数
    dfs(len, sum, a1-(n-i), a2);  // 递归左子树,后面两个参数依次是 左子树跟的位置  左子树在a2中的起点
    dfs(n-len-1, sum, a1-1, a2+i+1);//递归右子树
}
int main()
{
    char ch;
    int s1[N], s2[N];         //1中序,2后序
    while(~scanf("%d%c",s1,&ch))
    {
        minpath = 0x7fffffff;minleaf = 0x7fffffff;
        int n;
        for(n = 1; ch != '\n'; n++)
            scanf("%d%c", s1+n, &ch);
        for(int i = 0; i < n; i++)
            scanf("%d", s2+i);
        dfs(n, 0, &s2[n-1], s1);
        printf("%d\n", minleaf);
    }
    return 0;
}




 

 

你可能感兴趣的:(uva 548 - Tree)