dijkstra堆优化(multiset实现->大大减小代码量)

例题:



Time Limit: 1 second

Memory Limit: 128 MB

【问题描述】

在电视时代,没有多少人观看戏剧表演。Malidinesia古董喜剧演员意识到这一事实，他们想宣传剧院,尤其是古色古香的喜剧片。他们已经打印请帖和所有必要的信息和计划。许多学生被雇来分发这些请柬。每个学生志愿者被指定一个确切的公共汽车站，他或她将留在那里一整天,邀请人们参与。 　　这里的公交系统是非常特殊的:所有的线路都是单向的，连接两个站点。公共汽车离开起始点，到达目的地之后又空车返回起始点。　 学生每天早上从总部出发，乘公交车到一个预定的站点邀请乘客。每个站点都被安排了一名学生。在一天结束的时候,所有的学生都回到总部。现在需要知道的是，学生所需的公交费用的总和最小是多少。

【输入格式】

第1行有两个整数n、m(1<=n,m<=1000000)，n是站点的个数，m是线路的个数。 然后有m行，每行描述一个线路，包括3个整数，起始点，目的地和价格。 总部在第1个站点，价钱都是整数，且小于1000000000。

【输出格式】

输出一行，表示最小费用。

【数据规模】

Sample Input1

4 6

1 2 10

2 1 60

1 3 20

3 4 10

2 4 5

4 1 50

Sample Output1

210

【样例说明】

学生各自从总部被派遣到2,3,4站点，然后又回到总部

1-2-4-1:10+5+50=65

1-3-4-1:20+10+50=80

1-2-4-1:10+5+50=65

65+80+65=210

此题数据规模较大，需要使用较为高效的算法，此题不设小规模数据分数。

【题解】



这题的图算是比较的密集的图。如果用spfa的话。第一个点无法通过。而应该用dijkstra+堆优化来实现。

因为后者更擅长解决密集的图的问题。

做法是这样。

一开始输入图的时候，建一个正图和一个反图。

然后分别在正图和反图上做从起点1到其他点的最短路。

然后获取的dis[0][1..n],dis[1][1..n]分别表示在正图和反图上1到其他点的最短路。

最后答案累加dis[0][i]+disi；

难点在dijkstra的堆优化(堆操作)

->这里用multiset+它的upper_bound函数来实现堆优化;

注意重载判断的时候要把第二个关键字也照顾到，不然upper_bound会失效

获取multiset的头结点的地址

设h为一个结构体multiset

用

__typeof(h.begin()) c=begin();

c就是头结点的地址;

然后x=(*c)就是头元素;

则可以直接访问结构体里的元素了

x.name

x.point

什么的.

upper_bound出来的类型也一样;

也是__typeof(h.begin())



【完整代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define rei(x) scanf("%d",&x)
#define rel(x) scanf("%I64d",&x)

typedef pair<int,int> pii;
typedef pair<LL,LL> pll;

const int MAXN = 1e6+10;
const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};
const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};
const double pi = acos(-1.0);

vector <LL> w[2][MAXN];
vector <int> g[2][MAXN];
__int64 dis[2][1000001] = { 0 },ans = 0;//0是正图，1是反图
int n,m;

struct rec{
    LL dis,x;
    friend bool operator < (rec a,rec b)
    {
        return a.dis < b.dis || (a.dis==b.dis && a.x < b.x);
    }
};
multiset<rec>h;

void dijkstra(int fx) //fx == 0 表示正方向 fx==1表示反方向。
{
    memset(dis[fx],255,sizeof dis[fx]);
    dis[fx][1] = 0;//初始化dis[0] == 0;
    h.clear();
    h.insert({0,1});
    while (!h.empty())
    {
         __typeof(h.begin()) c=h.begin();
        rec tou = (*c);
        h.erase(c);
        int x = tou.x;
        int len = g[fx][x].size();
        rep1(i,0,len-1)
        {
            int y = g[fx][x][i];
            LL co = w[fx][x][i];
            if (dis[fx][y] == -1)
            {
                dis[fx][y] = dis[fx][x] + co;
                h.insert({dis[fx][y],y});
            }
            else
                if (dis[fx][y] > dis[fx][x] + co)
                {
                    c = h.upper_bound({dis[fx][y],y});c--;h.erase(c);
                    dis[fx][y] = dis[fx][x]+co;
                    h.insert({dis[fx][y],y});
                }
        }
    }
}

int main()
{
    //freopen("F:\\rush.txt","r",stdin);
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= m; i++)//用邻接表来存正图和反图
    {
        int x, y;LL z;
        rei(x);rei(y);rel(z);
        g[0][x].pb(y);
        w[0][x].pb(z);
        g[1][y].pb(x);
        w[1][y].pb(z);
    }

    dijkstra(0);//在正图和反图上做最短路
    dijkstra(1);

    for (int i = 2; i <= n; i++)//最后输出到达和返回的最短路的和即可。
        ans += (dis[0][i] + dis[1][i]);
    printf("%I64d\n", ans);
    return 0;
}