1053 Path of Equal Weight (30分) (樹的靜態寫法,先根排序,DFS)

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <queue>
#include <map>
using namespace std;
const int maxn=100010;

int n,m,s,path[maxn];
struct node
{
    int data;
    vector<int>child;
}T[maxn];

bool cmp(int a,int b)
{
    return T[a].data > T[b].data;
}

void DFS(int root,int num,int sum)
{
    if(sum > s) return; //大於sum就返回
    else if(sum == s)
    {
        if(T[root].child.size() != 0) return;  //不是葉節點返回
        for(int i = 0;i<num;i++)
        {
            printf("%d",T[path[i]].data);
            if(i!=num-1)
                printf(" ");
        }
        printf("\n");
        return;
    }

   for(int i = 0;i<T[root].child.size();i++)
   {
       int c = T[root].child[i];   //先根排列達到DFS的目的
       path[num] = c; //記錄路徑
       DFS(c,num+1,sum+T[c].data);
   }
}


int main()
{
    freopen("1.txt","r",stdin);

    scanf("%d%d%d",&n,&m,&s);
    for(int i = 0;i<n;i++)
    {
        int temp;
        scanf("%d",&temp);
        T[i].data = temp;
        T[i].child.clear();
    }
    for(int i = 0;i<m;i++)
    {
        int id,k;
        scanf("%d%d",&id,&k);
        for(int j = 0;j<k;j++)
        {
            int cid;
            scanf("%d",&cid);
            T[id].child.push_back(cid);
        }
        sort(T[id].child.begin(),T[id].child.end(),cmp); //直接子節點從大到小排序達到結果是降序排列
    }
    path[0]=0;
    DFS(0,1,T[0].data);

}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章