http://acm.hdu.edu.cn/showproblem.php?pid=2066
求多源多汇的最短路,n最大为1000,floyd三重循环会超时。继续dijkstra吧。
#include <stdio.h>
#include <algorithm>
#include <set>
#include <map>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#define LL long long
#define _LL __int64
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1100;
int Map[maxn][maxn];
int t,s,d;
int ss[maxn],dd[maxn];
int n;
int dis[maxn];
int vis[maxn];
void dijkstra(int src)
{
memset(dis,INF,sizeof(dis));
memset(vis,0,sizeof(vis));
vis[src] = 1;
for(int i = 1; i <= n; i++)
dis[i] = Map[src][i];
dis[src] = 0;
for(int i = 1; i <= n; i++)
{
int Min = INF,pos;
for(int j = 1; j <= n; j++)
{
if(!vis[j] && dis[j] < Min)
{
Min = dis[j];
pos = j;
}
}
vis[pos] = 1;
for(int j = 1; j <= n; j++)
{
if(!vis[j] && dis[j] > dis[pos] + Map[pos][j])
dis[j] = dis[pos] + Map[pos][j];
}
}
}
int main()
{
while(~scanf("%d %d %d",&t,&s,&d))
{
n = -1;
memset(Map,INF,sizeof(Map));
int u,v,w;
for(int i = 0; i < t; i++)
{
scanf("%d %d %d",&u,&v,&w);
if(Map[u][v] > w)
Map[u][v] = Map[v][u] = w;
n = max( max(u,v),n );
}
for(int i = 0; i < s; i++)
scanf("%d",&ss[i]);
for(int i = 0; i < d; i++)
scanf("%d",&dd[i]);
for(int i = 1; i <= n; i++)
Map[i][i] = 0;
int ans = INF;
for(int i = 0; i < s; i++)
{
dijkstra(ss[i]);
for(int j = 0; j < d; j++)
ans = min(ans,dis[dd[j]]);
}
printf("%d\n",ans);
}
return 0;
}