PAT A 1053 Path of Equal Weight

题目链接 https://pintia.cn/problem-sets/994805342720868352/problems/994805424153280512

思路

这一题因为节点数较小，所以使用静态法构造树的结构。
对每一个结点有w权值属性以及孩子结点的编号。
所以构造一个结点结构Node

static class  Node {
		int w;
		List<Integer>children ;
		Node(){
			children=new ArrayList<Integer>();
		}
	}

根据输入数据构造树结构
使用数组path存储每一条路上经历结点的编号。
当访问新的结点时，当结点的编号加入数组或者可以覆盖原有的值，可以保证每一条路都是正确的结果。
通过dfs遍历树的结构，如果满足条件输出path中的值。
dfs有三个参数
第一个表示所有遍历过的结点的权值之和，
第二个表示当前结点所在的层次数，
第三个表示当前结点的编号。
在dfs中，需要进行一些剪枝
当当前sum小于S时可以直接return
当sum等于S但当前节点并不是叶子节点时，直接return，否则输出权值序列。
需要注意的是，因为输出结果是需要降序的输出权值，所以需要每一个结点的孩子编号数组根据权值进行排序。

代码

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Main {

	static int n;//节点数
	static int m;//非叶子结点数
	static int S;
	static class  Node {
		int w;
		List<Integer>children ;
		Node(){
			children=new ArrayList<Integer>();
		}
		
	}
	static Node[] nodes=new Node[105];
	static int [] path=new int[105];
	static void dfs(int sum,int numi,int k ) {
		path[numi]=k;
		if(sum>S) 
			return;
		if(sum==S) {
			if(nodes[k].children.size()!=0)
				return;
			for(int i=0;i<=numi;i++) {
				System.out.print(nodes[path[i]].w);
				if(i!=numi) {
					System.out.print(" ");
				}else {
					System.out.println();
				}
			}
		}
		int len=nodes[k].children.size();
		for(int i=0;i<len;i++) {
			int child=nodes[k].children.get(i);
			dfs(sum+nodes[child].w,numi+1,child);
		}
		
	}
	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
		String[] token1=br.readLine().split(" ");	
		n=Integer.parseInt(token1[0]);
		m=Integer.parseInt(token1[1]);
		S=Integer.parseInt(token1[2]);
		String[] token2=br.readLine().split(" ");
		for(int i=0;i<n;i++) {
			nodes[i]=new Node();
			nodes[i].w=Integer.parseInt(token2[i]);
		}
		for(int i=0;i<m;i++) {
			String [] token=br.readLine().split(" ");
			int id=Integer.parseInt(token[0]);
			int len=Integer.parseInt(token[1]);
			for(int j=2;j<len+2;j++) {
				nodes[id].children.add(Integer.parseInt(token[j]));
			}
			Collections.sort(nodes[id].children,(Object a,Object  b)->{
				return nodes[(int)b].w-nodes[(int)a].w;
			});
			
		}
		dfs(nodes[0].w,0,0);
	}
}