标签:his pre vector this def minimum ios cto i++
3 4 2 1 2 2 3 3 4 4 2 1 2 1 3 1 4 6 3 1 2 2 3 3 4 3 5 6 2Sample Output
1 0 1
分析: 问题可以转化为求有多少条边左右两边都有k个以上的点。因为这样的边一定是符合要求的。
题目保证这是一个树,树是无向无环图,所以我们可以从一个点开始DFS搜索,进而找到每一个点左右两边的点数
代码如下:
#include<cstdio> #include<iostream> #include<algorithm> #include<vector> #include<cstring> using namespace std; const int MAXN=2e5+10; int vis[MAXN]; int num[MAXN]; int cnt; vector<int>V[MAXN]; struct node { int u; int v; }poi[MAXN]; int t,n,k,a,b,ans; void dfs(int x) { int u,v; num[x]=1; for(int i=0;i<V[x].size();i++) { u=x; v=V[x][i]; /* if(v==pre) continue;*/ if(!vis[v]) { vis[v]=1; dfs(v); num[u]+=num[v]; } } if(num[u]>=k&&n-num[u]>=k) ans++; } int main() { scanf("%d",&t); while(t--) { memset(vis,0,sizeof(vis)); ans=0; scanf("%d%d",&n,&k); for(int i=0;i<=n;i++)V[i].clear(); int h=n-1; for(int i=0;i<n-1;i++){ scanf("%d%d",&poi[i].u,&poi[i].v); V[poi[i].u].push_back(poi[i].v); V[poi[i].v].push_back(poi[i].u); } vis[1]=1; dfs(1); printf("%d\n",ans); } return 0; }
标签:his pre vector this def minimum ios cto i++
原文地址:http://www.cnblogs.com/a249189046/p/7978898.html