写树写树
这个题其实没啥技巧,就是先建树再dfs就好啦~~~
不过要提几个注意点(这为clion不能正常使用的我雪上加霜)
- dfs一定不要漏边界了!!!(不是说不知道weight为0的时候输出,而是你疯狂忘记weight<0的时候直接返回)
- 以及不要忘了判断是不是叶子节点哈
-它要求按照一定顺序输出的时候,你可以搜索的时候就按照这个顺序搜
-注意啊,这里的child可是int形的,所以你cmp一定要是两个Int为参数啊
否则报错会给你跳到这里:
怎么还是排序的锅
快测测这个测试样例:
input:
4 1 2
1 1 1 1
00 3 01 02 03
output:
1 1
1 1
1 1
具体就是这样啦:
#include<vector>
#include<iostream>
#include <algorithm>
#include <cmath>
#include<map>
#include <queue>
#include <stack>
using namespace std;
struct node{
int weight;
vector<int> child;
}Node[100];
int m,n,S;//m is the number of non-leaf nodes,n is the count of nodes
bool cmp(int a,int b){//因为你比较的child 是int类型,所以这里也要写Int类型
return Node[a].weight>Node[b].weight;
}
vector<int> ans;//存放路径
void dfs(int nn,int weight) {
//n是到哪个节点了,weight是还剩多少权值没走
if(weight<0) return;
if (weight == 0) {
if(!Node[nn].child.empty()) return ;//注意这里要限制一下,不到叶子节点
//判断的时候不要用v.size来判断有没有
int len = ans.size();
if(len!=0)
cout<<Node[0].weight<<" ";
else{
cout<<Node[0].weight;
}
for (int i = 0; i < len; ++i) {
cout << Node[ans[i]].weight;
if (i != len - 1) {
cout << " ";
}
}
cout<<endl;
return;
}
else{
int si=Node[nn].child.size();
for (int i = 0; i < si; ++i) {
int ide=Node[nn].child[i];
ans.push_back(ide);
dfs(ide,weight-Node[ide].weight);
ans.pop_back();
}
}
}
int main(){
cin>>n>>m>>S;
//cout<<n<<m<<S;
for (int i = 0; i < n; ++i) {
cin>>Node[i].weight;
}
for (int i = 0; i < m; ++i) {//child
int q;
cin>>q;
int x;
cin>>x;
for (int j = 0; j < x; ++j) {
int a;
cin>>a;
Node[q].child.push_back(a);
}
sort(Node[q].child.begin(),Node[q].child.end(),cmp);
}
//其实如果说是按照一定顺序输出所有的结果的话,你就可以先按照这个顺序排序 这样可以dfs按这个顺序搜索
//sort(Node,Node+n,cmp);一开始写成这个QAQ,别把整个换顺序了啊,你的下标和权值还是一一对应的呢
//cout<<"read"<<endl;
//ans.push_back(Node[0].weight);
//cout<<Node[0].weight<<endl;
dfs(0,S-Node[0].weight);//注意要求记录的是权值
return 0;
}