刘汝佳第六章UVA-1599

这个题疯狂RE.。。。。网上找的代码也不好使,不懂了。

无向图,数据量太大,开邻接矩阵会TLE。所以开邻接向量
先一次BFS,将每一个点的距离找出来。
第二次BFS,将以每一个点的距离为参照,先判断是否是最短路径(距离减1),然后再判断颜色。然后再对这个点所有最短路径且最小颜色的点BFS。注意还要把ans[i]设置为自动更新对应距离的最小颜色。
d[i]和mc设为-1,为初始值。
输出也可以参考一下

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=200000+5;
//无向图
//数据量太大,开邻接矩阵会TLE。所以开邻接向量
vector <int>  g[maxn];//g[i]储存的事第i个节点指向的边
vector <int>  c[maxn];//g[i]储存的是第i个节点指向边对应的权重
int n,m,vis[maxn],d[maxn],ans[maxn];//vis用来储存是否遍历到,d储存各点的距离,ans储存答案的序列,从0到n-1顺序


void init(){
    memset(vis,0,sizeof(vis));
    memset(d,-1,sizeof(d));
    memset(ans,0,sizeof(ans));
    for(int i=0;i<=n;i++){
        g[i].clear();
        c[i].clear();
    }
    for(int i=0;i<m;i++){
        int from,to,w;
        scanf("%d %d %d",&from,&to,&w);
        g[from].push_back(to);
        g[to].push_back(from);
        c[from].push_back(w);
        c[to].push_back(w);
    }
}


void bfs1(int n){
    queue <int >q;
    q.push(n);
    d[n]=0;
    while(!q.empty()){
        int u=q.front();
        q.pop();
        int sz=g[u].size();
        for(int i=0;i<sz;i++){//用d[i]==-1来巧妙判断这个节点是否被遍历过
                int temp=g[u][i];
                if(temp==1){
                    d[1]=d[u]+1;
                    return;
                }
                else if(d[temp]==-1){
                    d[temp]=d[u]+1;
                    q.push(temp);
                }
        }
    }
    return;
}

void bfs2(){
    queue<int> q;
    q.push(1);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        if(u==n){//思考出口
            return;
        }
        int mc=-1;//mc为最小颜色的目标节点的颜色
        int sz=g[u].size();
        for(int i=0;i<sz;i++){
            int tg=g[u][i];
//            printf("tg=%d d[tg]=%d ",tg,d[tg]);
            if(d[tg]==d[u]-1){
                if(mc==-1){
//                    printf("c[%d][%d]=%d\n",u,i,c[u][i]);
                    mc=c[u][i];
                }
                else if(mc>c[u][i]){
                    mc=c[u][i];
                }
            }
        }
//        printf("mc=%d\n",mc);
        int t=d[1]-d[u];
        if(ans[t]==0) ans[t]=mc;
        else{
            ans[t]=min(mc,ans[t]);//这个是为后面重新BFS服务的。第一次到这里的ans[t]都应该是0,但是后面会重新BFS。
        }
        for(int i=0;i<sz;i++){
            int tg=g[u][i];
            if(c[u][i]==mc&&d[tg]==d[u]-1){//满足条件  1.最短路径  2.最小颜色
                q.push(tg);
//                vis[tg]=1;
            }
        }
    }
}

int main(void){
    while(scanf("%d %d",&n,&m)){
        init();
        bfs1(n);
        bfs2();
        printf("%d\n",d[1]);
        for(int i=0;i<d[1];i++){
            if(i) printf(" ");
            printf("%d",ans[i]);
        }
        printf("\n");
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
《算法竞赛入门经典——训练指南》代码仓库 例题代码 限于篇幅,书上并没有给出所有例题的代码,这里给出了所有例题的代码,并且改进了书上的一些代码。 第一章 32题 38份代码 第二章 28题 30份代码 第三章 22题 23份代码 第四章 19题 21份代码 第五章 34题 39份代码 第六章 24题 26份代码 共159题 177份代码 为了最大限度保证代码风格的一致性,所有例题代码均由刘汝佳用C++语言编写。 所有代码均通过了UVa/La的测试,但不能保证程序是正确的(比如数据可能不够强),有疑问请致信rujia.liu@gmail.com,或在googlecode中提出: http://code.google.com/p/aoapc-book/ [最新更新] 2013-04-23 增加字符串中例题10(UVa11992 Fast Matrix Operations)的另一个版本的程序,执行效率较低,但更具一般性,可读性也更好 2013-04-22 增加字符串部分“简易搜索引擎”代码,可提交到UVa10679 2013-04-13 修正Treap中优先级比较的bug(原来的代码实际上是在比较指针的大小!),加入纯名次树代码 2013-03-31 修正UVa1549标程的bug,即buf数组不够大。 增加线段树部分“动态范围最小值”的完整代码 2013-03-23 修正UVa10054标程的bug,即没有判断是否每个点的度数均为偶数。UVa数据已经更新 LA3401修正了代码和文字不一致的问题 UVa11270增加了答案缓存 2013-03-21 增加线段树部分中两个经典问题的完整代码:快速序列操作I和快速序列操作II 2013-02-28 补全所有159道例题的代码

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值