PAT A1053 Path of Equal Weight (30point(s))

题目链接
思路:用深度遍历算法遍历树,其实就是树的先根遍历,用一个数组模拟栈来记录遍历时从根到结点的路径,一旦当前结点是叶子结点且当前的路径权重等于题目所给权重就输出路径,否则就对孩子结点递归,递归出口就是一个是当前路径权重大于题给的权重或者输出路径。
注意点:
1.要用数组去模拟路径结点栈,而不能用STL的stack或者queue,他们都不能用迭代器访问,必须边取端部元素边出队或出栈,而我们希望保留路径并输出。
2.为避免超出内存,非叶结点的孩子数组用vector来实现。
3.题目是按照字典序非递减的顺序输出满足题意路径各结点的权值,只需要将非叶结点的孩子数组按照权值非递减排序再按序递归就可以使得字典序大的路径总是在前面输出。
4.要特别注意结点的出栈,处理完一个结点,无论是遍历完其所有的孩子结点,还是找到满足题意的路径输出路径,之后都要将当前的结点出栈,表示包含当前结点在内的所有路径都被处理完了。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
int n,m,s;//题目所给参数
int path[100],top=0;//数组模拟路径结点栈,top指向栈顶
struct node{//结点结构体
    int weight;//权重
    vector<int> children;//孩子结点下标数组
}Node[100];//结点数组
void buildtree(){//根据题目给的信息建树
    for(int i=0;i<n;i++){//输入权重
        scanf("%d",&Node[i].weight);
    }
    for(int i=0;i<m;i++){//为每个非叶子结点建立孩子结点数组
        int id,childID,num;
        scanf("%d%d",&id,&num);
        for(int i=0;i<num;i++){
            scanf("%d",&childID);
            Node[id].children.push_back(childID);
        }
    }
}
void print(){
    for(int i=0;i<top;i++){//输出路径
        printf("%d",path[i]);
         if(i!=top-1)printf(" ");
    }
    printf("\n");
}
bool cmp(int a,int b){//比较函数
    return Node[a].weight>Node[b].weight;
    //两个结点,权重大的在前面
}
void preorder(int id,int totalweigth){//先序遍历树
    int Nowweight=totalweigth+Node[id].weight;
    //计算当前路径权重
    if(Nowweight<=s){//当前路径权重小于等于S才会处理
       path[top++]=Node[id].weight;//将当前结点下标加入到路径中
        if(Nowweight==s){//当前路径权重等于题目所给权重
            if(Node[id].children.size()!=0) {
                //但并未到达叶子结点,不是一条完整路径,故不需要再往下递归,当前结点直接出栈
                top--;
                return;
            }
            else print();//找到一条满足题意的完整路径,输出
        }
        else{//当前路径权重小于题目所给权重
            sort(Node[id].children.begin(),Node[id].children.end(),cmp);
            //将孩子结点按权重从大到小排列
            for(vector<int>::iterator it=Node[id].children.begin();it!=Node[id].children.end();it++){
                //按顺序递归遍历孩子结点
                preorder(*it,Nowweight);
            }
            }
            top--;//处理完包含当前结点的所有路径,当前结点出栈
        }
}
int main(){
    scanf("%d%d%d",&n,&m,&s);
    buildtree();
    preorder(0,0);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值