新年好(最短路+dfs)

重慶城裏有 nn 個車站,mm 條 雙向 公路連接其中的某些車站。

每兩個車站最多用一條公路連接,從任何一個車站出發都可以經過一條或者多條公路到達其他車站,但不同的路徑需要花費的時間可能不同。

在一條路徑上花費的時間等於路徑上所有公路需要的時間之和。

佳佳的家在車站 11,他有五個親戚,分別住在車站 a,b,c,d,ea,b,c,d,e。

過年了,他需要從自己的家出發,拜訪每個親戚(順序任意),給他們送去節日的祝福。

怎樣走,才需要最少的時間?

輸入格式

第一行:包含兩個整數 n,mn,m,分別表示車站數目和公路數目。

第二行:包含五個整數 a,b,c,d,ea,b,c,d,e,分別表示五個親戚所在車站編號。

以下 mm 行,每行三個整數 x,y,tx,y,t,表示公路連接的兩個車站編號和時間。

輸出格式

輸出僅一行,包含一個整數 TT,表示最少的總時間。

數據範圍

1≤n≤500001≤n≤50000,
1≤m≤1051≤m≤105,
1<a,b,c,d,e≤n1<a,b,c,d,e≤n,
1≤x,y≤n1≤x,y≤n,
1≤t≤1001≤t≤100

輸入樣例:

6 6
2 3 4 5 6
1 2 8
2 3 3
3 4 4
4 5 5
5 6 2
1 6 7

輸出樣例:

21

tips:spfa+dfs時間複雜度=O(m*2)+5!; 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 50050,M=200020;

int n,m,res;
int h[N],e[M],w[M],ne[M],idx;
int q[N],st[N],dist[6][N],sourse[6];


void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void spfa(int s,int dist[])
{
    memset(dist,0x3f,N*4);
    memset(st,0,sizeof(st));
    dist[s]=0;
    int hh = 0 ,tt = 1;
    q[hh]=s;st[s]=1;
    while(hh!=tt)
    {
        int t = q[hh++];
        if(hh==N) hh=0;
        st[t]=0;
        for(int i=h[t];~i;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[t]+w[i])
            {
                dist[j]=dist[t]+w[i];
                if(!st[j])
                {
                    q[tt++]=j;
                    st[j]=1;
                    if(tt==N)
                        tt=0;
                }
            }
        }

    }

}
int  dfs(int num,int now,int dis)
{
    if(num==6)
        return dis;
    for(int i=1;i<=5;i++)
    {
        if(!st[i])
        {
            int next = sourse[i];
            st[i]=1;
            res = min(res,dfs(num+1,i,dis+dist[now][next]));
            st[i]=0;
        }
    }
    return res;
}

int main()
{
    memset(h,-1,sizeof(h));
    cin>>n>>m;
    for(int i=1;i<=5;i++) cin>>sourse[i];
    sourse[0]=1;
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    for(int i=0;i<=5;i++)
    {
        spfa(sourse[i],dist[i]);
    }
    res=0x3f3f3f3f;
    memset(st,0,sizeof(st));
    dfs(1,0,0);
    cout<<res<<endl;
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章