首页 > 代码库 > XDOJ_1104_dfs
XDOJ_1104_dfs
http://acm.xidian.edu.cn/problem.php?id=1104
答案为边的两边的节点数相乘,dfs一遍的节点数。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<vector> #include<cmath> using namespace std; int n,m,from[1005],to[1005]; vector<int> v[1005]; int dfs(int s,int t) { int ans = 1; for(int i = 0;i < v[s].size();i++) { if(v[s][i] == t) continue; ans += dfs(v[s][i],s); } return ans; } int main() { int T; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); for(int i = 1;i <= n;i++) v[i].clear(); for(int i = 1;i < n;i++) { int a,b; scanf("%d%d",&a,&b); from[i] = a; to[i] = b; v[a].push_back(b); v[b].push_back(a); } for(int i = 1;i <= m;i++) { int x; scanf("%d",&x); int t = dfs(to[x],from[x]); printf("%d\n",t*(n-t)); } } return 0; }
XDOJ_1104_dfs
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。