uva 699 The Falling Leaves

最新推荐文章于 2024-04-11 16:23:11 发布

原创最新推荐文章于 2024-04-11 16:23:11 发布 · 308 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#uva #树 #dfs

本文介绍了一种通过深度优先搜索（DFS）遍历树结构的方法来计算具有相同横坐标节点的权值之和。利用C++实现了一个具体的示例程序。

题目：The Falling Leaves

题意：树中根节点和左右子节点的距离定为1，输出有同一横坐标的节点的权值和。

思路：dfs遍历。

代码：

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<deque>
using namespace std;

map<int,int> sum;
void dfs(int pos,int value) {
	if(!sum.count(pos)) sum[pos]=0;
	sum[pos]+=value;
	int x,y;
	scanf("%d",&x);
	if(x!=-1) dfs(pos-1,x);
	scanf("%d",&y);
	if(y!=-1) dfs(pos+1,y);
	return ;
}

int main() {
	int T=0;
	while(true) {
		int x;
		scanf("%d",&x);
		if(x==-1) break;
		sum.clear();
		dfs(0,x);
		printf("Case %d:\n",++T);
		for(map<int,int>::iterator it=sum.begin(); it!=sum.end(); it++) {
			if(it!=sum.begin()) printf(" ");
			printf("%d",(it->second));
		}
		printf("\n\n");
	}
	return 0;
}