2018 Multi-University Training Contest 7 HDU 6386 Age of Moyu

Bryce1010模板

http://acm.hdu.edu.cn/showproblem.php?pid=6386
题意：
给出一个图，问你从节点1到节点n话费多少？
图中的每条边有归属，如果走过的连续两条边归属同一人，则只需花费1，否则花费2.

思路：
总体上是Dijkstra，然后修改修改模板。
在建图时，对于每条入图中的边，给一个归属标记owner，然后在进行优先队列的节点入队操作时，给节点添加一个归属标记owner，表明上一条边的归属，当与下一条边比较时，比较节点归属（表示上一条边归属）是否与当前比较的边归属相同，不相同则距离贡献+1；继续执行Dijkstra算法流程。

#include<bits/stdc++.h>

using namespace std;
const int MAXN=1e5+10;
const int INF=0x3f3f3f3f;
//建边
struct Edge
{
    int owner,to;
    Edge(int o=0,int t=0):owner(o),to(t){}
};

//建节点
struct Node
{
     int owner,d,u;
    bool operator <(const Node& rhs)const
    {
        return d>rhs.d;
    }
    Node(int o=0,int d=0,int u=0):owner(o),d(d),u(u){}

};

int n,m;
vector<int>G[MAXN];
vector<Edge>edges;
bool vis[MAXN];
int dis[MAXN];//s到各点的距离
int p[MAXN];//最短路中的上一条弧


void init(int n)
{
    for(int i=0;i<=n;i++)
    {
        G[i].clear();
        vis[i]=0;
        dis[i]=-1;
    }
    edges.clear();
}

//无向图
void addEdge(int from,int to,int book)
{
    edges.push_back(Edge{book,to});
    G[from].push_back(edges.size()-1);
    edges.push_back(Edge{book,from});
    G[to].push_back(edges.size()-1);
}
//st.owner=-1  st.d=0   st.u=1
void Dijkstra(Node st)
{
    priority_queue<Node>Que;
    memset(vis,0,sizeof(vis));
    Que.push(st);
    int cnt=0;
    while(!Que.empty()&&cnt<=n)
    {
        Node x=Que.top();Que.pop();
        int u=x.u;//next node
        if(vis[u])continue;
        vis[u]=true;
        dis[u]=x.d;
        cnt++;
        for(int i=0;i<G[u].size();i++)
        {
            Edge& e=edges[G[u][i]];
            if(vis[e.to])continue;
            int cost=x.d;
            if(e.owner!=x.owner)
            {
                cost++;
            }
            Que.push(Node{e.owner,cost,e.to});
        }
    }
}


int main()
{
    ios::sync_with_stdio(false);
    while(cin>>n>>m)
    {
        memset(dis,-1,sizeof(dis));
        init(n);
        int A,B,C;
        for(int i=1;i<=m;i++)
        {
            cin>>A>>B>>C;
            addEdge(A,B,C);
        }
        Node st=Node{-1,0,1};
        Dijkstra(st);

        cout<<dis[n]<<endl;

    }
    return 0;
}