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

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值