目录
Problem D 网络传输
题目描述
网络的高效互联与智能传输是提升海量用户服务请求映射效率的重要措施。在这个任务中,你要用最少的传输时间,将特定的数据源发送到指定的网络节点中。
我们给定的网络一共包含N个节点(从1到N编号),其中节点1为数据源。网络中有M条无向边(u,v,w),表示一条传输线连接节点u和节点v,且数据通过这条传输线的平均时间为w。由于传送机制的限制,当一个节点接收到数据之后,它只能选择与它互连的一个节点,并将数据转发到该节点。节点1在初始化时只会发送一次数据,但在传输过程中它可以作为转发节点。
网络中有k个目标节点,你需要计算出该数据从节点1传送到所有K歌节点所需要的最短时间。注意目标节点可以按任意顺序进行传送,数据也可以多次经过同一节点。
输入格式
输入数据第一行是一个整数T(T<=5),表示测试数据的组数。
对于每组测试数据:
第一行是三个正整数N,M,K(2<=N<=1000,1<=M<=N(N-1)/2,K<=10),分别表示节点数,边数和目标节点数。
接下来M行,每行三个整数u,v,w(1<=u,v<=N, 0<=w<=1000,u!=v)。如上所述给出每条传输线。任意两个网络节点之间最多只会有一条边相连。
最后一行是K个整数,给出所有的目标节点的编号,所有目标节点的编号都在2到N之间。
输出格式
对于每组测试数据,输出数据传送到所有K个目标节点的最短时间。
样例输入
2
3 2 2
1 3 1
1 2 3
2 3
6 6 4
1 5 1
5 6 2
2 1 20
2 3 5
3 4 5
6 3 1
2 3 4 6
样例输出
5
19
样例说明
在第一组样例中,最短路线是:1->3->1->2
在第二组样例中,最短路线是:1->5->6->3->2->3->4,或者1->5->6->3->4->3->2
思路
借鉴了大神的博客才做出来,Dijkstra算法+全排列,先将K个目标结点进行全排列,利用Dijkstra算法得到组合间相邻节点的代价,相加比较求最小。注意先利用Dijkstra算法得到距离,存至一个二维数组中,各个组合在二维数组中取数。
代码
#include<stdio.h>
#include<bits/stdc++.h>
using namespace std;
const int MAXV=1010;
const int INF=1000000000;
int n,m,s,G[MAXV][MAXV];
int d[MAXV];
bool vis[MAXV]={false};
int dis[MAXV][MAXV]; //记录各点间最小距离
void Dijkstra(int s){ //s为起点
fill(d,d+MAXV,INF); //fill函数将整个d数组赋为INF
fill(vis,vis+MAXV,false);
d[s]=0; //起点s到自身距离为0
for(int i=1;i<=n;i++){
int u=-1,MIN=INF; //u使d[u]最小,MIN存放该最小的d[u]
for(int j=1;j<=n;j++){ //找到未访问的顶点中d[]最小的
if(vis[j]==false && d[j]<MIN){
u=j;
MIN=d[j];
}
}
//找不到小于INF的d[u],说明剩下的顶点和起点s不连通
if(u==-1) return;
vis[u]=true; //标记u为已访问
for(int v=1;v<=n;v++){
//如果v未访问 && u能到达v && 已u为中介点可以使d[v]更优
if(vis[v]==false && G[u][v]!=INF && d[u]+G[u][v]<d[v]){
d[v]=d[u]+G[u][v]; //优化d[v]
dis[s][v]=d[v];
dis[v][s]=d[v];
}
}
}
}
void Perm(int* a,int k,int n,int & min){ //全排列
if(k==n-1){
int qz=0;
qz+=dis[1][a[0]];
if(qz==INF)return;
for(int i=0;i<n-1;i++){
if(dis[a[i]][a[i+1]]==INF) return;
qz+=dis[a[i]][a[i+1]];
}
if(qz<min){
min=qz;
}
}else{
for(int i=k;i<n;i++){
swap(a[k],a[i]);
Perm(a,k+1,n,min);
swap(a[k],a[i]);
}
}
}
int main(){
int T;
scanf("%d",&T);
while(T--){
int K;
scanf("%d%d%d",&n,&m,&K);
fill(G[0],G[0]+MAXV*MAXV,INF);
for(int i=0;i<m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
G[u][v]=w;
G[v][u]=w;
}
int a[K];
for(int i=0;i<K;i++){
scanf("%d",a+i);
}
Dijkstra(1);
for(int i=0;i<K;i++){
Dijkstra(a[i]);
}
int minqz=INF;
Perm(a,0,K,minqz);
printf("%d\n",minqz);
}
return 0;
}