hdu 3274

Problem from:http://acm.hdu.edu.cn/showproblem.php?pid=3274

从government选取的villages中任意选一个village作为树的根节点,在树上深搜

此题并不用把树生成出来,把每条边以两条边(双向)的方式存入图中

在图中进行深搜(每个节点只遍历一次)

#include<cstdio>
#include<cstring>
const int MAXN=1010;
int head[MAXN],n,m,t,s,ans;//s在输入被选择的点的时候用,在开始深搜时,s存储的是最后一个被选择的点(作为树的根节点)
bool cho[MAXN],vis[MAXN];//cho[]标记是否被选择,vis[]标记是否已遍历
struct Edge{
    int to,next,w;
}e[MAXN<<1];
void addedge(int u,int v,int c){
    e[t].to=v;
    e[t].w=c;
    e[t].next=head[u];
    head[u]=t++;
}
bool dfs(int rt){//返回以rt为根的树中是否含有被选择的点
    vis[rt]=true;
    int k=head[rt];
    bool flag=false;//标记以rt为根的子树中是否含有被选择的点
    while(k>=0){
        if(!vis[e[k].to] && dfs(e[k].to)){
            flag=true;
            ans+=e[k].w;
        }
        k=e[k].next;
    }
    return (flag||cho[rt]);
}
void init(){
    memset(cho,false,sizeof(cho));
    memset(vis,false,sizeof(vis));
    memset(head,-1,sizeof(head));
    for(int i=0;i<m;i++){
        scanf("%d",&s);
        cho[s]=true;
    }
    int u,v,c;
    t=0;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&u,&v,&c);
        addedge(u,v,c);
        addedge(v,u,c);
    }
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF){
        init();
        ans=0;
        dfs(s);
        printf("%d\n",ans);
    }
    return 0;
}
View Code

 

 

转载于:https://www.cnblogs.com/cshhr/p/3538953.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值