A1053 Path of Equal Weight (30分)

一、技术总结

  1. 但是首先得明白这里的题意是什么,已经读懂的前提下,这里是给出一棵树,每个结点都有自己的权重,然后权重值之和等于所给数值,输出路径中每个结点的权重。
  2. 首先的明白怎么存储,定义数据域,然后用vector容器进行子结点的存储
  3. 题中要求结果输出要进行非增进行输出,就是这句话,只要第一个数不同,然后输出按照大的进行输出,这就意味着需要排序
  4. 使用sort函数,编写cmp进行排序,将大的权重排到每个子结点的前面。
  5. 编写DFS函数,要明确什么是递归边界,怎么进行遍历。这里无非就是sum大于所给值,还有就是恰好等于时,输出结果。

三、参考代码

#include<bits/stdc++.h>
using namespace std;
const int 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, num, s;//n是结点的数量。num是非叶子结点的数量,s是路径权重和 
int path[maxn];//用于记录路径
//当前访问结点为index,numNode为当前路径path上结点的个数
//sum为当前权重的和 
void DFS(int index, int numNode, int sum){
    if(sum > s) return;
    if(sum == s){
        if(Node[index].child.size() != 0) return;//还没到达叶子结点直接返回
        //到达叶子结点,此时path[]中存放了一条完整路径,输出它
        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];//结点index的第i的子结点编号 
        path[numNode] = child;//将child加到路径path末尾 
        DFS(child, numNode+1, sum+Node[child].data);//递归进入下一层 
    }
}
int main(){
    scanf("%d%d%d", &n, &num, &s);
    int w;//权重 
    for(int i = 0; i < n; i++){
        scanf("%d", &w);
        Node[i].data = w; 
    }
    int index1, sum;//记录下标 
    for(int i = 0; i < num; i++){
        scanf("%d%d", &index1, &sum);
        int index2;
        for(int j = 0; j < sum; j++){
            scanf("%d", &index2);
            Node[index1].child.push_back(index2);
        }
        sort(Node[index1].child.begin(), Node[index1].child.end(), cmp);
    }
    path[0] = 0;
    DFS(0, 1, Node[0].data);
    return 0;
} 

猜你喜欢

转载自www.cnblogs.com/tsruixi/p/12319150.html