题意:给一个树型关系图,父亲结点管理儿子结点,问有多少个点管理k个点。
解法:dfs。
代码:
#include<stdio.h> #include<iostream> #include<algorithm> #include<string> #include<string.h> #include<math.h> #include<limits.h> #include<time.h> #include<stdlib.h> #include<map> #include<queue> #include<set> #include<stack> #include<vector> #define LL long long using namespace std; int n, k; vector <int> v[105]; int ans; int dfs(int u) { int res = 0; for(int i = 0; i < v[u].size(); i++) { res += dfs(v[u][i]); } if(res == k) ans++; return res + 1; } int main() { while(~scanf("%d%d", &n, &k)) { for(int i = 0; i < 105; i++) v[i].clear(); int degree[105] = {0}; ans = 0; for(int i = 0; i < n - 1; i++) { int a, b; scanf("%d%d", &a, &b); v[a].push_back(b); degree[b]++; } for(int i = 1; i <= n; i++) { if(degree[i] == 0) dfs(i); } printf("%d ", ans); } return 0; }
现场是队友做的……后来问他的时候被误导了……以为不只一棵树……然而就是一棵啦……= =