将满二叉树转换为求和树 python

给满出二叉树，编写算法将其转化为求和树

什么是求和树：二叉树的求和树， 是一颗同样结构的二叉树，其树中的每个节点将包含原始树中的左子树和右子树的和。

二叉树：
10
/
-2 6
/ \ / \
8 -4 7 5

求和树：
20(4-2+12+6)
/
4(8-4) 12(7+5)
/ \ / \
0 0 0 0

二叉树给出前序和中序输入，求和树要求中序输出；
所有处理数据不会大于int；

输入描述:
2行整数，第1行表示二叉树的前序遍历，第2行表示二叉树的中序遍历，以空格分割
输出描述:
1行整数，表示求和树的中序遍历，以空格分割
示例1
输入
10 -2 8 -4 6 7 5
8 -2 -4 10 7 6 5
输出
0 4 0 20 0 12 0

每个根节点的值是其左右子树所有值的和，好在树用数组的形式给出，只要求出根节点，该根节点的值为数组所有值减去该值，再递归左右子树即可。

import sys
def func(a,b):
	if len(a)==1:
		return [0]
	elif len(a)==0:
		return []
	mid = b.index(a[0])
	b[mid] = sum(b[:])-b[mid]
	return func(a[1:mid+1],b[:mid]) +\
				[b[mid]] +\
				func(a[mid+1:],b[mid+1:])

if __name__=="__main__":
	a = list(map(int,input().strip().split()))
	b = list(map(int,input().strip().split()))
	ans = func(a,b)
	print(' '.join(map(str,ans)))