POJ 树上求和

题目链接 ： http://dsa.openjudge.cn/2022dsachapter6/2/

题目大致意思：给定一颗树，判断是否存在一条从根节点到叶子节点的路径，使得沿路径节点权值和等于给定的数值

思路：由于必须是根节点到叶子节点的路径，枚举所有情况本质上就是遍历这颗树，所以采用先序遍历的递归形式作为框架。这题的一大难点在于处理这个输入格式，不用getchar几乎必定TLE，用了getchar就要一个一个字符处理，较为恶心。作者的思路就是顺着他输入的格式：读左括号 -> 读数值 -> 递归进入左子树 -> 递归进入右子树 -> 读右括号。中间注意处理负数和空树的情况即可。此外还有一点是如何在递归中判断一个节点是叶子节点。作者的方法是利用递归的返回值，只有空树返回1其余返回0。所以一个节点是叶子节点等价于他进入左右子树的返回值都是1，于是我们就可以对这个节点检查当前的路径和是否等于目标值。

AC代码：

#include<bits/stdc++.h>
using namespace std;

int target, found; 

/* 相当于 cin>>c */ 
char read_char(){   
	char c = getchar();
	while(c == ' ' || c == '\n')
		c = getchar();
	return c;
}

/* 由于是根到叶子的路径， 所以每个节点都必须算上，只用先序遍历一遍树即可 */
int solve(int sum){
	char c = 0;
	c = read_char(); //此时c为左括号
	
	c = read_char(); //此时c为负号，数字 或者 右括号
	
	if(c == ')')
		return 1; // 返回1来标识这是一个空树 
	
	int minus = 1, val = 0;
	if(c == '-'){ 
		minus = -1;  // 负数 
		c = getchar();
	}
	while(c >= '0' && c <= '9'){ 
		val = 10 * val + (c - '0'); //读节点值 
		c = getchar(); 
	} 
	
	ungetc(c, stdin); //此时c为读完数字后的下一个字符，可能多读了一个，送回去
	
	sum += minus * val; // 更新路径和 
	
	int r1 = solve(sum); // 检查左子树 
	int r2 = solve(sum); // 检查右子树 
	if(sum == target && r1 && r2 ) //左右子树都为空树，为叶子节点 
		found = 1; 
	
	c = read_char(); //把自己的右括号读掉 
	
	return 0;
}
int main(){
	while(cin.peek() != EOF){
		found = 0;
		scanf("%d", &target);
		solve(0);
		printf((found ? "yes\n" : "no\n"));
		getchar();  //末尾回车 
	}
	return 0;
}