【HDU 6228、2017ACM/ICPC亚洲区沈阳站】Tree

只要能将题目转化成找关键边即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define N 200003

int cnt
, n, k, ans;
vector<int> V
;

void dfs(int u, int f) {
int i, v;
cnt[u] = 1;
for (i = 0;i < V[u].size();i++) {
v = V[u][i];
if (v == f) continue;
dfs(v, u);
cnt[u] += cnt[v];
if (cnt[v] >= k && n-cnt[v] >= k) ans++;
}
}

int main() {
int T, u, v, i;
scanf("%d", &T);
while (T--) {
scanf("%d%d", &n, &k);
for (i = 1;i <= n;i++) V[i].clear();
for (i = 1;i < n;i++) {
scanf("%d%d", &u, &v);
V[u].push_back(v);
V[v].push_back(u);
}
ans = 0;
memset(cnt, 0, sizeof(cnt));
dfs(1, 0);
printf("%d\n", ans);
}
}