2021.2.2 poj1511

题意：
t组样例
每组有n个点，有m条单向边
m组输入表示从u到v的边权值为w
求往返最小花费
思路：
最短路问题，这题选用dij，spss均可。
建图，正向图和反向图，正跑一次dij，反跑一次dij，所加即是答案。

数据较大，dij需要堆优化省点时间，加用链式前向星建图，vector会T，此时就感觉学一下链式前向星建图十分有必要。

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#include <vector>
#include <list>
#include <set>
#include <utility> // pair
#include <map>
#include <iostream>
#include <sstream>
#include <algorithm> // sort
#include <string>
#include <stack>
#include <queue>
#include <fstream>
#include <bitset>
#include <iomanip>
#include <cmath>
using namespace std;
#define ll long long

const int maxn = 1e6+5, maxm = 2e6+5;
const int INF = 1e9;
//链式前向星
struct node{
    int to, next;
    ll w;
}e[maxm << 1];

int head[maxn << 1], vis[maxm << 1], dis[maxm << 1];

int n, m, cnt;

void init()//初始化操作
{
    for (int i = 0; i <= n<<1; i++)
    {
        head[i] = -1;
        vis[i] = 0;
    }
    cnt = 0;
}

void add_e(int u, int v, ll w) //建边
{
    e[cnt].to = v;
    e[cnt].w = w;
    e[cnt].next = head[u];
    head[u] = cnt;
    cnt++;
}


void dij(int s) //利用优先队列实现优化
{
    priority_queue <pair<int, int>, vector< pair<int, int> >, greater<pair<int, int> > > q;
    for (int i = 0; i <= n<<1; i++)
        dis[i] = INF;

    dis[s] = 0;
    q.push(make_pair(0, s));

    while(!q.empty())
    {
        int u = q.top().second;
        q.pop();

        if (vis[u]) continue;
        vis[u] = 1;

        for (int i = head[u]; i != -1; i = e[i].next)
        {
            int v = e[i].to;
            if (dis[v] > dis[u] + e[i].w)
            {
                dis[v] = dis[u] + e[i].w;
                q.push(make_pair(dis[v], v));
            }
        }
    }
}


int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d", &n, &m);
        init(); 
        for (int i = 0; i < m; i++)
        {
            int u, v;
            ll w;
            scanf("%d %d %lld",&u, &v, &w);
            add_e(u, v, w); 
            add_e(v + n, u + n, w); //这里没有直接开两个图，而是建在一张图上有了两个图的效果，貌似可以省点时间吧
        }

        ll ans = 0;

        dij(1); //正跑
        for (int i = 1; i <= n; i++) ans += dis[i];

        dij(1+n); //反跑
        for (int i = n+2; i <= n << 1 ; i++) ans += dis[i];

        printf("%lld\n",ans);
    }
    return 0;
}