Description
一颗树n个点,n-1条边,经过每条边都要花费一定的时间,任意两个点都是联通的。
有K个人(分布在K个不同的点)要集中到一个点举行聚会。
聚会结束后需要一辆车从举行聚会的这点出发,把这K个人分别送回去。
请你回答,对于i=1~n,如果在第i个点举行聚会,司机最少需要多少时间把K个人都送回家。
Input
第一行两个数,n,K。
接下来n-1行,每行三个数,x,y,z表示x到y之间有一条需要花费z时间的边。
接下来K行,每行一个数,表示K个人的分布。
Output
输出n个数,第i行的数表示:如果在第i个点举行聚会,司机需要的最少时间。
Sample Input
7 2
1 2 4
1 3 1
2 5 1
2 4 2
4 7 3
4 6 2
3
7
1 2 4
1 3 1
2 5 1
2 4 2
4 7 3
4 6 2
3
7
Sample Output
11
15
10
13
16
15
10
15
10
13
16
15
10
HINT
【数据规模】
K <= N <= 500000
1 <= x,y <= N, 1 <= z <= 1000000
先DP求出从每一个点出发处理完子树再回到该点的时间,然后用dfs序线段树维护到当前点距离最远的人的家
将没有人要回的节点点权设为-INF 当dfs到一个点时 将该节点以及子树中所有节点权值减去当前节点父亲到其距离,其他节点加上这个距离,然后求1-n的最大值,即为当前点与必须要到的点的最远距离;
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
using
namespace
std;
const
int
maxn=500000+10;
int
usize[maxn]={0};
long
long
f[maxn]={0};
vector<
int
>A[maxn];
vector<
int
>C[maxn];
int
n,k;
int
w[maxn];
int
size[maxn];
long
long
F[maxn];
long
long
maxx[maxn]={0};
int
ti=0;
int
tid[maxn];
long
long
Maxx[maxn<<2];
long
long
addv[maxn<<2];
long
long
dep[maxn];
long
long
num[maxn];
bool
isplace[maxn]={0};
inline
void
read(
int
&x){
x=0;
char
c=
getchar
();
while
(c<
'0'
||c>
'9'
)
c=
getchar
();
while
(c>=
'0'
&&c<=
'9'
){
x=x*10+c-
'0'
;
c=
getchar
();
}
}
inline
void
dfs(
int
x,
int
fa,
int
di){
tid[x]=++ti;
size[x]=1;
dep[x]=dep[fa]+di;
for
(
int
i=0;i<A[x].size();i++){
int
u=A[x][i];
if
(u==fa)
continue
;
dfs(u,x,C[x][i]);
usize[x]+=usize[u];
size[x]+=size[u];
}
}
inline
void
dp(
int
x,
int
fa){
if
(size[x]==1)
f[x]=0;
long
long
res=0;
for
(
int
i=0;i<A[x].size();i++){
int
u=A[x][i];
if
(u==fa)
continue
;
dp(u,x);
if
(usize[u])
res+=f[u]+C[x][i]*2;
}
f[x]=min(f[x],res);
if
(usize[x]==0)
f[x]=0;
}
inline
void
getans(
int
x,
int
fa,
int
di){
w[x]=di;
if
(x!=1)
F[x]=F[fa]-2*di*(usize[x]!=0)+2*di*(k-usize[x]!=0);
else
F[x]=f[x];
for
(
int
i=0;i<A[x].size();i++){
int
u=A[x][i];
if
(u==fa)
continue
;
getans(u,x,C[x][i]);
}
}
int
ql,qr;
inline
void
add(
int
o,
int
l,
int
r,
long
long
v){
if
(l>=ql&&r<=qr){
Maxx[o]+=v;
addv[o]+=v;
return
;
}
else
if
(l!=r){
if
(addv[o]!=0){
addv[o<<1]+=addv[o];
addv[o<<1|1]+=addv[o];
Maxx[o<<1]+=addv[o];
Maxx[o<<1|1]+=addv[o];
addv[o]=0;
}
int
mid=(l+r)>>1;
if
(ql<=mid)
add(o<<1,l,mid,v);
if
(qr>mid)
add(o<<1|1,mid+1,r,v);
Maxx[o]=max(Maxx[o<<1],Maxx[o<<1|1]);
}
}
inline
long
long
maxd(
int
o,
int
l,
int
r){
if
(l>=ql&&r<=qr)
return
Maxx[o];
else
{
int
mid=(l+r)>>1;
if
(addv[o]!=0){
addv[o<<1]+=addv[o];
addv[o<<1|1]+=addv[o];
Maxx[o<<1]+=addv[o];
Maxx[o<<1|1]+=addv[o];
addv[o]=0;
}
long
long
ans=0;
if
(ql<=mid)
ans=max(ans,maxd(o<<1,l,mid));
if
(qr>mid)
ans=max(ans,maxd(o<<1|1,mid+1,r));
return
ans;
}
}
inline
void
getmaxdis(
int
x,
int
fa){
ql=tid[x],qr=tid[x]+size[x]-1;
add(1,1,n,-w[x]);
ql=1,qr=tid[x]-1;
if
(ql<=qr)
add(1,1,n,w[x]);
ql=tid[x]+size[x],qr=n;
if
(ql<=qr)
add(1,1,n,w[x]);
ql=1,qr=n;
maxx[x]=maxd(1,1,n);
for
(
int
i=0;i<A[x].size();i++){
int
u=A[x][i];
if
(u!=fa)
getmaxdis(u,x);
}
ql=tid[x],qr=tid[x]+size[x]-1;
add(1,1,n,w[x]);
ql=1,qr=tid[x]-1;
if
(ql<=qr)
add(1,1,n,-w[x]);
ql=tid[x]+size[x],qr=n;
if
(ql<=qr)
add(1,1,n,-w[x]);
}
inline
void
build(
int
o,
int
l,
int
r){
if
(l==r){
Maxx[o]=num[l];
addv[o]=0;
return
;
}
else
{
int
mid=(l+r)>>1;
build(o<<1,l,mid);
build(o<<1|1,mid+1,r);
Maxx[o]=max(Maxx[o<<1],Maxx[o<<1|1]);
}
}
int
main(){
//freopen("kamp.in","r",stdin);
//freopen("kamp.out","w",stdout);
memset
(f,0x7f,
sizeof
(f));
read(n),read(k);
int
x,y,z;
for
(
int
i=1;i<n;i++){
read(x),read(y),read(z);
A[x].push_back(y);
A[y].push_back(x);
C[x].push_back(z);
C[y].push_back(z);
}
for
(
int
i=1;i<=k;i++){
read(x);
usize[x]++;
isplace[x]=1;
}
dfs(1,1,0);
dp(1,1);
getans(1,1,0);
memset
(num,-127/2,
sizeof
(num));
for
(
int
i=1;i<=n;i++)
if
(isplace[i])
num[tid[i]]=dep[i];
build(1,1,n);
getmaxdis(1,1);
for
(
int
i=1;i<=n;i++)
printf
(
"%lld\n"
,F[i]-maxx[i]);
return
0;
}