博客园 首页 私信博主 显示目录 隐藏目录 管理 动画

Codeforces.786B.Legacy(线段树优化建图 最短路Dijkstra)

题目链接


\(Description\)
\(n\)个点。你有\(Q\)种项目可以选择(边都是有向边,每次给定\(t,u,v/lr,w\)):
t=1,建一条\(u\to v\)的边,花费\(w\)
t=2,由\(u\)\([l,r]\)中任意一些点连边,每次花费\(w\)
t=3,由\([l,r]\)中任意一些点向u连边,每次花费\(w\)
最后求使给定的\(s\)到达点\(i(1\leq i\leq n)\)的最小花费。

\(Solution\)
花费看成每条边的边权,全都连上可以最后直接求最短路。
看到区间操作,能想到线段树。。吗?
原n个点我们还保留。假如连边\((u\to [l,r],w)\),那么像区间修改一样,\(u\)向代表\([l,r]\)的节点连边权为\(w\)的边。
但只这样连上去了却回不到\(n\)个点上。可以把每个点向其左右儿子连边权为\(0\)的边,表示\([l,r]\)的节点就会真的和\([l,r]\)这些点相连(就直接代表了这些点)。
因为是有向边,对于连边\([l,r]\to u\)我们需要另一棵线段树完成。
线段树节点从\(n+1\)编号,叶节点直接用\(1\sim n\)编号很方便,对操作\(1\)可以直接连边。
要给所有节点标不同的号,就不用struct了。


//249ms	73800KB
#include <queue>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
//#define MAXIN 200000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
#define mp std::make_pair
#define pr std::pair<LL,int>
typedef long long LL;
const int N=100005*9,M=100005*(8+34);//M=8N+2Qlogn
const LL INF=0x3f3f3f3f3f3f3f3fll;//longlong!

int Enum,H[N],nxt[M],to[M],len[M],tot,son[N][2];
LL dis[N];
std::priority_queue<pr> q;
//char IN[MAXIN],*SS=IN,*TT=IN;
#define AddEdge(u,v,w) (to[++Enum]=v,nxt[Enum]=H[u],H[u]=Enum,len[Enum]=w)//()!用到三目运算符 
#define lson son[x][0]
#define rson son[x][1]
void Build2(int &x,int l,int r)
{
	if(l==r) {x=l; return;}
	x=++tot;
	Build2(lson,l,l+r>>1), Build2(rson,(l+r>>1)+1,r);
	AddEdge(x,lson,0), AddEdge(x,rson,0);
}
void Build3(int &x,int l,int r)
{
	if(l==r) {x=l; return;}
	x=++tot;
	Build3(lson,l,l+r>>1), Build3(rson,(l+r>>1)+1,r);
	AddEdge(lson,x,0), AddEdge(rson,x,0);
}
void Modify(int l,int r,int x,int L,int R,int t,int u,int w)
{
	if(L<=l && r<=R)
	{
		t==2?AddEdge(u,x,w):AddEdge(x,u,w);
		return;
	}
	int m=l+r>>1;
	if(L<=m) Modify(l,m,lson,L,R,t,u,w);//not rt<<1!
	if(m<R) Modify(m+1,r,rson,L,R,t,u,w);
}

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
void Dijkstra(int s)
{
	static bool vis[N];
	memset(dis,0x3f,sizeof dis);
	dis[s]=0, q.push(mp(0,s));
	while(!q.empty())
	{
		int x=q.top().second; q.pop();
		if(vis[x]) continue;
		vis[x]=1;
		for(int v,i=H[x]; i; i=nxt[i])
			if(dis[v=to[i]]>dis[x]+len[i])
				dis[v]=dis[x]+len[i], q.push(mp(-dis[v],v));
	}
}

int main()
{
	int n=read(),Q=read(),S=read(),rt2,rt3;
	tot=n, Build2(rt2,1,n), Build3(rt3,1,n);
	for(int t,u,v,l,r,w; Q--; )
	{
		if((t=read())==1) u=read(),v=read(),w=read(),AddEdge(u,v,w);
		else u=read(),l=read(),r=read(),w=read(),Modify(1,n,t==2?rt2:rt3,l,r,t,u,w);
	}
	Dijkstra(S);
	for(int i=1; i<=n; ++i) printf("%I64d ",dis[i]==INF?-1ll:dis[i]);

	return 0;
}
posted @ 2018-07-19 17:24  SovietPower  阅读(390)  评论(0编辑  收藏  举报