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;
}