版权声明：未经本人同意，禁止转载。 https://blog.csdn.net/qq_34022601/article/details/84670605
算法与数据结构实验题 7.3 旅行二
★实验任务
王尼玛又想出去旅游了，由于王尼玛是个抠门的人，所以王尼玛想要花最少的钱去一个自 己想去的地方。王尼玛决定使用火车去旅行，地图上总共有 n 个城市，其中有 k 个城市是王尼玛想去玩的城市。由于王尼玛天天为公司加班加点，公司决定奖励他，免费送他去其中的
m 个城市（也就是说王尼玛能选择这 m 个城市中的一个作起点）。王尼玛数学不好，让你帮他选一条花费最少的路，这条路最终能到某一个他喜欢的城市。
★数据输入
第一行有四个整数 n，T，m 和 k，表示有 n 个城市和 T 条路，公司奖励王尼玛其中 m 个城市，王尼玛喜欢其中的 k 个城市。接下来有 T 行，每行有三个整数 a，b，c，表示 a 城市和 b 城市火车通行的花费（火车通行是双向的），接下来一行有 m 个整数，表示公司奖励的 m 个城市的编号。最后一行有 k 个整数，表示王尼玛喜欢的 k 个城市的编号。( 1<=a,b<=n , 0<=c<=10000 , 2<=n<=50000,1<=T<=200000, 1<=m,k<=n ，数据保证王尼玛一
定能到喜欢的城市)
★数据输出
每个样例输出一行，输出王尼玛能去某个喜欢的城市的最短花费。
| 输入示例 |
输出示例 |
| 10 6 2 3 |
9 |
| 1 3 5 |
|
| 1 4 7 |
|
| 2 8 12 |
|
| 3 8 4 |
|
| 4 9 12 |
|
| 9 10 2 |
|
| 1 2 |
|
| 8 9 10 |
|
解题思路：
和前两天博客文章（一个人的旅行）简直一摸一样。
把奖励的城市连接起来，并设置为边权为0.
#include <cstdio>
#include <map>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int INF=1<<29;
int k[50010],m[50010];
map <int,int> mp[50010];
struct G{
vector <int> adj;
}g[50010];
struct cmp
{
bool operator()(const pair <long long,int> p1, const pair <long long,int> p2)
{
return p1.first > p2.first; //first的小值优先
}
};
void Dijkstra (int n,int kn,int mn)
{
long long dis[50010];
int vis[50010]={0};
pair <long long,int> p;
priority_queue <pair <long long,int> ,vector<pair<long long,int> > ,cmp > q;
for (int i=1;i<=n+1;i++)
{
dis[i]=INF;
}
for (int i=1;i<=mn;i++)
{
dis[m[i]]=0;
p.first=dis[m[i]];
p.second=m[i];
q.push(p);
}
vector <int> :: iterator it;
while (!q.empty())
{
int v=q.top().second;
q.pop();
if (vis[v]==1)
continue;
vis[v]=1;
for (it=g[v].adj.begin();it!=g[v].adj.end();it++)
{
if (vis[*it]==0 && dis[*it] > dis[v] + mp[*it][v])
{
dis[*it] = dis[v] + mp[*it][v];
p.first=dis[*it];
p.second=*it;
q.push(p);
}
}
}
long long ans=INF;
for (int i=1;i<=kn;i++)
{
ans=min(ans,dis[k[i]]);
}
printf ("%lld\n",ans);
}
int main()
{
int n,mn,t,kn,a,b,c;
bool flag=false;
scanf ("%d%d%d%d",&n,&t,&mn,&kn);
for (int i=0;i<t;i++)
{
scanf ("%d%d%d",&a,&b,&c);
g[a].adj.push_back(b);
g[b].adj.push_back(a);
mp[a][b]=c;
mp[b][a]=c;
}
for (int i=1;i<=mn;i++)
{
scanf ("%d",&m[i]);
}
for (int i=1;i<=kn;i++)
{
scanf ("%d",&k[i]);
}
Dijkstra(n,kn,mn);
return 0;
}