CF786B Legacy


思路

线段树优化建图
基本思想就是要把一个区间连边拆成log个节点连边,
然后一颗入线段树,一颗出线段树,出线段树都由子节点向父节点连边(可以从子区间出发),入线段树从父节点向子节点连边(可以到达子区间),入线段树上每个节点向出线段树的每个对应节点连边(进来之后可以出去),题目里的边由出线段树连向入线段树
然后最短路就好了

代码

#include 
#include 
#include 
#include 
#define int long long
using namespace std;
int id1[800100*4],id2[800100*4],fir[800100*4],nxt[800100*4],v[800100*4],w[800100*4],lson[800100*4],rson[800100*4],root1,root2,Nodecnt,cnt,n,m,s;
void addedge(int ui,int vi,int wi){
    ++cnt;
    v[cnt]=vi;
    w[cnt]=wi;
    nxt[cnt]=fir[ui];
    fir[ui]=cnt;
}
void build(int l,int r,int &o1,int &o2){
    o1=++Nodecnt;
    o2=++Nodecnt;
    addedge(o2,o1,0);
    if(l==r){
        id1[l]=o1;
        id2[l]=o2;
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,lson[o1],lson[o2]);
    build(mid+1,r,rson[o1],rson[o2]);
    addedge(lson[o1],o1,0);
    addedge(rson[o1],o1,0);
    addedge(o2,lson[o2],0);
    addedge(o2,rson[o2],0);
}
void link(int L,int R,int l,int r,int o,int opt,int v,int w){//0:[l,r]->o  1:o->[l,r]
    if(L<=l&&r<=R){
        if(!opt)
            addedge(o,v,w);
        else
            addedge(v,o,w);
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid)
        link(L,R,l,mid,lson[o],opt,v,w);
    if(R>mid)
        link(L,R,mid+1,r,rson[o],opt,v,w);
}
struct QNode{
    int p,val;
    bool operator < (const QNode &b) const{
        return val>b.val;
    }
};
priority_queue q;
int dis[800100*4],vis[800100*4];
void dijkstra(int s){
    memset(dis,0x3f,sizeof(dis));
    memset(vis,0,sizeof(vis));
    dis[s]=0;
    q.push((QNode){s,0});
    while(!q.empty()){
        QNode x=q.top();
        q.pop();
        if(vis[x.p])
            continue;
        vis[x.p]=true;
        for(int i=fir[x.p];i;i=nxt[i]){
            if(dis[v[i]]>dis[x.p]+w[i]){
                dis[v[i]]=dis[x.p]+w[i];
                q.push((QNode){v[i],dis[v[i]]});
            }
        }
    }
}
signed main(){
    scanf("%lld %lld %lld",&n,&m,&s);
    build(1,n,root1,root2);
    for(int i=1;i<=m;i++){
        int opt;
        scanf("%lld",&opt);
        if(opt==1){
            int u,v,w;
            scanf("%lld %lld %lld",&u,&v,&w);
            addedge(id1[u],id2[v],w);
        }
        else if(opt==2){
            int u,l,r,w;
            scanf("%lld %lld %lld %lld",&u,&l,&r,&w);
            link(l,r,1,n,root2,1,id1[u],w);
        }
        else if(opt==3){
            int l,r,v,w;
            scanf("%lld %lld %lld %lld",&v,&l,&r,&w);
            link(l,r,1,n,root1,0,id2[v],w);
        }
    }
    dijkstra(id2[s]);
    for(int i=1;i<=n;i++)
        printf("%lld ",dis[id1[i]]==0x3f3f3f3f3f3f3f3fLL?-1:dis[id1[i]]);
    return 0;
}

相关