bzoj3073: [Pa2011]Journeys 线段树优化建图

bzoj3073: [Pa2011]Journeys

链接

BZOJ

思路

区间和区间连边。如何线段树优化建图。

和单点连区间类似的，我们新建一个点，区间->新点->区间。

又转化成了单点连区间的了。

代码

#include <bits/stdc++.h>

using namespace std;

const int N=2e6+7;

int read() {

    int x=0,f=1;char s=getchar();

    for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;

    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';

    return x*f;

}

int n,q,S,dis[N];

struct node {

    int v,nxt,q;

}G[N<<3];

int head[N<<2],tot,cnt;

void add(int u,int v,int q,int opt) {

    if(opt) swap(u,v);

    G[++tot].v=v;

    G[tot].q=q;

    G[tot].nxt=head[u];

    head[u]=tot;

}

struct seg {

    #define ls rt<<1

    #define rs rt<<1|1

    int id[N<<3];

    void build(int l,int r,int rt,int opt) {

        if(l==r) return id[rt]=l,void();

        id[rt]=++cnt;

        int mid=(l+r)>>1;

        build(l,mid,ls,opt);

        build(mid+1,r,rs,opt);

        add(id[rt],id[ls],0,opt);

        add(id[rt],id[rs],0,opt);

    }

    void modify(int L,int R,int u,int q,int l,int r,int rt,int opt) {

        if(L<=l&&r<=R) return add(u,id[rt],q,opt);

        int mid=(l+r)>>1;

        if(L<=mid) modify(L,R,u,q,l,mid,ls,opt);

        if(R>mid) modify(L,R,u,q,mid+1,r,rs,opt);

    }

}a,b;

struct edge {

    int id;

    int val;

    edge(int a=0,int b=0) {id=a,val=b;}

    bool operator < (const edge &b) const {

        return val>b.val;

    }

};

priority_queue<edge> Q;

void dij() {

    memset(dis,0x3f,sizeof(dis));

    dis[S]=0;

    Q.push(edge(S,0));

    while(!Q.empty()) {

        edge u=Q.top();

        Q.pop();

        if(dis[u.id]!=u.val) continue;

        for(int i=head[u.id];i;i=G[i].nxt) {

            int v=G[i].v;

            if(dis[v]>dis[u.id]+G[i].q) {

                dis[v]=dis[u.id]+G[i].q;

                Q.push(edge(v,dis[v]));

            }

        }

    }

}

int main() {

    n=cnt=read(),q=read(),S=read();

    a.build(1,n,1,0);

    b.build(1,n,1,1);

    for(int i=1;i<=q;++i) {

        int x=read(),y=read(),l=read(),r=read();

        ++cnt;

        a.modify(x,y,cnt,1,1,n,1,0);

        b.modify(l,r,cnt,1,1,n,1,1);

        ++cnt;

        a.modify(l,r,cnt,1,1,n,1,0);

        b.modify(x,y,cnt,1,1,n,1,1);

    }

    dij();

    for(int i=1;i<=n;++i) printf("%d\n",dis[i]/2);

    return 0;

}