今天主要学习了线段树优化建图,对于一个区间的点对一个点连边的问题,就可以通过线段树来优化建图。
不过用线段树建图时,是有向边。需要两棵线段树,一棵连接[l,r]到u的边,一棵连接u到[l,r]的边,这样才不会出现0权值走到终点的情况。
附上代码
int build(int l,int r,int tp){
if(l==r) return l;
int node=++tot;
t[tp][node].lr=build(l,mid,tp);
t[tp][node].rr=build(mid+1,r,tp);
int lr=t[tp][node].lr,rr=t[tp][node].rr;
if(tp==1){
G[node].push_back(mk(lr,0));
G[node].push_back(mk(rr,0));
// cout<<node<<" "<<lr<<" 0"<<endl;
// cout<<node<<" "<<rr<<" 0"<<endl;
}
else{
G[lr].push_back(mk(node,0));
G[rr].push_back(mk(node,0));
// cout<<lr<<" "<<node<<" 0"<<endl;
// cout<<rr<<" "<<node<<" 0"<<endl;
}
return node;
}
void link(int node,int l,int r,int L,int R,int x,int w,int tp){
if(l==L && r==R){
if(tp==1){
G[x].push_back(mk(node,w));
//cout<<x<<" "<<node<<" "<<w<<endl;
}
else{
G[node].push_back(mk(x,w));
//cout<<node<<" "<<x<<" "<<w<<endl;
}
return ;
}
int lr=t[tp][node].lr,rr=t[tp][node].rr;
//cout<<l<<" "<<r<<" "<<node<<" "<<lr<<" "<<rr<<endl;
if(R<=mid) link(lr,l,mid,L,R,x,w,tp);
else if(L>mid) link(rr,mid+1,r,L,R,x,w,tp);
else link(lr,l,mid,L,mid,x,w,tp),link(rr,mid+1,r,mid+1,R,x,w,tp);
}