首页 > 代码库 > codeforces 700B Connecting Universities
codeforces 700B Connecting Universities
蛮有意思的。。。。考虑每条边的贡献。
#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#define maxv 200500#define maxe 400500using namespace std;int n,k,x,y,nume=0,g[maxv],size[maxv],fath[maxv];long long ans=0;bool vis[maxv];struct edge{ int v,nxt;}e[maxe];void addedge(int u,int v){ e[++nume].v=v; e[nume].nxt=g[u]; g[u]=nume;}void dfs1(int x){ size[x]=0; for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (v!=fath[x]) { fath[v]=x; dfs1(v); size[x]+=size[v]; } } if (vis[x]) size[x]++;}void dfs2(int x){ for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (v!=fath[x]) { ans+=min(size[v],2*k-size[v]); dfs2(v); } }}int main(){ scanf("%d%d",&n,&k); for (int i=1;i<=2*k;i++) {scanf("%d",&x);vis[x]=true;} for (int i=1;i<=n-1;i++) { scanf("%d%d",&x,&y); addedge(x,y);addedge(y,x); } dfs1(1); dfs2(1); printf("%I64d\n",ans); return 0;}
codeforces 700B Connecting Universities
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。