树形结构 | 北邮OJ | 92. 统计节点个数

最新推荐文章于 2022-04-28 10:55:20 发布

原创最新推荐文章于 2022-04-28 10:55:20 发布 · 624 阅读

2 ·

CC 4.0 BY-SA版权

OJ刷题专栏收录该内容

49 篇文章

订阅专栏

本文介绍了一个使用深度优先遍历(DFS)算法来分析树形结构数据的C++程序实例。通过预处理阶段的遍历，该程序能够统计特定条件下节点的数量，为理解和实现树的遍历提供了实用的参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://vpn.bupt.edu.cn/http/10.105.242.80/problem/p/92/

#include <bits/stdc++.h>
#define FF(a,b) for(int a=0;a<b;a++)
#define F(a,b) for(int a=1;a<=b;a++)
#define LEN 510000
#define INF 1000000
#define bug(x) cout<<#x<<"="<<x<<endl;

using namespace std;

typedef struct Node{
    int i;
    int p;
    int d;
    vector<int> c;
    Node(){
        p=-1;
        d=0;
    }
}Node;

int whoIsRoot(int * in,int n){
    FF(i,n){
        if(in[i]==0)
            return i;
    }
}

int cnt=0;


void preOrder(Node* a,int r){
    vector<int> &vc=a[r].c;
    bool ok=1;
    FF(i,vc.size()) if(a[r].d<a[vc[i]].d) ok=0;
    if(a[r].d<a[a[r].p].d) ok=0;
    if(ok) cnt++;
//    cout<<a[r].i<<endl;
    FF(i,vc.size()){
        int c=vc[i];
        preOrder(a,c);
    }
}

int main()
{
//    freopen("./in","r",stdin);
    int N;
    scanf("%d",&N);
    while(N--){
        Node nd[1010];
        int indegree[1010]={0};
        int n,x,y;
        scanf("%d",&n);
        FF(i,n-1){
            nd[i].i=i;
            scanf("%d%d",&x,&y);
            nd[x].d++;
            nd[y].d++;
            indegree[y]++;
            nd[x].c.push_back(y);
            nd[y].p=x;
        }
        nd[n-1].i=n-1;
        int root=whoIsRoot(indegree,n);
        cnt=0;
        preOrder(nd,root);
        printf("%d\n",cnt);
    }
    return 0;
}