Dij+堆优化

  • Post author:
  • Post category:其他


struct rec
{
    int v,w;//终点,花费。
};
vector<rec> edge[N*2];//起点,注意清空
struct cmp
{
    bool operator()(int a,int b)
    {
        return dis[a]>dis[b];
    }
};
void add(int x,int y)
{
    edge[x].push_back((rec){y+n,1});
    edge[y+n].push_back((rec){x,1});

}
void Dijkstra()
{
    priority_queue<int,vector<int>,cmp> Q;
    memset(dis,-1,sizeof(dis));
    memset(vis,0,sizeof(vis));
    int i,u,v;
    Q.push(1);
    dis[1]=0;
    while(!Q.empty())
    {
        u=Q.top();
        Q.pop();
        vis[u]=0;
        if(u==n)
            break;
        for(i=0;i<edge[u].size();i++)
        {
            v=edge[u][i].v;
            if(dis[v]==-1 || dis[v]>dis[u]+edge[u][i].w)
            {
                dis[v]=dis[u]+edge[u][i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    Q.push(v);
                }
            }
        }
    }
}



版权声明:本文为qq_33183401原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。