这种题目是重点题型,跟前面刷到的DFS和BFS有很多相同的地方,树的先根遍历对应DFS,树的层序遍历对应BFS,这种题目也有很强的规律可循,刷的多了,就会越来越得心应手了。
题目描述如下:
题目大致意思:
给出一个带权重的树,求出从根节点到叶子节点的路径上的权重和等于给定值的路径。
大致思路:
这道题可以用DFS来做,具体的实现过程如下面的代码所示:
提交结果:
有一个测试用例未能通过,暂时放在第二轮刷题时解决。
提交的代码如下:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct Node
{
int weight;
vector<int> child;
};
Node node[100];
vector<int> temp;
vector<vector<int> >ans;
int n,m,s;
void DFS(int index,int sum);
bool cmp(vector<int> arr1,vector<int> arr2)
{
for(int i=0;i<arr1.size();i++)
{
if(arr1[i]>arr2[i])
return true;
if(arr1[i]<arr2[i])
return false;
}
}
int main()
{
cin>>n>>m>>s;
for(int i=0;i<n;i++)
{
cin>>node[i].weight;
}
for(int i=0;i<m;i++)
{
int id,k;
cin>>id>>k;
for(int j=0;j<k;j++)
{
int temp;
cin>>temp;
node[id].child.push_back(temp);
}
}
temp.push_back(node[0].weight);
DFS(0,node[0].weight);
sort(ans.begin(),ans.end(),cmp);
for(int i=0;i<ans.size();i++)
{
for(int j=0;j<ans[i].size()-1;j++)
{
cout<<ans[i][j]<<" ";
}
cout<<ans[i][ans[i].size()-1]<<endl;
}
}
void DFS(int index,int sum)
{
if(node[index].child.size()==0)
{
if(sum==s)
ans.push_back(temp);
return;
}
for(int i=0;i<node[index].child.size();i++)
{
temp.push_back(node[node[index].child[i]].weight);
DFS(node[index].child[i],sum+node[node[index].child[i]].weight);
temp.pop_back();
}
}
本次提交后累计得分为1444。