hdu 5423 Rikka with Tree(dfs)

解析:

显然一棵树是独特的当且仅当任意处于每一个深度的点数是”1 1 1 1 … 1 1 x”。所以直接DFS一下求出每一个点到根的距离然后判断一下就好了 。

my code

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#define pb push_back
using namespace std;
const int N = 1005;

vector<int> G[N];
int depth[N], maxde;
int n;

void init() {
    memset(depth, 0, sizeof(depth));
    maxde = 0;
    for(int i = 1; i <= n; i++) {
        G[i].clear();
    }
}

void dfs(int u, int pre, int de) {
    maxde = max(de, maxde);
    depth[de]++;
    for(int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        if(v == pre) continue;
        dfs(v, u, de+1);
    }
}

bool judge() {
    for(int i = 1; i < maxde; i++) {
        if(depth[i] != 1) return false;
    }
    return true;
}

int main() {
    int u, v;
    while(scanf("%d", &n) != EOF) {
        init();
        for(int i = 1; i < n; i++) {
            scanf("%d%d", &u, &v);
            G[u].pb(v);
            G[v].pb(u);
        }
        dfs(1, -1, 1);
        puts(judge() ? "YES" : "NO");
    }
    return 0;
}

你可能感兴趣的:(HDU,5423)