POJ 1511（最短路+Dijkstra or SPFA）

题意：

给定节点数n,和边数m,边是单向边.问从1节点出发到2,3,...n 这些节点路程和从从这些节点回来到节点1的路程和最小值。

思路：

先从1Dijkstra算出dis1，再把边反向，还是从1Dijkstra算出dis2，两个加起来就是答案。

Dijkstra版

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
#define ll long long
const ll INF=0x3f3f3f3f;
const int maxn=1000005;
using namespace std;
#define pa pair<int,int>
struct node
{
    int to,next,v;
};
node edge[maxn];
int cnt=1,head[maxn];
ll dist1[maxn];
void init()
{
    memset(head,-1,sizeof(head));
    cnt=0;
}
void addedge(int u,int v,int w)
{
    cnt++;
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
    edge[cnt].v=w;
}
void addedge2(int u,int v,int w)
{
    cnt++;
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
    edge[cnt].v=w;
}
void Dijkstra(int n,int now,ll dis[])
{
    priority_queue<pa,vector<pa>,greater<pa> >q;
    int i;
    for (i=1; i<=n; i++)
        dis[i]=INF;
    dis[now]=0;
    q.push(make_pair(0,now));
    while (!q.empty())
    {
        now=q.top().second;
        q.pop();
        for (i=head[now]; i; i=edge[i].next)
            if (dis[now]+edge[i].v<dis[edge[i].to])
            {
                dis[edge[i].to]=dis[now]+edge[i].v;
                q.push(make_pair(dis[edge[i].to],edge[i].to));
            }
    }
}
ll dis1[maxn],dis2[maxn];
int a[maxn][3];
int main()
{
    int n,p,q;
    cin>>n;
    while(n--)
    {

        scanf("%d%d",&p,&q);
        init();
        for(int i=0; i<q; i++)
        {
            scanf("%d%d%d",&a[i][0],&a[i][1],&a[i][2]);
            addedge(a[i][0],a[i][1],a[i][2]);
        }

        Dijkstra(p,1,dis1);
        init();
        for(int i=0; i<q; i++)
        {
            addedge(a[i][1],a[i][0],a[i][2]);
        }
        Dijkstra(p,1,dis2);
        ll sum=0;
        for(int i=1; i<=p; i++)
        {
            sum+=(dis1[i]+dis2[i]);
        }
        printf("%lld\n",sum);
    }
    return 0;
}

SPFA版

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=1000005;
const double eps=1e-8;
const double PI = acos(-1.0);
int a[maxn][3];
struct Edge
{
    int v;
    int cost;
    Edge(int _v=0,int _cost=0):v(_v),cost(_cost) {}
};
vector<Edge> E[maxn];
void addedge(int u,int v,int w)
{
    E[u].push_back(Edge(v,w));
}
bool vis[maxn];//在队列标志
int cnt[maxn];//每个点的入队列次数
ll dist1[maxn],dist2[maxn];
bool SPFA(int start,int n,ll dist[])
{
    memset(vis,false,sizeof(vis));
    for(int i=1; i<=n; i++)dist[i]=inf;
    vis[start]=true;
    dist[start]=0;
    queue<int>que;
    while(!que.empty())que.pop();
    que.push(start);
    memset(cnt,0,sizeof(cnt));
    cnt[start]=1;
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=0; i<E[u].size(); i++)
        {
            int v=E[u][i].v;
            if(dist[v]>dist[u]+E[u][i].cost)
            {
                dist[v]=dist[u]+E[u][i].cost;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                    if(++cnt[v]>n)return false;//cnt[i] 为入队列次数，用来判定是否存在负环回路
                }
            }
        }
    }
    return true;
}
void init()
{
    for(int i=0;i<maxn;i++)
    {
        E[i].clear();
    }
}
int main()
{
    int n,m,t;
    cin>>t;
    while(t--)
    {
        init();
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&a[i][0],&a[i][1],&a[i][2]);
            addedge(a[i][0],a[i][1],a[i][2]);
        }
        SPFA(1,n,dist1);
        init();
        for(int i=0;i<m;i++)
        {
            addedge(a[i][1],a[i][0],a[i][2]);
        }
        ll ans=0;
        SPFA(1,n,dist2);
        for(int i=1;i<=n;i++)
        {
            ans+=(dist1[i]+dist2[i]);
        }
        printf("%lld\n",ans);
    }
    return 0;
}