首页 > 代码库 > poj 1511 Dijkstra的另一种用法---求其他点到源点的最短路

poj 1511 Dijkstra的另一种用法---求其他点到源点的最短路

传送门:http://poj.org/problem?id=1511


题目其实到现在我都没读懂,到底是哪里看出来的,ans是源点到各个点最短路的和外加各个点到源点的最短路的和,不过重要的是学到dijkstra的另一种用法--求各个点到源点的距离,原理不难理解,就是把dijkstra求单源最短路径的过程逆过来:

1、取源点加入S集合,设其余点在T集合

2、找到达源点的最小边,将该边连的点加入S,更新T到S的各个点的最短路径,取最短的边,继续2的过程


而dijastra求单源最短路径的过程


1、取源点加入S集合,设其余点在T集合

2、找到达源点的最小边,将该边连的点加入S,更新S到T的各个点的最短路径,取最短的边,继续2的过程


用有向边表示的话,直接用同一个Dijkstra代码就行,只是在建图的时候,注意边反向

关于这个题,可能是我自己INF设置的问题,反证int会WA掉,,,该long long 就过了


//
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;

#define ll long long
const ll INF = 1000000100ll;
const int N =  1000000+10;
ll head[N],lowcost[N],path[N],vis[N],n,m;
ll headf[N],lowcostf[N],pathf[N],visf[N];

struct Edge{
    int to,next;
    int w;
}e[N],ef[N];
struct qnode{
    int v,c;//c--cost
    qnode(int vv=0,int cc=0):v(vv),c(cc){}
    bool operator< (const qnode& r) const{ return c>r.c; }
};
void Addedge(int u, int v, int w, int k)
{
    e[k].w=w;
    e[k].to=v;
    e[k].next=head[u];
    head[u]=k;
    ef[k].w=w;
    ef[k].to=u;
    ef[k].next=headf[v];
    headf[v]=k;
}

void Init()
{
    int u,v,w;
    memset(head,-1,sizeof(head));
    memset(path,-1,sizeof(path));
    memset(vis, 0, sizeof(vis));
    memset(headf,-1,sizeof(headf));
    memset(pathf,-1,sizeof(pathf));
    memset(visf, 0, sizeof(visf));
    scanf("%d%d",&n,&m);
    for(int i=0;i<=n;i++)lowcost[i]=lowcostf[i]=INF;
    for(int i=0;i<m;i++)
        scanf("%d%d%d",&u,&v,&w),Addedge(u-1,v-1,w,i);
}


void Dij(const int src, ll head[], ll vis[], ll lowcost[],Edge e[])
{
    qnode mv;
    int i,j,k,pre;
    priority_queue<qnode>que;
    vis[src]=1,lowcost[src]=0;
    que.push(qnode(src,0));
    for(pre=src,i=1;i<n;i++)
    {
        for(j=head[pre];j!=-1;j=e[j].next)
        {
            k=e[j].to;
            if(!vis[k] && lowcost[pre]+e[j].w<lowcost[k])
            {
                lowcost[k]=lowcost[pre]+e[j].w;
                que.push(qnode(e[j].to,lowcost[k]));
                path[k]=pre;
            }
        }
        while(!que.empty() && vis[que.top().v] == 1)que.pop();//没有直接让最小的在队列的前面因为每次都是要把队列清空的
        if(que.empty())break;
        mv=que.top();que.pop();
        vis[pre=mv.v]=1;
    }
}

ll solve()
{
    ll ans=0;
    for(int i=1;i<n;i++)
        ans=ans+lowcostf[i]+lowcost[i];
    return ans;
}

int main()
{
   // freopen("poj1511.txt","r",stdin);
    int ncase;
    scanf("%d",&ncase);
    while(ncase--)
    {
        Init();
        Dij(0,head,vis,lowcost,e);
        Dij(0,headf,visf,lowcostf,ef);
        printf("%lld\n",solve());
    }
    return 0;
}