HDU 6228 Tree
题目:http://acm.hdu.edu.cn/showproblem.php?pid=6228
题意:给一个无根树,让你用k种颜色对树进行染色,求每种颜色形成的子树的最大公共边数量
从答案考虑,在答案内的边一定左右两边都有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 | #include<iostream> #include<cstdio> #include<cstring> #include<string> #include<cmath> #include<algorithm> #include<vector> #include<queue> #include<stack> #include<map> #include<set> using namespace std; const int N=2e5+5; vector< int > a[N]; int s[N]; void dfs( int x, int fa) { s[x]=1; for ( int i=0;i<a[x].size();i++) { int t=a[x][i]; if (t==fa) continue ; dfs(t,x); s[x]+=s[t]; } } int main() { int T; scanf ( "%d" ,&T); while (T--) { int n,k; scanf ( "%d%d" ,&n,&k); for ( int i=1;i<=n;i++) a[i].clear(); memset (s,0, sizeof (s)); for ( int i=1;i<n;i++) { int x,y; scanf ( "%d%d" ,&x,&y); a[x].push_back(y); a[y].push_back(x); } dfs(1,0); int ans=0; for ( int i=1;i<=n;i++) if (s[i]>=k&&n-s[i]>=k) ans++; printf ( "%d\n" ,ans); } return 0; } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步