Codeforce.787——D. Legacy 線段樹上優化建圖

大致題意:有n個星球,現在編號爲m的星球是地球,然後有三種傳送方式,1是從v星球到u星球,2是從v到 [l,r] 區間的星球,3是從 [l,r]區間的星球到v星球。現在求地球到其他星球的最短距離。

很明顯,暴力建邊肯定是不行的。不然也對不起這2600分。這題意就告訴我們該去思考區間建圖,而區間的問題就需要用到線段樹了。所以大致思路就是線段樹上優化建圖再跑最短路就可以了。

有空想起來再詳細的寫寫線段樹上優化建圖,先挖個坑。

最後,代碼: 

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+100;
struct node
{
    int v,next,w;
}arr[maxn*50];
int intree[maxn<<2];
int outree[maxn<<2];
int cur;
int cnt;
int head[maxn*20];
int inum[maxn];
int onum[maxn];
void add(int u,int v,int w)
{
    arr[cur].next=head[u];
    arr[cur].v=v;
    arr[cur].w=w;
    head[u]=cur++;
}
void build_intree(int l,int r,int x)
{
    intree[x]=cnt++;
    if(l==r)
    {
        inum[l]=intree[x];
        return ;
    }
    int tp=x<<1;
    int mid=l+r>>1;
    build_intree(l,mid,tp);
    build_intree(mid+1,r,tp+1);
    add(intree[tp],intree[x],0);
    add(intree[tp+1],intree[x],0);
}
void build_outree(int l,int r,int x)
{
    outree[x]=cnt++;
    if(l==r)
    {
        onum[l]=outree[x];
        return ;
    }
    int tp=x<<1;
    int mid=l+r>>1;
    build_outree(l,mid,tp);
    build_outree(mid+1,r,tp+1);
    add(outree[x],outree[tp],0);
    add(outree[x],outree[tp+1],0);
}
void add_edge_outree(int l,int r,int x,int L,int R,int u,int w)
{
    if(r<L||l>R)
        return ;
    if(l>=L&&r<=R)
    {
        add(u,outree[x],w);
        return ;
    }
    int tp=x<<1;
    int mid=l+r>>1;
    add_edge_outree(l,mid,tp,L,R,u,w);
    add_edge_outree(mid+1,r,tp+1,L,R,u,w);
}
void add_edge_intree(int l,int r,int x,int L,int R,int u,int w)
{
    if(r<L||l>R)
        return ;
    if(l>=L&&r<=R)
    {
        //cout<<intree[x]<<" "<<u<<" "<<w<<" "<<x<<"***"<<l<<" "<<r<<" "<<L<<" "<<R<<endl;
        add(intree[x],u,w);
        return ;
    }
    int tp=x<<1;
    int mid=l+r>>1;
    add_edge_intree(l,mid,tp,L,R,u,w);
    add_edge_intree(mid+1,r,tp+1,L,R,u,w);
}
struct Nde
{
    long long int ds,v;
    bool operator <(const Nde &a)const
    {
        return ds>a.ds;
    }
};
priority_queue<Nde>q;
const long long int inf=0x3f3f3f3f3f3f3f3f;
long long int dis[maxn*10];
void dij_heap(int u)
{
    memset(dis,inf,sizeof dis);
    Nde nw,f;
    nw.v=u;
    nw.ds=0;
    dis[u]=0;
    q.push(nw);
    while(!q.empty())
    {
        f=q.top();
        q.pop();
        for(int i=head[f.v];i!=-1;i=arr[i].next)
        {
            if(dis[arr[i].v]>f.ds+arr[i].w)
            {
                dis[arr[i].v]=f.ds+arr[i].w;
                q.push((Nde){dis[arr[i].v],arr[i].v});
            }
        }
    }
}
int main()
{
    int n,q,s;
    int opt,u,l,r,w,v;
    cin>>n>>q>>s;
    memset(head,-1,sizeof head);
    build_intree(1,n,1);
    build_outree(1,n,1);
    for(int i=1;i<=n;++i)
    {
        add(inum[i],onum[i],0);
        add(onum[i],inum[i],0);
    }
    for(int i=0;i<q;++i)
    {
        cin>>opt;
        if(opt==1)
        {
            cin>>v>>u>>w;
            add(inum[v],onum[u],w);
        }
        else if(opt==2)
        {
            cin>>v>>l>>r>>w;
          //  cout<<"?????"<<inum[v]<<" "<<v<<endl;
            add_edge_outree(1,n,1,l,r,inum[v],w);
        }
        else
        {
            cin>>v>>l>>r>>w;
            add_edge_intree(1,n,1,l,r,onum[v],w);
        }
    }
    dij_heap(inum[s]);
    for(int i=1;i<=n;++i)
    {
        if(dis[onum[i]]!=inf)
            cout<<dis[onum[i]]<<" ";
        else
            cout<<-1<<" ";
    }
    cout<<endl;
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章