TonyY是一个喜欢到处浪的男人,他的梦想是带着兰兰姐姐浪遍天朝的各个角落,不过在此之前,他需要做好规划。
现在他的手上有一份天朝地图,上面有n个城市,m条交通路径,每条交通路径都是单行道。他已经预先规划好了一些点作为旅游的起点和终点,他想选择其中一个起点和一个终点,并找出从起点到终点的一条路线亲身体验浪的过程。但是他时间有限,所以想选择耗时最小的,你能告诉他最小的耗时是多少吗?
包含多组测试数据。
输入第一行包括两个整数n和m,表示有n个地点,m条可行路径。点的编号为1 - n。
接下来m行每行包括三个整数i, j, cost,表示从地点i到地点j需要耗时cost。
接下来一行第一个数为S,表示可能的起点数,之后S个数,表示可能的起点。
接下来一行第一个数为E,表示可能的终点数,之后E个数,表示可能的终点。
0<S, E≤n≤100000,0<m≤100000,0<cost≤100。
输出他需要的最短耗时。
4 4 1 3 1 1 4 2 2 3 3 2 4 4 2 1 2 2 3 4
1
着实感受到了打CF与刷专题的力量,今天比赛遇到直接切了,虽然题比较简单,但还是很开心。
思路:
建立一个总起点,与个个起点链接,建立一个总终点,与个个终点链接,跑一个spfa,从起点开始跑,最后求出。
#include<stdio.h>
#include<queue>
using namespace std;
struct aa
{
int u,v,w,next;
} save[400000];
int cnt;
int per[100086],dis[100086],vis[100086];
void add(int u,int v,int w)
{
save[cnt].u=u;
save[cnt].v=v;
save[cnt].w=w;
save[cnt].next=per[u];
per[u]=cnt++;
}
void spfa(int s,int e)
{
dis[s]=0;
queue<int>ycq;
ycq.push(s);
vis[s]=1;
while(!ycq.empty())
{
int u=ycq.front();
vis[u]=0;
ycq.pop();
for(int i=per[u]; i!=-1; i=save[i].next)
{
int v=save[i].v;
int w=save[i].w;
// printf("**%d %d %d\n***",u,v,w);
if(dis[u]+w<dis[v])
{
dis[v]=dis[u]+w;
if(!vis[v])
{
vis[v]=1;
ycq.push(v);
}
}
}
}
printf("%d\n",dis[e]);
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
cnt=0;
for(int i=0; i<=n+1; i++)
{
dis[i]=0x3f3f3f3f;
per[i]=-1;
vis[i]=0;
}
while(m--)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
}
int s,e;
scanf("%d",&s);
while(s--)
{
int temp;
scanf("%d",&temp);
add(0,temp,0);
}
scanf("%d",&e);
while(e--)
{
int temp;
scanf("%d",&temp);
add(temp,n+1,0);
}
spfa(0,n+1);
}
}