PAT (Advanced Level) 1053. Path of Equal Weight (30)

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;

const int maxn=100+10;
vector<int>Tree[maxn];
long long val[maxn];
long long path[maxn];
int n,m;
long long W;

bool cmp(const int &a,const int &b)
{
return val[a]>val[b];
}

void read()
{
scanf("%d%d%lld",&n,&m,&W);
for(int i=0;i<n;i++) scanf("%lld",&val[i]);
for(int i=1;i<=m;i++)
{
int id; scanf("%d",&id);
int k; scanf("%d",&k);
while(k--)
{
int to; scanf("%d",&to);
Tree[id].push_back(to);
}
}
}

void dfs(long long sum,int x,int deep)
{
if(sum>W) return;
if(sum==W)
{
if(Tree[x].size()==0){
printf("%lld",val[0]);
for(int i=0;i<deep;i++)
printf(" %lld",path[i]);
printf("\n");
}
return;
}

for(int i=0;i<Tree[x].size();i++)
{
path[deep]=val[Tree[x][i]];
dfs(sum+val[Tree[x][i]],Tree[x][i],deep+1);
}
}

void work()
{
for(int i=0;i<n;i++)
sort(Tree[i].begin(),Tree[i].end(),cmp);
dfs(val[0],0,0);
}

int main()
{
read();
work();
return 0;
}