PAT——A1053 Path of Equal Weight

  • Post author:
  • Post category:其他



题目链接:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<set>
using namespace std;
#define maxn 110
struct node{
    int data;
    vector<int>child;
}Node[maxn];
bool cmp(int a,int b)
{
    return Node[a].data>Node[b].data;
}
int n,m,S;
int path[maxn];
void DFS(int index,int numNode,int sum)
{
    if(sum>S)
        return;
    if(sum==S)
    {
        if(Node[index].child.size()!=0)
            return;
        for(int i=0;i<numNode;i++)
        {
            printf("%d",Node[path[i]].data);
            if(i<numNode-1)
                printf(" ");
            else
                printf("\n");
        }
        return;
    }
    for(int i=0;i<Node[index].child.size();i++)
    {
        int child=Node[index].child[i];
        path[numNode]=child;
        DFS(child,numNode+1,sum+Node[child].data);
    }
}
int main()
{
   scanf("%d%d%d",&n,&m,&S);
   for(int i=0;i<n;i++)
   {
       scanf("%d",&Node[i].data);
   }
   int id,k,child;
   for(int i=0;i<m;i++)
   {
       scanf("%d%d",&id,&k);
       for(int j=0;j<k;j++)
       {
           scanf("%d",&child);
           Node[id].child.push_back(child);
       }
       sort(Node[id].child.begin(),Node[id].child.end(),cmp);
   }
   path[0]=0;
   DFS(0,1,Node[0].data);
   return 0;
}



版权声明:本文为qq_42232118原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。