HDU 5326 Work
题意:给一个树型关系图,父亲结点管理儿子结点,问有多少个点管理k个点。
解法:dfs。
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | #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\n" , ans); } return 0; } |
现场是队友做的……后来问他的时候被误导了……以为不只一棵树……然而就是一棵啦……= =
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步