1053 Path of Equal Weight (30 分)难度: 一般 / 树的遍历

Posted 2021-10-26 辉小歌

https://pintia.cn/problem-sets/994805342720868352/problems/994805424153280512

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
vector<int>ve[N];
vector< vector<int> >ans;
vector<int>path;
int n,m,s,w[N];
bool cmp(vector<int> a,vector<int> b) {return a>b;}
void dfs(int u,int sum)
{
    if(ve[u].size()==0)//说明是叶子结点
    {
        if(sum+w[0]==s) ans.push_back(path);//和相等
        return;
    }
    for(int i=0;i<ve[u].size();i++)
    {
        path.push_back(w[ve[u][i]]);//保存路径权重
        dfs(ve[u][i],sum+w[ve[u][i]]);
        path.pop_back();//恢复现场
    }
}
int main(void)
{
    cin>>n>>m>>s;
    for(int i=0;i<n;i++) cin>>w[i];
    for(int i=0;i<m;i++)
    {
        int id,k,x; cin>>id>>k;
        for(int j=0;j<k;j++) cin>>x,ve[id].push_back(x);
    }
    dfs(0,0);
    sort(ans.begin(),ans.end(),cmp);
    for(int i=0;i<ans.size();i++)
    {
        cout<<w[0];
        for(int j=0;j<ans[i].size();j++) cout<<" "<<ans[i][j];
        cout<<endl;
    }
    return 0;
}