bzoj4070 [Apio2015]雅加达的摩天楼 分块+spfa

Description

---

印尼首都雅加达市有 N 座摩天楼，它们排列成一条直线，我们从左到右依次将它们编号为 0 到 N−1。除了这 N 座摩天楼外，雅加达市没有其他摩天楼。

有 M 只叫做 “doge” 的神秘生物在雅加达市居住，它们的编号依次是 0 到 M−1。编号为 i 的 doge 最初居住于编号为 Bi 的摩天楼。每只 doge 都有一种神秘的力量，使它们能够在摩天楼之间跳跃，编号为 i 的 doge 的跳跃能力为 Pi （Pi>0）。
在一次跳跃中，位于摩天楼 b 而跳跃能力为 p 的 doge 可以跳跃到编号为 b−p （如果 0≤b−p < N）或 b+p （如果 0≤b+p < N）的摩天楼。
编号为 0 的 doge 是所有 doge 的首领，它有一条紧急的消息要尽快传送给编 号为 1 的 doge。任何一个收到消息的 doge 有以下两个选择:
跳跃到其他摩天楼上；
将消息传递给它当前所在的摩天楼上的其他 doge。
请帮助 doge 们计算将消息从 0 号 doge 传递到 1 号 doge 所需要的最少总跳跃步数，或者告诉它们消息永远不可能传递到 1 号 doge。

所有数据都保证 0≤Bi < N。
1≤N≤30000
1≤Pi≤30000
2≤M≤30000

Solution

---

直接建图最短路是会T的，边数太大了
回想之前的线段树建图，这里可以分块建图
设块的大小为size，对于b[i]>=size的边可以直接连，这样最多只有n√n条
对于b[i]< size的边，我们给每个点开size个辅助点，连i的第b[i]个辅助点和i+b[i]的第b[i]个辅助点，这样最多只有√n种取值，也就是n√n条
这样就比较优秀了，理论上讲堆优dij和spfa都能上但是不知为何我就是错了

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <queue>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))
#define fi first
#define se second

const int N=3030005;
const int E=3030005;

typedef std:: pair <int,int> pair;

std:: priority_queue <pair> heap;
std:: queue <int> que;

struct edge {int y,w,next;} e[E*5];

int dis[N],ls[N],edCnt;
int size,n,m;

bool vis[N];

void add_edge(int x,int y,int w) {
    e[++edCnt]=(edge) {y,w,ls[x]}; ls[x]=edCnt;
}

int get(int x,int y) {
    return x*n+y;
}

void dijkstra(int st,int ed) {
    fill(dis,63); dis[st]=0;
    heap.push(pair(0,st));
    while (!heap.empty()) {
        pair top=heap.top(); heap.pop();
        if (vis[top.se]) continue;
        vis[top.se]=1;
        for (int i=ls[top.se];i;i=e[i].next) {
            if (!vis[e[i].y]&&top.fi+e[i].w<dis[e[i].y]) {
                dis[e[i].y]=top.fi+e[i].w;
                heap.push(pair(dis[e[i].y],e[i].y));
            }
        }
    }
    if (dis[ed]!=dis[0]) {
        printf("%d\n", dis[ed]);
    } else puts("-1");
}

void spfa(int st,int ed) {
    fill(dis,63); dis[st]=0;
    que.push(st); vis[st]=1;
    while (!que.empty()) {
        int now=que.front(); que.pop();
        for (int i=ls[now];i;i=e[i].next) {
            if (dis[now]+e[i].w<dis[e[i].y]) {
                dis[e[i].y]=dis[now]+e[i].w;
                if (!vis[e[i].y]) {
                    que.push(e[i].y);
                    vis[e[i].y]=1;
                }
            }
        }
        vis[now]=0;
    }
    if (dis[ed]!=dis[0]) {
        printf("%d\n", dis[ed]);
    } else puts("-1");
}

int main(void) {
    scanf("%d%d",&n,&m);
    size=std:: min(100,(int)sqrt(n));
    rep(i,1,size) rep(j,1,n) {
        add_edge(get(i,j),j,0);
    }
    rep(i,1,size) rep(j,1,n-i) {
        add_edge(get(i,j),get(i,i+j),1);
        add_edge(get(i,i+j),get(i,j),1);
    }
    int st,ed;
    rep(i,1,m) {
        int x,y; scanf("%d%d",&x,&y); x++;
        if (i==1) st=x;
        if (i==2) ed=x;
        if (y>size) {
            for (int j=1;j*y+x<=n;j++) add_edge(x,x+j*y,j);
            for (int j=1;x-j*y>=1;j++) add_edge(x,x-j*y,j);
        } else add_edge(x,get(y,x),0);
    }
    spfa(st,ed);
    // dijkstra(st,ed);
    return 0;
}