bzoj3040 最短路＋配对堆优化

3040: 最短路(road)

Time Limit: 60 Sec  Memory Limit: 200 MB
Submit: 1859  Solved: 564
[Submit][Status][Discuss]

Description

N个点，M条边的有向图，求点1到点N的最短路（保证存在）。
1<=N<=1000000，1<=M<=10000000

Input

第一行两个整数N、M，表示点数和边数。
第二行六个整数T、rxa、rxc、rya、ryc、rp。

前T条边采用如下方式生成：
1.初始化x=y=z=0。
2.重复以下过程T次：
x=(x*rxa+rxc)%rp;
y=(y*rya+ryc)%rp;
a=min(x%n+1,y%n+1);
b=max(y%n+1,y%n+1);
则有一条从a到b的，长度为1e8-100*a的有向边。

后M-T条边采用读入方式：
接下来M-T行每行三个整数x,y,z，表示一条从x到y长度为z的有向边。

1<=x,y<=N，0<z,rxa,rxc,rya,ryc,rp<2^31

 

Output

一个整数，表示1~N的最短路。

Sample Input

3 3
0 1 2 3 5 7
1 2 1
1 3 3
2 3 1

Sample Output

2

请采用高效的堆来优化Dijkstra算法。

 

 

 

第一眼以为是裸堆优化dijkstra

看到hzwer题解发现要用奇怪的堆优化

据说priority_queue里有五大堆，这里用的是配对堆

#include <stdio.h>
#include <ext/pb_ds/priority_queue.hpp>
#include <string.h>
#define inf 10000000000000000LL
#include <algorithm>
using namespace std;
typedef pair<long long , int > pii;
//priority_queue <pii, vector<pii>, greater<pii > > q;
using namespace __gnu_pbds;
typedef __gnu_pbds::priority_queue <pii,  greater<pii >, pairing_heap_tag > heap;
heap::point_iterator id[1000005];
struct node 
{
    int u, v, w, next;
}a[10000005];
int tot, n, m, rxa, rxc, rya, ryc;
long long dis[1000005];
int rp, T, first[1000005];
void addedge(int st, int end, int val)
{
    a[++tot].u = st;a[tot].v = end;a[tot].w = val;
    a[tot].next =first[st];first[st] = tot;
}
int dij(int S)
{
    heap q;
    for (int i = 1; i <= n; i++)dis[i] = inf;
    id[1] = q.push(make_pair(0, 1));
    dis[1] = 0;
    while (!q.empty())
    {
        int u = q.top().second;q.pop();
        for (int e = first[u]; e != -1; e = a[e].next)
        {
            int v = a[e].v;
            if (dis[u] + a[e].w < dis[v])
            {
                dis[v] = dis[u] + a[e].w;
                if (id[v] != 0)
                    q.modify(id[v], make_pair(dis[v], v));
                else id[v] = q.push(make_pair(dis[v], v));
            }
        }
    }
}
int main()
{
    scanf("%d %d", &n, &m);
    scanf("%d %d %d %d %d %d", &T, &rxa, &rxc, &rya, &ryc, &rp);
    int x , y , z , A, B;
    x = y = z = 0;
    memset(first, -1, sizeof(first));
    for (int i = 1; i <= T; i++)
    {
        x = ((long long )x*rxa + rxc) % rp;
        y = ((long long )y*rya + ryc) % rp;
        A = min(x%n+1, y%n+1);
        B = max(y%n+1, y%n+1);
        addedge(A, B, 100000000 - 100*A);   
     }
     for (int i = 1; i <= m - T; i++)
     {
        scanf("%d %d %d", &x, &y, &z);
        addedge(x, y, z);
     }
     dij(1);
     printf("%lld\n", dis[n]);
}