树的DFS——Path of Equal Weight

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
const int MAXN = 110;

struct node{
    
    
	int weight;
	vector<int> child;
}Node[MAXN];  // 静态树

bool cmp(int a, int b)
{
    
    
   return Node[a].weight > Node[b].weight;	
} 

int n, m, S;
int path[MAXN];

void DFS(int index, int numNode, int sum)
{
    
    
	if(sum > S) return;
	if(sum == S)
	{
    
    
		if(Node[index].child.size() != 0) return;
		else{
    
    
			for(int i = 0; i < numNode; i++)
			{
    
    
				printf("%d", Node[path[i]].weight);
				if(i < numNode - 1) printf(" ");
				else printf("\n");
			}
			return;
		}
	}
	for(int i = 0; i < Node[index].child.size(); i++)
	{
    
    
		int child = Node[index].child[i];
		path[numNode] = child;
		DFS(child, numNode + 1, sum + Node[child].weight);
	}
}

int main()
{
    
    
	// 建树 
	scanf("%d%d%d", &n, &m, &S);
	for(int i = 0; i < n; i++)
	{
    
    
		scanf("%d", &Node[i].weight);
	}
	int id, k, child;
	for(int i = 0; i < m; i++)
	{
    
    
		scanf("%d%d", &id, &k);
		for(int j = 0; j < k; j++)
		{
    
    
			scanf("%d", &child);
			Node[id].child.push_back(child); 
		} // 存放子节点地址 
		sort(Node[id].child.begin(), Node[id].child.end(), cmp);
	}
	
	// DFS
	path[0] = 0;
	DFS(0, 1, Node[0].weight);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/tian__si/article/details/114276636