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
#include
#include
#include
#include
#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<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[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<