题意:求从根节点出发到叶子节点权值之和等于给定的值的节点的权值顺序,要求非递增
思路:用vector数组保存结构体,数组id为该节点id,一个结构体表示该节点的权值以及孩子节点,深度搜索解决,注意的是根节点要到叶子节点,中途可能有等于权值的时候,但是此情况不符合条件。还有一个是要求非递增,在输入孩子节点的时候就根据孩子节点的权值进行排序,这样一开始得到的顺序就是非递增的了。
代码:
#include<iostream> #include<vector> #include<algorithm> using namespace std; struct node { int w; vector<int> child;//该节点的孩子 }; vector<node> v; vector<int> path; bool com(int a,int b) { return v[a].w > v[b].w; } int n = 0, m = 0, s = 0,root = 0,childNum = 0; void dfs(int index,int nodenum,int sum) { if (sum > s)return; if (sum == s) { if (v[index].child.size()!= 0)return; for (int i = 0; i < nodenum; i++) { printf("%d%c", v[path[i]].w, i == nodenum - 1 ? '\n' : ' '); } return; } for (int i = 0; i < v[index].child.size(); i++) { int node = v[index].child[i]; path[nodenum] = node; dfs(node,nodenum+1,sum+v[node].w); } } int main() { scanf("%d%d%d",&n,&m,&s); v.resize(n); path.resize(n); for (int i = 0; i < n; i++) {//输入权重 scanf("%d",&v[i].w); } for (int i = 0; i < m; i++) { scanf("%d%d",&root,&childNum); for (int j = 0,tem = 0; j < childNum; j++) {//输入该根节点的孩子节点id scanf("%d",&tem); v[root].child.push_back(tem); } sort(v[root].child.begin(), v[root].child.end(),com);//对该节点的孩子节点按权重排序,这样寻找的顺序既是非递增的 } dfs(0,1,v[0].w); system("pause"); return 0; }
