2021-03-05PAT甲级1053 Path of Equal Weight (30 分)思路（两种方法））

PAT甲级1053 Path of Equal Weight (30 分)思路（两种方法，结构体与非结构体）

自己一开始写代码的时候，直接用的简简单单的动态数组，在输出的时候遇到了麻烦，不能将得到的数按照非增顺序输出，转换string又有语法错误，参考《算法笔记》明白了在输入数据的同时要对数据进行处理，将子节点按照重量由大到小排序即可…这样有重新用结构体写了代码…后来再去改之前不用结构体的代码，发现也可以实现。
整道题用的DFS搜索.
学会了cmp排序的新写法，加深了对sort+cmp的理解。
代码如下：

#include<bits/stdc++.h>//两种方法
using namespace std;
int father[110];
struct Node
{
    int w;
    vector<int>child;
}node[110];
int n,m,W;
int cun[110];
int weight=0;
bool cmp(int a,int b)//这题考察的排序真的好强..自己写了好久字符串想转换一下都转换不成
                                    //原来考察的还有读入数据时用sort排序的思维，OK！
{
    return node[a].w>node[b].w;
}
void DFS(int index)
{
    int p;
    // printf("----\n");
    if(node[index].child.size()==0)
    {
        p=0;
        if(weight+node[index].w==W)
        {
            for(int i=index;i!=0;i=father[i])
            {
                cun[p++]=node[i].w;
            }
            cun[p++]=node[0].w;
            for(int i=p-1;i>=0;i--)
            {
                if(i==0)
                    printf("%d\n",cun[i]);
                else
                    printf("%d ",cun[i]);
            }
        }
        return;
    }
    else
    {
        for(int i=0;i<node[index].child.size();i++)
        {
          //  printf("----\n");
            weight=weight+node[index].w;
            DFS(node[index].child[i]);
            weight=weight-node[index].w;
        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&W);
    for(int i=0;i<n;i++)
        scanf("%d",&node[i].w);
    for(int i=0;i<m;i++)
    {
        int temp,t,k;
        scanf("%d%d",&temp,&k);
        for(int j=0;j<k;j++)
        {
            scanf("%d",&t);
            node[temp].child.push_back(t);
            father[t]=temp;
        }
        sort(node[temp].child.begin(),node[temp].child.end(),cmp);
    }
    DFS(0);
}

#include<bits/stdc++.h>
using namespace std;
vector<int>child[110];
char step[110][110];
int n,m;
int W,w[110],w_totle[110]={0};
int weight=0;
int num=0;
int father[110];
int cun[110];
bool cmp(int a,int b)
{
    return w[a]>w[b];
}
void DFS(int index)
{
    int k;
    if(child[index].size()==0)
    {
        num++;k=0;
        if(weight+w[index]==W)
        {
            for(int i=index;i!=0;i=father[i])
                cun[k++]=w[i];
            cun[k++]=w[0];
            for(int i=k-1;i>0;i--)
            printf("%d ",cun[i]);
            printf("%d\n",cun[0]);
        }
        return;
    }
    else
    {
        weight=weight+w[index];
        for(int i=0;i<child[index].size();i++)
        {
            DFS(child[index][i]);
        }
        weight=weight-w[index];
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&W);
    for(int i=0;i<n;i++)
        scanf("%d",&w[i]);
    int t,temp,k;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&temp,&t);
        for(int j=0;j<t;j++)
        {
            scanf("%d",&k);
            child[temp].push_back(k);
            father[k]=temp;
        }
        sort(child[temp].begin(),child[temp].end(),cmp);
        //对于类似child[temp].begin(),child[temp].end()的范围表示的意义，
        //可以在写cmp函数的时候，将其用child[temp]具体的含义表示，比如这里就值得是结点号
    }
    DFS(0);
    return 0;
}