题目
就是给你一个所谓的二叉树的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');
}
}