C++——NOIP2015提高组day2 t3——运输计划

题目描述

公元2044年，人类进入了宇宙纪元。
L国有 n 个星球，还有 n-1 条双向航道，每条航道建立在两个星球之间，这 n-1 条航道连通了L国的所有星球。
小P掌管一家物流公司，该公司有很多个运输计划，每个运输计划形如：有一艘物流飞船需要从 ui 号星球沿最快的宇航路径飞行到 vi 号星球去。显然，飞船驶过一条航道是需要时间的，对于航道 j ，任意飞船驶过它所花费的时间为 tj ，并且任意两艘飞船之间不会产生任何干扰。
为了鼓励科技创新，L国国王同意小P的物流公司参与L国的航道建设，即允许小P把某一条航道改造成虫洞，飞船驶过虫洞不消耗时间。
在虫洞的建设完成前小P的物流公司就预接了 m 个运输计划。在虫洞建设完成后，这 m 个运输计划会同时开始，所有飞船一起出发。当这 m 个运输计划都完成时，小P的物流公司的阶段性工作就完成了。
如果小P可以自由选择将哪一条航道改造成虫洞，试求出小P的物流公司完成阶段性工作所需要的最短时间是多少？

输入格式

第一行包括两个正整数 n、m，表示L国中星球的数量及小P公司预接的运输计划的数量，星球从 1 到 n 编号。
接下来 n-1 行描述航道的建设情况，其中第 i 行包含三个整数 ai, bi 和 ti，表示第 i 条双向航道修建在 ai 与 bi 两个星球之间，任意飞船驶过它所花费的时间为 ti。
接下来 m 行描述运输计划的情况，其中第 j 行包含两个正整数 uj 和 vj ，表示第 j 个运输计划是从 uj 号星球飞往 vj 号星球。

输出格式

共 1 行，包含 1 个整数，表示小P的物流公司完成阶段性工作所需要的最短时间。

样例数据 1

输入

6 3 

1 2 3 

1 6 4 

3 1 7 

4 3 6 

3 5 5 

3 6 

2 5 

4 5
输出

11

备注

【样例1说明】

将第1条航道改造成虫洞：则三个计划耗时分别为：11、12、11，故需要花费的时间为12。

将第2条航道改造成虫洞：则三个计划耗时分别为：7、15、11，故需要花费的时间为15。

将第3条航道改造成虫洞：则三个计划耗时分别为：4、8、11，故需要花费的时间为11。

将第4条航道改造成虫洞：则三个计划耗时分别为：11、15、5，故需要花费的时间为15。

将第5条航道改造成虫洞：则三个计划耗时分别为：11、10、6，故需要花费的时间为11。

故将第3条或第5条航道改造成虫洞均可使得完成阶段性工作的耗时最短，需要花费的时间为11。
【数据范围】 

所有测试数据的范围和特点如下表所示：



请注意常数因子带来的程序效率上的影响。
#include<stdio.h>
#include<algorithm>
#define mx 600006
struct node{
__int32 x,y,l;
}edge[mx];
__int32 n,m,x,y,z,tot,Nd,ans;
__int32 next[mx],first[mx],deep[mx],to[mx],f[mx],size[mx],son[mx],pos1[mx],pos2[mx],top[mx],value[mx],cost[mx],dis[mx],s[mx],q[mx];
extern "C" _CRTIMP void* __cdecl __MINGW_NOTHROW memset (void*, int, size_t);
__int32 readint()
{
__int32 i=0;
__int8 ch;
for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
for(;ch>='0'&&ch<='9';ch=getchar()) i=(i<<3)+(i<<1)+ch-'0';
return i;
}
__int32 max(__int32 a,__int32 b)
{
return a > b ? a : b ;
}
void add(__int32 x,__int32 y,__int32 z)
{
to[++tot]=y;
value[tot]=z;
next[tot]=first[x];
first[x]=tot;
}
void dfs1(__int32 x)
{
deep[x]=deep[f[x]]+1;
size[x]=1;
for (__int32 i=first[x];i;i=next[i])
{
__int32 v
4000
=to[i];
if(v==f[x]) continue;
f[v]=x;
dis[v]=dis[x]+value[i];
cost[v]=value[i];
dfs1(v);
size[x]+=size[v];
if(size[v]>size[son[x]]) son[x]=v;
}
}
void dfs2(__int32 x,__int32 tp)
{
top[x]=tp;
pos1[x]=++Nd;
pos2[Nd]=cost[x];
if(son[x]==0) return;
dfs2(son[x],tp);
for (__int32 i=first[x];i;i=next[i])
{
__int32 v=to[i];
if (v==f[x]||v==son[x])continue;
dfs2(v,v);
}
}
__int32 lca(__int32 x,__int32 y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])
{
register int z=y;
y=x;
x=z;
}
x=f[top[x]];
}
return deep[x]<deep[y] ? x : y;
}
bool cmp(node a,node b)
{
return a.l>b.l;
}
void work(__int32 x,__int32 y)
{
while (top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])
{
register int z=y;
y=x;
x=z;
}
++s[pos1[top[x]]];
--s[pos1[x]+1];
x=f[top[x]];
}
if(x==y) return;
if(deep[x]>deep[y])
{
register int z=y;
y=x;
x=z;
}
++s[pos1[x]+1];
--s[pos1[y]+1];
}
__int32 check(__int32 mid)
{
register __int32 S=0;
while(edge[S+1].l>mid) ++S;
if(q[S]!=0) return q[S];
memset(s,0,sizeof(s));
for(__int32 i=1;i<=S;i++) work(edge[i].x,edge[i].y);
register __int32 Max=0,total=0;
for(__int32 i=1;i<=n;i++)
{
total+=s[i];
if(total==S)
Max=max(Max,pos2[i]);
}
q[S]=Max;
return Max;
}
int main()
{
n=readint();
m=readint();
for(__int32 i=1;i<n;++i)
{
x=readint();
y=readint();
z=readint();
add(x,y,z);
add(y,x,z);
}
dfs1(1);
dfs2(1,1);
for(__int32 i=1;i<=m;++i)
{
edge[i].x=readint();
edge[i].y=readint();
edge[i].l=dis[edge[i].x]+dis[edge[i].y]-2*dis[lca(edge[i].x,edge[i].y)];
}
std::sort(edge+1,edge+1+m,cmp);
register __int32 L=0,R=edge[1].l;
while(L<=R)
{
register __int32 mid=(L+R)>>1;
if(edge[1].l-check(mid)>mid)
L=mid+1;
else R=mid-1,ans=mid;
}
__int32 num=0;
__int8 c[12];
do c[++num]=(ans%10)+48,ans/=10; while(ans);
while(num) putchar(c[num--]);
return 0;
}