bzoj 3694: 最短路（树链剖分+线段树）

3694: 最短路

Time Limit: 5 Sec   Memory Limit: 256 MB
Submit: 135   Solved: 66
[ Submit][ Status][ Discuss]

Description

给出一个n个点m条边的无向图，n个点的编号从1~n，定义源点为1。定义最短路树如下：从源点1经过边集T到任意一点i有且仅有一条路径，且这条路径是整个图1到i的最短路径，边集T构成最短路树。  给出最短路树，求对于除了源点1外的每个点i，求最短路，要求不经过给出的最短路树上的1到i的路径的最后一条边。

 

Input

第一行包含两个数n和m，表示图中有n个点和m条边。

接下来m行，每行有四个数ai，bi，li，ti，表示图中第i条边连接ai和bi权值为li，ti为1表示这条边是最短路树上的边，ti为0表示不是最短路树上的边。

Output

输出n-1个数，第i个数表示从1到i+1的要求的最短路。无法到达输出-1。

Sample Input

5 9
3 1 3 1
1 4 2 1
2 1 6 0
2 3 4 0
5 2 3 0
3 2 2 1
5 3 1 1
3 5 2 0
4 5 4 0

Sample Output

6 7 8 5

HINT

 对于100%的数据，n≤4000，m≤100000，1≤li≤100000

Source

[ Submit][ Status][ Discuss]

题解：树链剖分+线段树

这个题应该是安全路径的简化版。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 200003
using namespace std;
int tot,n,m,cnt,son[N],next[N],point[N],v[N],c[N];
int len[N],x[N],y[N],t[N],dis[N],f[N],belong[N];
int tr[N],tag[N],inf,pos[N],deep[N],size[N];
void add(int x,int y,int z)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
	tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
	//cout<<x<<" "<<y<<" "<<z<<endl;
}
void dfs(int x,int fa)
{
	deep[x]=deep[fa]+1; size[x]=1;
	for (int i=point[x];i;i=next[i])
	 if (v[i]!=fa) {
	 	f[v[i]]=x;
	 	dis[v[i]]=dis[x]+c[i];
	 	dfs(v[i],x);
	 	size[x]+=size[v[i]];
	 	if (size[son[x]]<size[v[i]]) son[x]=v[i];
	 } 
}
void dfs1(int x,int chain)
{
	belong[x]=chain; pos[x]=++cnt;
	if (!son[x]) return;
	dfs1(son[x],chain);
	for (int i=point[x];i;i=next[i])
	 if (v[i]!=f[x]&&v[i]!=son[x]) 
	  dfs1(v[i],v[i]);
}
void update(int now)
{
	tr[now]=min(tr[now<<1],tr[now<<1|1]);
}
void pushdown(int now)
{
	if(tag[now]!=inf) {
		tr[now<<1]=min(tr[now<<1],tag[now]);
		tr[now<<1|1]=min(tr[now<<1|1],tag[now]);
		tag[now<<1]=min(tag[now<<1],tag[now]);
		tag[now<<1|1]=min(tag[now<<1|1],tag[now]);
		tag[now]=inf;
	}
}
void change(int now,int l,int r,int ll,int rr,int v)
{
	if (ll<=l&&r<=rr) {
		tr[now]=min(tr[now],v);
		tag[now]=min(tag[now],v);
		return;
	}
	int mid=(l+r)/2;
	pushdown(now);
	if (ll<=mid) change(now<<1,l,mid,ll,rr,v);
	if (rr>mid) change(now<<1|1,mid+1,r,ll,rr,v);
//	update(now);
}
int find(int now,int l,int r,int x)
{
	if (l==r) return tr[now];
	int mid=(l+r)/2;
	pushdown(now);
	if (x<=mid) return find(now<<1,l,mid,x);
	else return find(now<<1|1,mid+1,r,x);
}
void solve(int x,int y,int v)
{
	while (belong[x]!=belong[y]) {
		if (deep[belong[x]]<deep[belong[y]]) swap(x,y);
		change(1,1,n,pos[belong[x]],pos[x],v);
		x=f[belong[x]];
	}
	if (deep[x]>deep[y]) swap(x,y);
	if (x==y) return;
	change(1,1,n,pos[x]+1,pos[y],v);
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;i++) {
		scanf("%d%d%d%d",&x[i],&y[i],&len[i],&t[i]);
		if (t[i]) add(x[i],y[i],len[i]);
	}
	dfs(1,0); 
	dfs1(1,1);
	memset(tr,127/3,sizeof(tr));
	memset(tag,127/3,sizeof(tag)); inf=tag[0];
	for (int i=1;i<=m;i++)
	  if (!t[i]) solve(x[i],y[i],dis[x[i]]+dis[y[i]]+len[i]);//cout<<x[i]<<" "<<y[i]<<" "<<dis[x[i]]+dis[y[i]]+len[i]<<endl;
	for (int i=2;i<=n;i++) {
		int t=find(1,1,n,pos[i]);
		if (t==inf) printf("-1 ");
		else printf("%d ",t-dis[i]);
	}
	printf("\n");
}