题目:
TonyY是一个喜欢到处浪的男人,他的梦想是带着兰兰姐姐浪遍天朝的各个角落,不过在此之前,他需要做好规划。
现在他的手上有一份天朝地图,上面有n个城市,m条交通路径,每条交通路径都是单行道。他已经预先规划好了一些点作为旅游的起点和终点,他想选择其中一个起点和一个终点,并找出从起点到终点的一条路线亲身体验浪的过程。但是他时间有限,所以想选择耗时最小的,你能告诉他最小的耗时是多少吗?
Input
包含多组测试数据。
输入第一行包括两个整数n和m,表示有n个地点,m条可行路径。点的编号为1 - n。
接下来m行每行包括三个整数i, j, cost,表示从地点i到地点j需要耗时cost。
接下来一行第一个数为S,表示可能的起点数,之后S个数,表示可能的起点。
接下来一行第一个数为E,表示可能的终点数,之后E个数,表示可能的终点。
0
思路:
比赛的时候队友是直接对每个起点bfs搜索暴力过的,过后看了题解,有一个很巧妙的思路。因为题目规定了给出的点范围为1-n,那么我们可以设定一个编号从0开始的起点,并且将0到他给出的起点之间建一条权值为0的边,设定一个编号n+1的终点,并且对每一个他给出的终点与n+1这个点建一条权值为0的边。那么最后问题就转化为求0到n+1这个点的最短路。这里直接spfa跑一边就过了。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<cctype>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 0x3f3f3f3f;
const int maxn = 123456;
int T,n,m;
int ans;
int tot;
int dis[maxn],vis[maxn],head[maxn];
//int path[maxn];//记录路径
//int counts[maxn];//记录每个点的访问次数
struct Edge
{
int to;
int w;
int next;
}edge[maxn];
void init()
{
tot=0;
ans=INF;
memset(dis,INF,sizeof(dis));
memset(vis,0,sizeof(vis));
memset(head,-1,sizeof(head));
//memset(counts,0,sizeof(counts));
//memset(path,0,sizeof(path));
}
void add_edge(int u,int v,int w)
{
edge[tot].to=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
void spfa()
{
queue<int>q;
q.push(0);
dis[0]=0;
vis[0]=1;
//counts[0]=1;
//bool ok=true;//判断是否会出现负权边
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
if(u==n+1)
{
ans=min(ans,dis[u]);
continue;
}
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].to;
int w=edge[i].w;
//path[v]=u;
if(dis[u]+w<dis[v])
{
dis[v]=dis[u]+w;
if(vis[v]==0)
{
q.push(v);
vis[v]=1;
/*counts[v++];
if(counts[v]>NN)//超过入队次数上限,说明有负权边
{
ok=false;
break;
}*/
}
}
}
}
}
/*void print_path(int k)
{
if(path[k])
print_path(path[k]);
printf("%d ",k);
}*/
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
int u,v,w;
int s,e;
int num_s,num_e;
init();
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,w);
}
scanf("%d",&num_s);
for(int i=0;i<num_s;i++)
{
scanf("%d",&s);
add_edge(0,s,0);
}
scanf("%d",&num_e);
for(int i=0;i<num_e;i++)
{
scanf("%d",&e);
add_edge(e,n+1,0);
}
spfa();
printf("%d\n",ans);
}
return 0;
}