【BZOJ1758】【WC2010】重建计划(点分治,单调队列)
题面
Description
Input
第一行包含一个正整数N,表示X国的城市个数. 第二行包含两个正整数L和U,表示政策要求的第一期重建方案中修建道路数的上下限 接下来的N-1行描述重建小组的原有方案,每行三个正整数Ai,Bi,Vi分别表示道路(Ai,Bi),其价值为Vi 其中城市由1..N进行标号
Output
输出最大平均估值,保留三位小数
Sample Input
4
2 3
1 2 1
1 3 2
1 4 3
Sample Output
2.500
HINT
N<=100000,1<=L<=U<=N-1,Vi<=1000000
题解
我这鬼代码在BZOJ上跑不过去
因为(BZOJ)添加了一组很鬼畜的数据,导致(BZOJ)上会(TLE)
洛谷上能过。
表示完全不会点分治了,这道题目就当复习用。
每次我们二分一个答案,将所有的边权全部减去这个二分的值
此时题目相当于询问是否存在一条边数在([L,U])之间,权值和大于(0)的路径。
考虑每个分治重心的贡献,依次计算当前重心的每一棵子树,
求出所有点的深度(经过的边数),以及权值和
对于每个深度,维护一个前面所有子树的最大权值和。
为了方便计算,按照所有点按照深度排序,这样就用(bfs)便利子树就行了。
开始考虑前面的所有子树与当前子树的贡献
用一个指针从大到小维护所有可以的前面子树中的链
同时用单调队列维护一下单调性
每次取出满足经过的边数在([L,U])之间,并且权值和最大的边出来进行组合,计算一下是否满足二分的答案就好了。。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 111111
inline int read()
{
RG int x=0,t=1;RG char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=-1,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return x*t;
}
int n,L,U;
double ans;
struct Line{int v,next,w;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int size[MAX],Size,root,mx,MD;
bool vis[MAX];
double t[MAX],dis[MAX];
void Getroot(int u,int ff)
{
size[u]=1;int ret=1;
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;if(vis[v]||v==ff)continue;
Getroot(v,u);size[u]+=size[v];
ret=max(ret,size[v]);
}
ret=max(ret,Size-size[u]);
if(mx>ret)mx=ret,root=u;
}
int Q[MAX],H,T,Vis[MAX],dep[MAX];
void bfs(int u,double d)
{
Q[H=T=1]=u;Vis[u]=true;
while(H<=T)
{
int u=Q[H++];
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;if(vis[v]||Vis[v])continue;
dep[v]=dep[u]+1;dis[v]=dis[u]+e[i].w-d;
Q[++T]=v;Vis[v]=true;
}
}
}
int q[MAX];
bool check(int u,double d)
{
bool fl=false;int md=0;
for(int ee=h[u];ee&&!fl;ee=e[ee].next)
{
int v=e[ee].v;if(vis[v])continue;
dis[v]=e[ee].w-d;dep[v]=1;
bfs(v,d);
int hh=1,tt=0,j=md;
for(int i=1;i<=T;++i)
{
while(j>=0&&j+dep[Q[i]]>=L)
{
while(hh<=tt&&t[q[tt]]<t[j])--tt;
q[++tt]=j;--j;
}
while(hh<=tt&&dep[Q[i]]+q[hh]>U)++hh;
if(hh<=tt&&dis[Q[i]]+t[q[hh]]>=0)fl=true;
}
md=max(md,dep[Q[T]]);
for(int i=1;i<=T;++i)
{
Vis[Q[i]]=false;
t[dep[Q[i]]]=max(t[dep[Q[i]]],dis[Q[i]]);
}
}
for(int i=1;i<=md;++i)t[i]=-1e12;
return fl;
}
void Calc(int u)
{
double l=ans,r=MD;
while(r-l>1e-4)
{
double mid=(l+r)/2;
if(check(u,mid))l=mid;
else r=mid;
}
ans=l;
}
void DFS(int u)
{
vis[u]=true;Calc(u);
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;if(vis[v])continue;
Size=size[v];mx=n;Getroot(v,0);
DFS(root);
}
}
int main()
{
n=read();L=read();U=read();
for(int i=1;i<n;++i)
{
int u=read(),v=read(),w=read();
Add(u,v,w);Add(v,u,w);MD=max(MD,w);
}
Size=mx=n;Getroot(1,0);
DFS(root);
printf("%.3lf
",ans);
return 0;
}