3743: [Coci2015]Kamp
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 540 Solved: 254
[Submit][Status][Discuss]
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
f[i][0]表示以i为根的子树走完之后回来 f[i][1]表示不回来
f[i][1]=sigma(f[j][1]+2*val(边权))
f[i][0]=f[i][1]-max(f[j][1]-f[j][0]+val)
之后我懒的dp转移 直接用堆维护集合了 多了个log 就卡了卡常
考试时开心被卡
bzoj开心AC
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
#define fastcall __attribute__((optimize("-O3")))
#define IL __inline__ __attribute__((always_inline))
fastcall IL int read()
{
int x=0,c=0;
while(c<'0'||c>'9')c=getchar();
while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
return x;
}
fastcall void print(ll x)
{if(x>=10)print(x/10);putchar(x%10+'0');}
const int N=500100;
int last[N],ecnt;
struct EDGE{int to,nt,val;}e[N<<1];
fastcall IL void add(int u,int v,int val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
struct Que
{
priority_queue<ll>A,B;
fastcall IL void push(ll x){A.push(x);}
fastcall IL void pop(ll x){B.push(x);}
fastcall IL ll top()
{
while(!B.empty()&&!A.empty()&&A.top()==B.top())
A.pop(),B.pop();
return A.empty() ? 0 : A.top();
}
}q[N];
int fa[N],size[N];
ll f[N][2];
fastcall void dfs(int u)
{
for(int i=last[u],v,val;i;i=e[i].nt)
if((v=e[i].to)!=fa[u])
{
fa[v]=u;
dfs(v);
if(size[v])
{
val=e[i].val;
f[u][1]+=f[v][1]+(val<<1);
q[u].push(f[v][1]-f[v][0]+val);
size[u]+=size[v];
}
}
f[u][0]=f[u][1]-q[u].top();
}
ll ans[N];
fastcall void solve(int u)
{
ans[u]=f[u][0];
for(int i=last[u],v,val,tmp[2];i;i=e[i].nt)
if((v=e[i].to)!=fa[u])
{
val=e[i].val;
if(size[v])
{
f[u][1]-=f[v][1]+(val<<1);
q[u].pop(f[v][1]-f[v][0]+val);
f[u][0]=f[u][1]-q[u].top();
}
tmp[0]=size[u];tmp[1]=size[v];
size[u]-=size[v];
size[v]+=size[u];
if(size[u])
{
f[v][1]+=f[u][1]+(val<<1);
q[v].push(f[u][1]-f[u][0]+val);
f[v][0]=f[v][1]-q[v].top();
}
solve(v);
if(size[u])
{
f[v][1]-=f[u][1]+(val<<1);
q[v].pop(f[u][1]-f[u][0]+val);
f[v][0]=f[v][1]-q[v].top();
}
size[u]=tmp[0];size[v]=tmp[1];
if(size[v])
{
f[u][1]+=f[v][1]+(val<<1);
q[u].push(f[v][1]-f[v][0]+val);
f[u][0]=f[u][1]-q[u].top();
}
}
}
fastcall int main()
{
int n=read(),K=read();
register int i,u,v,val;
for(i=1;i<n;++i)
{
u=read();v=read();val=read();
add(u,v,val);add(v,u,val);
}
for(i=1;i<=K;++i)size[read()]=1;
dfs(1);
solve(1);
for(i=1;i<=n;++i)
print(ans[i]),puts("");
return 0;
}
/*
7 2
1 2 4
1 3 1
2 5 1
2 4 2
4 7 3
4 6 2
3
7
11
15
10
13
16
15
10
*/