POJ - 1511（Dijstra+堆优化（优先队列））

poj1511

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<cmath>
using namespace std;
typedef long long LL;
typedef pair<LL,int> p;
const int maxn=1000005;
const LL inf=0x3f3f3f3f;
int n,m,cnt=0;
struct Edge
{
    int s,e;
    LL w;
}ew[maxn];
struct node
{
    int to,next;
    LL w;
}es[maxn];
LL d[maxn];
int used[maxn],head[maxn];
LL sum;
void init()
{
    memset(d,inf,sizeof(d));
    memset(used,0,sizeof(used));
    memset(head,0,sizeof(head));
    cnt=0;
}
void addnode(int u,int v,LL w)
{
    cnt++;
    es[cnt].next=head[u];
    head[u]=cnt;
    es[cnt].to=v;
    es[cnt].w=w;
}
void Dijstra()
{
    priority_queue<p,vector<p>,greater<p> > q;
    while(!q.empty()){
        q.pop();
    }
    d[1]=0;
    q.push(p(d[1],1));
    while(!q.empty()){
        p cur=q.top();
        q.pop();
        int u=cur.second;
        if(used[u]==1)continue;
        used[u]=1;
        for(int i=head[u];i;i=es[i].next){
            LL w=es[i].w;
            int v=es[i].to;
            if(d[u]+w<d[v]){
                d[v]=d[u]+w;
                q.push(p(d[v],v));
            }
        }
    }
    /*for(int i=1;i<=n;i++){
        printf("%lld ",d[i]);
    }
    printf("\n");*/
    for(int i=1;i<=n;i++){
        sum+=d[i];
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        sum=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++){
            scanf("%d%d%lld",&ew[i].s,&ew[i].e,&ew[i].w);
        }
        init();
        for(int i=1;i<=m;i++){
            addnode(ew[i].s,ew[i].e,ew[i].w);
        }
        Dijstra();
        init();
        for(int i=1;i<=m;i++){
            addnode(ew[i].e,ew[i].s,ew[i].w);
        }
        Dijstra();
        printf("%lld\n",sum);
    }
    return 0;
}