UvaOJ 112 - Tree Summing

题目

就是给你一个所谓的二叉树的S-expression 让你判断是否存在指定的 根到叶子的路径和。

我在这里用到cin.putback() （C里有 int ungetcA(int ch, FILE *stream) 在cstdio中），方便把下一个字符拿出来判断后再塞回缓冲区，最后用格式化读。

特别注意的是，题目所说的叶子，必须具有(integer()()) 这样的形式才是一个叶子，也就是不是算到() 这里的。开始我一直拿()当路径的结尾，结果就wa了。也是，怎么能拿空节点当叶子呢....好像有点sb...

另外，ac后又测试了下，输入数据中 数字间都没有空格，不是在这里wa的。

代码：

//二叉树，递归。		C里有没有类似 cin.putback(char)  ??

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

#define MAXN 1000002	//没说多少？
int n;
char Rchar()
{
	char t;
	while(scanf("%c", &t) && (t==' ' || t=='\n')){}
	return t;
}
int solve(int root, int sum)		//(3(...)(...))
{
	Rchar();	//'('
	char t = Rchar();
	if(t==')')
	{
		if(sum+root == n) 
		{ 
			char buf[3];
			buf[0] = Rchar();
			buf[1] = Rchar();
			cin.putback(buf[1]);
			cin.putback(buf[0]);
			if(buf[0] == '(' && buf[1] == ')') return 1;		//证明是子叶
			else return 0;
		}
		else return 0;
	}
	cin.putback(t);		//!!!!!!		c 里有没有相应函数？
	int r;	scanf("%d", &r);
	int ret1 = solve(r, sum+root);
	int ret2 = solve(r, sum+root);
	Rchar();	//')'
	return ret1 || ret2;
}

int main()
{
	while(scanf("%d", &n)!=EOF)
	{
		printf(solve(0, 0)? "yes": "no");
		putchar('\n');
	}
}