[UVA112]树求和（BZOJ2655）

2655: [UVA112]树求和

时间限制: 1 Sec   内存限制: 128 MB
提交: 33   解决: 17
[ 提交][ 状态][ 我的提交]

题目描述

给定一棵二叉树，结点的值均为整数。问树上是否存在一条从根到叶的路径，使得路径上各结点之和为给定的整数。如图所示，二叉树共有4条从根到叶的路径。它们的和分别为27， 22， 26， 18。

在本题中，输入二叉树用括号表示法。上图的二叉树可表示为：(5 (4 (11 (7 () ()) (2 () ()) ) ()) (8 (13 () ()) (4 () (1 () ()) ) ) )，树上的叶结点表示为：(integer () () )

显然，一棵空树是没有路径的，所以对空树的询问，答案一定是"no"。

输入

多组数据，每组数据的格式为：
第1行：先是1个整数，表示指定的某个值，再是1个空格，再是1个字符串，以括号表达式的方式表示的一棵二叉树. 保证输入的二叉树是合法的。在括号表达式描述中，无空格。一棵树最多有256个节点，所有结点的值均不超过1000

输出

对每一行表示的二叉树，若存在从根到叶的路径和等于指定的整数，则输出"yes"，否则输出"no"

样例输入

Copy (如果复制到控制台无换行，可以先粘贴到文本编辑器，再复制)

22 (5(4(11(7()())(2()()))())(8(13()())(4()(1()()))))
20 (5(4(11(7()())(2()()))())(8(13()())(4()(1()()))))
10 (3(2(4()())(8()()))(1(6()())(4()())))
5 ()

样例输出

yes
no
yes
no

分析：其实这道题白痴到极点（23333333然而恶心的括号输入我写了半天qnq），一边建树（其实不用，模拟建树dfs就可以了233333333），到叶子节点时记录可达到的数23333 qwq

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
using namespace std;
int n,len;
char s[100010];
bool vis[2560010];
int find_pair(int x){
	int flag=0;
	for(int i=x;i<=len;i++){
		if(s[i]=='(')	flag++;
		else if(s[i]==')')	flag--;
		if(!flag)	return i;
	}
	return -1;
}
int find_num(int l,int r,int &v){
	sscanf(s+l,"%d",&v);
	for(int i=l;i<=r;i++)
		if(s[i]<'0'||s[i]>'9')
			return i;
	return -1;
}
void build(int l,int r,int now){
	if(l>r)	return ;
	if(l+1==r){
		vis[now]=1;return ;
	}
	if(s[l]=='('&&s[r]==')'){
		build(l+1,r-1,now);return ;
	}
	int v;
	int pos1=find_num(l,r,v);
	int pos2=find_pair(pos1);
	if(s[pos1]=='('&&s[pos1+1]==')'&&s[pos1+2]=='('&&s[pos1+3]==')'){
		vis[now+v]=1;return ;
	}
	build(pos1,pos2,now+v);
	build(pos2+1,r,now+v);
}
int main(){
    while(~scanf("%d%s",&n,s+1)){
		memset(vis,0,sizeof(vis));
		build(1,len=strlen(s+1),0);
		printf("%s\n",vis[n]?"yes":"no");
	}
    return 0;
}