首页 > 代码库 > 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;
}