浪裏個浪 FZU - 2261

題目:

TonyY是一個喜歡到處浪的男人,他的夢想是帶着蘭蘭姐姐浪遍天朝的各個角落,不過在此之前,他需要做好規劃。

現在他的手上有一份天朝地圖,上面有n個城市,m條交通路徑,每條交通路徑都是單行道。他已經預先規劃好了一些點作爲旅遊的起點和終點,他想選擇其中一個起點和一個終點,並找出從起點到終點的一條路線親身體驗浪的過程。但是他時間有限,所以想選擇耗時最小的,你能告訴他最小的耗時是多少嗎?

Input
包含多組測試數據。

輸入第一行包括兩個整數n和m,表示有n個地點,m條可行路徑。點的編號爲1 - n。

接下來m行每行包括三個整數i, j, cost,表示從地點i到地點j需要耗時cost。

接下來一行第一個數爲S,表示可能的起點數,之後S個數,表示可能的起點。

接下來一行第一個數爲E,表示可能的終點數,之後E個數,表示可能的終點。

0

思路:

比賽的時候隊友是直接對每個起點bfs搜索暴力過的,過後看了題解,有一個很巧妙的思路。因爲題目規定了給出的點範圍爲1-n,那麼我們可以設定一個編號從0開始的起點,並且將0到他給出的起點之間建一條權值爲0的邊,設定一個編號n+1的終點,並且對每一個他給出的終點與n+1這個點建一條權值爲0的邊。那麼最後問題就轉化爲求0到n+1這個點的最短路。這裏直接spfa跑一邊就過了。

代碼:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<cctype>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 0x3f3f3f3f;
const int maxn = 123456;
int T,n,m;
int ans;
int tot;
int dis[maxn],vis[maxn],head[maxn];
//int path[maxn];//記錄路徑
//int counts[maxn];//記錄每個點的訪問次數

struct Edge
{
    int to;
    int w;
    int next;
}edge[maxn];

void init()
{
    tot=0;
    ans=INF;
    memset(dis,INF,sizeof(dis));
    memset(vis,0,sizeof(vis));
    memset(head,-1,sizeof(head));
    //memset(counts,0,sizeof(counts));
    //memset(path,0,sizeof(path));
}

void add_edge(int u,int v,int w)
{
    edge[tot].to=v;
    edge[tot].w=w;
    edge[tot].next=head[u];
    head[u]=tot++;
}

void spfa()
{
    queue<int>q;
    q.push(0);
    dis[0]=0;
    vis[0]=1;
    //counts[0]=1;
    //bool ok=true;//判斷是否會出現負權邊
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        if(u==n+1)
        {
            ans=min(ans,dis[u]);
            continue;
        }
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].to;
            int w=edge[i].w;
            //path[v]=u;
            if(dis[u]+w<dis[v])
            {
                dis[v]=dis[u]+w;
                if(vis[v]==0)
                {
                    q.push(v);
                    vis[v]=1;
                    /*counts[v++];
                    if(counts[v]>NN)//超過入隊次數上限,說明有負權邊
                    {
                        ok=false;
                        break;
                    }*/
                }
            }
        }
    }
}

/*void print_path(int k)
{
    if(path[k])
        print_path(path[k]);
    printf("%d ",k);
}*/


int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        int u,v,w;
        int s,e;
        int num_s,num_e;
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
        }
        scanf("%d",&num_s);
        for(int i=0;i<num_s;i++)
        {
            scanf("%d",&s);
            add_edge(0,s,0);
        }
        scanf("%d",&num_e);
        for(int i=0;i<num_e;i++)
        {
            scanf("%d",&e);
            add_edge(e,n+1,0);
        }
        spfa();
        printf("%d\n",ans);
    }
    return 0;
}
發佈了90 篇原創文章 · 獲贊 3 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章