Dijkstra——最小堆优化

Dijkstra+堆优化

• 题目链接：文化之旅
• 写完才发现这道题数据范围好小…完全可以用floyed写。不过正好顺便练练自己的代码能力orz（刚学堆优化，所以主体是剽zyh dalao的代码2333）
• dijkstra+堆优化 思路主要是通过优先队列来贪心的把最短路放进队列，从而减少时间复杂度，变成nlog(n)。代码如下。

 #include
    #include
    #include
    #include
    using namespace std;
    const int INF=0x3f;
    int n,k,m,s,t;
    int culture[110],head[110],dis[110],vis[110],vis2[110],node=0;
    bool reject[110][110];
    struct pan
    {
        int to,next,val;
    }run[21000];
    struct HeapNode{
        int d,u;
        bool operator < (const HeapNode &rhs )const{
            return d>rhs.d;
        }//重载小于号 
    };//用结构体来初始化优先队列 ，使其成为小根堆，来优化时间复杂度（不过这道题数据范围很小，不优化应该也能过） 
    void init()
    {
        memset(dis,INF,sizeof(dis));
        for(int i=1;i<=n;i++)
        {
            head[i]=-1;
        }
    }//初始化 
    void add(int u,int v,int w)
    {
        run[++node].to=v;
        run[node].val=w;
        run[node].next=head[u];
        head[u]=node;
    }
    priority_queueQ;//创建优先队列 
    void dijkstra(int s)
    {
        dis[s]=0;
        Q.push(HeapNode{0,s});
        while(!Q.empty())//小根堆优化的dijkstra很像SPFA，也是进行入队出队的操作。 
        {
            HeapNode x=Q.top();
            int u=x.u;
            Q.pop();
            if(vis[u]) continue;
            vis[u]=1;
            for(int i=head[u];i!=-1;i=run[i].next)
            {
                if(culture[run[i].to]==culture[u]) continue;//如果两个地方的文化相同，那么就进行下一个循环 
                if(dis[run[i].to]>dis[u]+run[i].val&&reject[u][run[i].to]==false)
                {
                    dis[run[i].to]=dis[u]+run[i].val;
                    Q.push((HeapNode){dis[run[i].to],run[i].to});//入队 
                }
            }
        }
    }
    int main()
    {
        int u,v,w;
        cin>>n>>k>>m>>s>>t;
        init();
        for(int i=1;i<=n;i++)
        {
            cin>>culture[i];
        } 
        if(culture[s]==culture[t])
        {
            cout<<-1;
            return 0;
        }
        for(int i=1;i<=k;i++)
        {
            for(int j=1;j<=k;j++)
            {
                cin>>reject[j][i];//这个地方是j对i排斥！！！！ 被这里卡了一次QAQ 
            }
        }
        for(int i=1;i<=m;i++)
        {
            cin>>u>>v>>w;
            if(w<=run[head[u]].val||run[head[u]].val<=0)//由于两条路之间可能有多条途径，所以要特判一下 
            {
                add(u,v,w); 
            } 
            if(w<=run[head[v]].val||run[head[u]].val<=0)
            {
                add(v,u,w);             
            }

        }
        dijkstra(s);
        if(dis[t]==0x3f3f3f3f) cout<<-1;//如果无法到达，就输出-1 
        else cout<return 0;
    }