题目链接
http://acm.hdu.edu.cn/showproblem.php?pid=3768
题意
某人搬了新房,要去附近的超市购买东西,不幸的是:他需要的东西在几个不同的超市,所以他需要走几个超市,已知他的房子在0号位置,超市有S个,要求从0到达所有超市并且返回的距离最短。
思路
一开始在题集上看到解法是“最短路+dfs”时,我是拒绝的。写完后抱着必T的心态,准备去网上看题解学习新姿势,结果交上去ac了=_=.
思路很暴力,dfs枚举从0出发顺序的排列,计算之后保存一下之间的最短路值。然后取最小的。。
#include<stdio.h>
#include<cstring>
#include<vector>
#include<iostream>
#include<queue>
#include<map>
using namespace std;
const int INF = 2e9+2;
const int maxn = 100010;
struct edge
{
int v,w;
};
struct node
{
int d,u;
bool operator < (const node &b)const
{
return d>b.d;
}
};
int path[20];
map<int,int> mp[maxn];
int d[maxn],vis[maxn],vis2[maxn];
vector<edge> g[maxn];
int q,ans;
int n,m;
void addedge(int u,int v,int w)
{
g[u].push_back(edge{v,w});
g[v].push_back(edge{u,w});
}
int dijkstra(int s,int t,int n)
{
memset(vis,0,sizeof(vis));
for(int i=0; i<n; i++)d[i]=INF;
d[s]=0;
priority_queue<node> q;
q.push(node{0,s});
while(!q.empty())
{
node temp=q.top();q.pop();
int u=temp.u;
if(vis[u])continue;
vis[u]=1;
for(int i=0,L=g[u].size(); i<L; i++)
{
int v=g[u][i].v;
if(vis[v])continue;
int w=g[u][i].w;
if(d[v]>d[u]+w)
{
d[v]=d[u]+w;
q.push(node{d[v],v});
}
}
}
return d[t];
}
void dfs(int lpos,int num,int sum)
{
if(num==q)
{
int dis;
if(mp[lpos][0])
dis=mp[lpos][0];
else
dis=mp[lpos][0]=mp[0][lpos]=dijkstra(0,lpos,n);
if(ans>sum+dis)
{
ans=sum+dis;
}
return;
}
for(int i=0; i<q; i++)
{
if(!vis2[i])
{
vis2[i]=1;
if(mp[lpos][path[i]]==0)
mp[lpos][path[i]]=mp[path[i]][lpos]=dijkstra(lpos,path[i],n);
dfs(path[i],num+1,sum+mp[lpos][path[i]]);
vis2[i]=0;
}
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)g[i].clear(),mp[i].clear();
for(int i=0; i<m; i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w);
}
scanf("%d",&q);
for(int i=0; i<q; i++)
scanf("%d",&path[i]);
ans=INF;
dfs(0,0,0);
printf("%d\n",ans);
}
}