首页 > 代码库 > HDU 3274 City Planning
HDU 3274 City Planning
题意:给你一组数n m n的意思是有多少个村庄,并且给你n-1个关系,m的意思是要你连通的村庄。现在要你求出连通m个村庄所花费的钱
思路:题目一看数据,就像是要你去求最小生成树的子数,但是仔细审题会发现一句“Meanwhile you should use the least money. You may suppose that the initial transportation network makes up a tree.”好吧,原来给你的网络图是一棵树,这样题目就直接简单地太多了,直接DFS搜出这条路来
关键还在于要开一个数组来记录你的初始点所在的边,这样我只需要枚举给出的m个边就可以了
AC代码:
#include<iostream> #include<cstring> #include<algorithm> #include<cstdio> using namespace std; const int maxn=2005; int n,m,root,sum,s,vis[maxn],head[maxn]; //head记录U为起点的边,vis标记所求的边 struct node { int u,v,w,next; //next 记录以u为起点的下一条边 } Edge[maxn]; void addEdge(int u,int v,int w) { Edge[s].u=u,Edge[s].v=v,Edge[s].w=w,Edge[s].next=head[u]; head[u]=s++; //s的功能在于记录你以u为起点的边在哪里 } void dfs(int u,int fa) { int i,j; for(i=head[u];i!=-1;i=Edge[i].next) { int v=Edge[i].v; if(v==fa)continue; //如果再次找到了你的起点,那么就可以停止此次的搜索了,此路不通 dfs(v,u); if(vis[v]) { sum+=Edge[i].w; // printf("sum= %d\n",sum); vis[u]=1; } } } int main() { int i,j,a,b,c; while(scanf("%d%d",&n,&m)!=EOF) { sum=s=0; memset(vis,0,sizeof(vis)); memset(head,-1,sizeof(head)); for(i=1; i<=m; i++) { scanf("%d",&root); vis[root]=1; } for(i=1; i<n; i++) { scanf("%d%d%d",&a,&b,&c); addEdge(a,b,c); addEdge(b,a,c); } // printf("开始:\n"); dfs(root,-1); printf("%d\n",sum); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。