第十六题 UVA548 树 Tree

题意翻译
输入一个二叉树的中序和后序遍历，请你输出一个叶子节点，该叶子节点到根的数值总和最小，且这个叶子是编号最小的那个。 输入： 您的程序将从输入文件中读取两行(直到文件结尾)。第一行是树的中序遍历值序列,第二行是树的后序遍历值序列。所有值将不同，大于零且小于或等于10000.二叉树的节1<=N<=10000。 输出： 对于每个树描述，您应该输出最小值路径的叶节点的值。存在多路径最小的情况下，您应该选择终端叶子节点上具有最小值的那条路径，且输出那个最小值的终端叶子。

Translated by @凉凉

输入输出样例
输入 #1复制
3 2 1 4 5 7 6
3 1 2 5 6 7 4
7 8 11 3 5 16 12 18
8 3 11 7 16 18 12 5
255
255
输出 #1复制
1
3
255

根据中序遍历和后序遍历建树，然后按照题意求解
搞了半天搞出来了，然后我发现树节点的编号并不一定是1~n连续的，我心态崩了呀

//节点 1~n连续的代码   显然不能AC
#include<iostream>
#include<cstring>
#include<cstdio>
#define Maxn 100010
using namespace std;

struct  Node{
	int l,r,node;
}tre[Maxn];
int las[Maxn],mid[Maxn];
int pos = 0,Sum = 99999999;
int Creat(int l1,int r1,int l2,int r2) {
	if(l1 == r1) {
		int kase = mid[l1];
		tre[kase].l = tre[kase].r = -1;
		tre[kase].node = kase;
		return kase;
	}
	if(l1 > r1) return -1;
	int Root;
	for(Root=l1; Root<=r1; Root++)
		if(mid[Root] == las[r2]) break;
	tre[mid[Root]].l = Creat(l1, Root-1, l2, l2 + (Root - l1 - 1));
	tre[mid[Root]].r = Creat(Root + 1, r1,l2 + (Root - l1 - 1) + 1, r2 - 1);
	tre[mid[Root]].node = mid[Root];
	return mid[Root];
}

void DFS(int p,int sum) {
	if(tre[p].l == -1 && tre[p].r == -1) {
		if(sum <= Sum) {
			pos = p; Sum = sum;
		}
		return ;
	}
	if(tre[p].l != -1) DFS(tre[p].l,sum + tre[tre[p].l].node);
	if(tre[p].r != -1) DFS(tre[p].r,sum + tre[tre[p].r].node);
}

void dfs(int Node) {
	printf("%d ",Node);
	if(tre[Node].l != -1) dfs(tre[Node].l);
	if(tre[Node].r != -1) dfs(tre[Node].r);
}


int main() {
	int k1=0,k2=0;
	while(scanf("%d",&mid[1]) != EOF) {
		k1 = 1; k2 = 0;
		char ch;
		for(int i=1; i<=Maxn; i++) tre[i].l = tre[i].r = -1;
		while(1) {
			ch = getchar();
			if(ch == '\n') break;
			scanf("%d",&mid[++k1]);
		}
		scanf("%d",&las[++k2]);
		while(1) {
			ch = getchar();
			if(ch == '\n') break;
			scanf("%d",&las[++k2]);
		}
		int Head = Creat(1,k1,1,k2);
		DFS(Head,tre[Head].node);
		printf("%d\n",pos);
		
	}
	
	return 0;
}

// AC代码
#include<malloc.h>
#include<iostream>
#include<stdio.h>
#include<string>
#include<cstring>
using namespace std;
struct tree{
    int data;
    int left;
    int right;
} T[100100]; //数组模拟的二叉树
int m_sum=100000000,pos; //用于DFS时保存结果
int mid[100100],last[100100];
int create_tree(int m1,int m2,int l1,int l2){
    if (m1 == m2) {
        T[m1].left=T[m1].right=-1;
        T[m1].data=mid[m1];
        return m1;
    }
    if (m1>m2) return -1;
    int i;
    for (i=0; i<=m2; i++)
        if (mid[i]==last[l2]) break;
    
    T[i].left=create_tree(m1,i-1,l1,l1+i-m1-1);//递归建树！！注意四个参数，runtime好多遍
    T[i].right=create_tree(i+1,m2,l1+i-m1,l2-1);
    T[i].data=mid[i];
    return i;
}
int dfs(int head,int m) {
    m += T[head].data;
    if (T[head].left==-1&&T[head].right==-1) {
        if (m<m_sum) {
            m_sum=m;
            pos=T[head].data;
        }
        return T[head].data;
    }
    int sum=T[head].data;
    if (T[head].left!=-1&&T[head].right==-1) sum += dfs(T[head].left,m);
    else if (T[head].right!=-1&&T[head].left==-1) sum += dfs(T[head].right,m);
    else sum += min(dfs(T[head].left,m),dfs(T[head].right,m));
    return sum;
}
int main() {
    int k1=0,k2=0;
    while (scanf("%d",&mid[0])!=EOF) {
        pos=0,m_sum=100000000;
        k1=1;
        for (int i=0; i<=10000; i++)
            T[i].left=T[i].right=-1,T[i].data=0;
        while (1)
        {
            char ch=getchar();
            if (ch=='\n') break;
            scanf("%d",&mid[k1++]);
        }
        k1--,k2=0;
        while (1)
        {
            scanf("%d",&last[k2++]);
            char ch=getchar();
            if (ch=='\n') break;
        }
        k2--;
        int T_head=create_tree(0,k1,0,k2);
        dfs(T_head,0);
        printf("%d\n",pos);
    }
    return 0;
}