POJ 3768 Shopping dfs+最短路

题目链接

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);
    }
}