[NOIP2012 普及组] 文化之旅 题解

题目描述

有一位使者要游历各国，他每到一个国家，都能学到一种文化，但他不愿意学习任何一种文化超过一次（即如果他学习了某种文化，则他就不能到达其他有这种文化的国家）。不同的国家可能有相同的文化。不同文化的国家对其他文化的看法不同，有些文化会排斥外来文化（即如果他学习了某种文化，则他不能到达排斥这种文化的其他国家）。

现给定各个国家间的地理关系，各个国家的文化，每种文化对其他文化的看法，以及这位使者游历的起点和终点（在起点和终点也会学习当地的文化），国家间的道路距离，试求从起点到终点最少需走多少路。

输入格式

第一行为五个整数 N,K,M,S,TN,K,M,S,T，每两个整数之间用一个空格隔开，依次代表国家个数（国家编号为11 到 NN），文化种数（文化编号为 11 到 KK），道路的条数，以及起点和终点的编号（保证 SS 不等于 TT）；

第二行为 NN个整数，每两个整数之间用一个空格隔开，其中第 ii个数 CiCi​，表示国家 ii的文化为 CiCi​。

接下来的 K行，每行 K 个整数，每两个整数之间用一个空格隔开，记第i行的第 j 个数为 aij，aij=1表示文化 i 排斥外来文化j（i等于 j 时表示排斥相同文化的外来人），aij=0aij​=0 表示不排斥（注意 i 排斥 j 并不保证 j 一定也排斥 i）。

接下来的 MM 行，每行三个整数 u,v,d，每两个整数之间用一个空格隔开，表示国家 u与国家 v有一条距离为d的可双向通行的道路（保证u不等于 v，两个国家之间可能有多条道路）。

输出格式

一个整数，表示使者从起点国家到达终点国家最少需要走的距离数（如果无解则输出−1−1）。

输入输出样例

输入 #1

2 2 1 1 2 
1 2 
0 1 
1 0 
1 2 10 

输出 #1

-1

刚学了Dijkstra算法，于是本蒟蒻来实战演习（没想到一遍A了？？？）

和dalao们不同，因为本蒟蒻硬要用vector，所以加了一大堆stl（优先队列，set）但是竟然能过？！（因为数据太水）

为什么不用邻接矩阵？ 因为刘汝佳的紫书上说 大多数情况下使用priority_queue实现的Dijstra算法是比基于邻接矩阵的Dijstra算法快的。

（然而楼下dalao的题解更好理解些）

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<queue>
#include<set>
using namespace std;
struct edge{
    int from,to,dis;
};
struct data{
    int d,u;
    set <int> r;    //原来想用bool存储，然而总是出问题
    bool operator < (const data &tmp) const{
        return d>tmp.d;
    }  //自定义数值小的先出列
};
set <int> r1;
int a[102][102],c[102],d[102],k;
vector <edge> ed;     //存边
vector <int> g[102];   //存边的编号
priority_queue <data> h;  
void pc(data q,int x)     //标记当前使者会被哪些文化排斥
{
    q.r.insert(x);
    for(int j=1;j<=k;j++)
        if(a[j][x])
            q.r.insert(j);
}
int main()
{
    memset(d,127/3,sizeof(d));
    const int inf=d[0];
    int n,m,s,t,i,j,q1,q2,q3,q4=-1;
    scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
    for(i=1;i<=n;i++) scanf("%d",&c[i]);
    for(i=1;i<=k;i++)
        for(j=1;j<=k;j++)
            scanf("%d",&a[i][j]);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d",&q1,&q2,&q3);
        ed.push_back((edge){q1,q2,q3});
        g[q1].push_back(++q4);
        ed.push_back((edge){q2,q1,q3});
        g[q2].push_back(++q4);
    }     //无向图所以要存两次
    d[s]=0;
    r1.insert(c[s]);
    for(j=1;j<=k;j++)
        if(a[j][c[s]])
            r1.insert(j);
    h.push((data){0,s,r1});        
    while(!h.empty())
    {
        data x=h.top(); h.pop();
        int u=x.u,len=g[u].size();        
        if(x.d!=d[u]) continue;
        for(i=0;i<len;i++)
        {
            edge &e=ed[g[u][i]];
            data p=x;
            if(!p.r.count(c[e.to]))     //不被排斥
                if(d[e.to]>d[u]+e.dis)     //可以走并且这种走法更优
                {
                    d[e.to]=d[u]+e.dis;
                    pc(p,c[e.to]);   //标记
                    h.push((data){d[e.to],e.to,p.r});   //入队
                }
        }    
    }
    if(d[t]==inf) printf("-1");    //判断使者能否到达终点
    else printf("%d",d[t]);
    return 0;
}

在百忙之中写一篇题解也比较辛苦，麻烦拿拿你们皇帝般的纤纤玉手帮我点点赞。