码迷,mamicode.com
首页 > 其他好文 > 详细

bfs理解——hdu6386好题

时间:2019-06-01 19:30:59      阅读:70      评论:0      收藏:0      [点我收藏+]

标签:color   i++   std   bit   else   str   oid   hdu   its   

用队列维护,对于每块颜色相同的相连的边进行dfs并记录即可

注意这题要用vis来标记边,不可以标记点

因为点的深度是可以随时更新的(这样的做法不满足贪心条件)

#include<bits/stdc++.h>
using namespace std;
#define maxn 200005
#define INF 0x3f3f3f3f
struct Edge{
    int to,nxt,w,flag;
}edge[maxn<<1];
int head[maxn],tot,n,m;
void init(){memset(head,-1,sizeof head);tot=0;}
void add(int u,int v,int w){
    edge[tot].to=v;edge[tot].w=w;edge[tot].nxt=head[u];head[u]=tot++;
    edge[tot].flag=0;
}

queue<int>q;
int d[maxn];
        
void dfs(int u,int c,int deep){
    if(d[u]>deep){
        d[u]=deep;
        q.push(u);
    }    
    for(int i=head[u];i!=-1;i=edge[i].nxt){
        if(edge[i].flag)continue;
        if(edge[i].w==c){
            edge[i].flag=1;
            dfs(edge[i].to,c,deep);
        }
    }    
}        
void bfs(){
    while(q.size())q.pop();
    memset(d,0x3f,sizeof d);
    q.push(1);
    d[1]=0;
    while(q.size()){
        int u=q.front();q.pop();
        for(int i=head[u];i!=-1;i=edge[i].nxt){
            if(edge[i].flag)continue;
            int v=edge[i].to;
            edge[i].flag=1;
            dfs(v,edge[i].w,d[u]+1);
        }
    }
}



int main(){
    while(cin>>n>>m){
        memset(edge,0,sizeof edge);
        init();
        for(int i=1;i<=m;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);add(v,u,w);
        }
        bfs();
        if(d[n]==INF)puts("-1");
        else cout<<d[n]<<"\n";        
    }    
}

 

bfs理解——hdu6386好题

标签:color   i++   std   bit   else   str   oid   hdu   its   

原文地址:https://www.cnblogs.com/zsben991126/p/10960509.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!