题意:给你一组数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;
}