POJ - 1847 Tram（dijkstra）

题意：有向图有N个点，当电车进入交叉口（某点）时，它只能在开关指向的方向离开。如果驾驶员想要采取其他方式，他/她必须手动更换开关。当驾驶员从路口A驶向路口B时，他/她尝试选择将他/她不得不手动更换开关的次数最小化的路线。

编写一个程序，该程序将计算从交点A到交点B所需的最小开关更改次数。第i个交点处的开关最初指向列出的第一个交点的方向。

分析：对于某点i，去往其直接可到达的点列表中的第一个点时不需要更换开关，等价于边长为0；而其他的点需要更换开关，等价于边长为1。dijkstra裸题。

#include<cstdio>

#include<map>

#include<iostream>

#include<cstring>

#include<queue>

using namespace std;

const int MAXN = 100 + 10;

const int INF = 0x3f3f3f3f;

struct Edge{

    int from, to, dist;

    Edge(int f, int t, int d):from(f), to(t), dist(d){}

};

struct HeapNode{

    int d, u;

    HeapNode(int dd, int uu):d(dd), u(uu){}

    bool operator < (const HeapNode&rhs)const{

        return d > rhs.d;

    }

};

struct Dijkstra{

    int n, m;

    vector<int> G[MAXN];

    vector<Edge> edges;

    bool done[MAXN];

    int d[MAXN];

    int p[MAXN];

    void init(int n){

        this -> n = n;

        for(int i = 1; i <= n; ++i) G[i].clear();

        edges.clear();

    }

    void AddEdge(int from, int to, int dist){

        edges.push_back(Edge(from, to, dist));

        m = edges.size();

        G[from].push_back(m - 1);

    }

    void dijkstra(int s){

        priority_queue<HeapNode> q;

        for(int i = 1; i <= n; ++i) d[i] = INF;

        memset(done, false, sizeof done);

        d[s] = 0;

        q.push(HeapNode(0, s));

        while(!q.empty()){

            HeapNode top = q.top();

            q.pop();

            if(done[top.u]) continue;

            done[top.u] = true;

            int len = G[top.u].size();

            for(int i = 0; i < len; ++i){

                Edge e = edges[G[top.u][i]];

                if(d[top.u] + e.dist < d[e.to]){

                    d[e.to] = d[top.u] + e.dist;

                    p[e.to] = G[top.u][i];

                    q.push(HeapNode(d[e.to], e.to));

                }

            }

        }

    }

}dij;

int main(){

    int N, A, B;

    scanf("%d%d%d", &N, &A, &B);

    dij.init(N);

    for(int i = 1; i <= N; ++i){

        int k, x;

        scanf("%d", &k);

        for(int j = 0; j < k; ++j){

            scanf("%d", &x);

            if(j == 0) dij.AddEdge(i, x, 0);

            else dij.AddEdge(i, x, 1);

        }

    }

    dij.dijkstra(A);

    if(dij.d[B] == INF) printf("-1\n");

    else printf("%d\n", dij.d[B]);

    return 0;

}

巴特西

POJ - 1847 Tram（dijkstra）

最新文章

热门文章