Newcoder 109 E.求長度(spfa+狀壓)
阿新 • • 發佈:2018-11-09
Description
給定一幅 個點 條邊的圖和 個一定要經過的點,問從 號點出發,經過這 個點再回到 號點的最短路徑長度是多少。
Input
第一行一個整數 表示資料組數
對於每組資料,第一行兩個整數 表示點數和邊數
接下來 行,每行三個整數 表示 之間有一條長度為 的雙向邊
接下來一個整數
接下來 行每行一個整數表示一定要經過的點
資料保證有解
Output
行,每行一個整數表示答案。
Sample Input
1
4 6
0 1 1
1 2 1
2 3 1
3 0 1
0 2 5
1 3 5
3
1
2
3
Sample Output
4
Solution
首先分別以這
個點以及
點為起點跑一遍
得到這
個點到其他點的最短距離,不妨設這
個點為
,而
,以
表示已經經過
狀態的點(用
個
位表示這
個點的經過狀態)且當前在
點的最短路徑長度,那麼列舉下一個要經過的目標點
,使得
中第
位為
,進而有轉移
最後
即為答案,時間複雜度
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
#define maxn 100005
#define INF 2e9
struct edge
{
int to,next,cost;
}g[2*maxn];
int head[maxn],tol;
int dis[11][maxn];
bool vis[maxn];
int T,n,m,s,a[11],dp[2200][11];
void init()
{
memset(head,-1,sizeof(head));
tol=0;
}
void add(int u,int v,int c)
{
g[tol].cost=c;
g[tol].to=v;
g[tol].next=head[u];
head[u]=tol++;
}
void spfa(int s,int *dis)
{
memset(vis,0,sizeof(vis));
queue<int>que;
for(int i=0;i<n;i++)dis[i]=INF;
dis[s]=0,vis[s]=1;
que.push(s);
while(!que.empty())
{
int u=que.front();que.pop();
vis[u]=0;
for(int i=head[u];i!=-1;i=g[i].next)
{
int v=g[i].to,c=g[i].cost;
if(dis[v]>dis[u]+c)
{
dis[v]=dis[u]+c;
if(!vis[v])vis[v]=1,que.push(v);
}
}
}
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
init();
while(m--)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w),add(v,u,w);
}
scanf("%d",&s);
for(int i=1;i<=s;i++)
{
scanf("%d",&a[i]);
spfa(a[i],dis[i]);
}
s++;
a[0]=0;
spfa(0,dis[0]);
int N=1<<s;
for(int i=0;i<N;i++)
for(int j=0;j<s;j++)
dp[i][j]=INF;
dp[0][0]=0;
for(int S=0;S<N;S++)
for(int i=0;i<s;i++)
if(dp[S][i]!=INF)
for(int j=0;j<s;j++)
if(!((S>>j)&1))
dp[S^(1<<j)][j]=min(dp[S^(1<<j)][j],dp[S][i]+dis[i][a[j]]);
int ans=INF;
for(int i=0;i<s;i++)ans=min(ans,dp[N-1][i]+dis[i][0]);
printf("%d\n",ans);
}
return 0;
}