题目大意
N<=5000
V, T < LIMIT <= 10^5
【测试点1~3】N<=2000, T=0
暴力。
【测试点4~6】N<=2000, LIMIT, T<=100
先枚举 C。
求平均肯定先二分,二分完之后相当于要选一些路径,然后每条路径减去一个值,最后判断是否大于一个值。
设 f[i][0~2] 表示:以 i 为根的子树,0 表示点 i 不选,1 表示点 i 选了并且将来还要往上走,2 表示点 i 选了并且将来不往上走(即有条路径在点 i 处转弯)
转移比较显然。
【测试点7~12】N<=1000, LIMIT, T<=10^5
发现 C 不能枚举了。
可是,比如我现在有个 C,而所有的 V[i]+C 都不等于 LIMIT-1,那就说明我这个 C 肯定不是最优的。
因此有用的 C 最多只有 N 个。
dp照常。
【测试点13~20】N<=5000
上述方法被卡常了。
优化1:对于每个 C (已经最多 N 个了),在做二分之前先假设 mid 是当前的 ans,用二分的判断函数判断一下是否可行,若不可行则continue。这样可以去掉很多无用的二分。
优化2:用随机大法打乱 C 的枚举顺序,据说这样期望变成了 n log n。
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int maxn=5005, maxL=1e5+5;
const double eps=1e-8;
int n,L,v[maxn],T;
int tot,go[2*maxn],next[2*maxn],f1[maxn];
void ins(int x,int y)
{
go[++tot]=y;
next[tot]=f1[x];
f1[x]=tot;
}
double f[maxn][3],g[maxn],Mid;
void dfs(int k,int last,int c)
{
double rl=((v[k]+c)%L)/Mid, fir=0, sec=0, sum=0;
for(int p=f1[k]; p; p=next[p]) if (go[p]!=last)
{
dfs(go[p],k,c);
sum+=g[go[p]];
double t=f[go[p]][1]-g[go[p]];
if (t-fir>eps)
{
sec=fir;
fir=t;
} else if (t-sec>eps) sec=t;
}
f[k][0]=sum;
f[k][1]=sum+fir+rl;
f[k][2]=sum+fir+sec+rl;
g[k]=max(f[k][0],max(f[k][1]-1,f[k][2]-1));
}
bool check(double mid,double c)
{
Mid=mid;
memset(f,0,sizeof(f));
memset(g,0,sizeof(g));
dfs(1,0,c);
return (g[1]>1);
}
int vt[maxn],vt0;//存可能的C
bool bz[maxL];
int main()
{
srand(101);
scanf("%d %d",&n,&L);
fo(i,1,n) scanf("%d",&v[i]);
fo(i,1,n-1)
{
int x,y;
scanf("%d %d",&x,&y);
ins(x,y), ins(y,x);
}
scanf("%d",&T);
vt[++vt0]=0;
vt[++vt0]=T;
fo(i,1,n)
{
v[i]%=L;
if (L-1-v[i]<=T && !bz[v[i]]) vt[++vt0]=L-1-v[i], bz[v[i]]=1;
}
random_shuffle(vt+1,vt+1+vt0);//优化2
double ans=0;
fo(ci,1,vt0)
{
if (!check(ans,vt[ci])) continue;//优化1
double l=0, r=(double)n*(L-1);
while (l+eps<=r)
{
double mid=(l+r)/2;
if (check(mid,vt[ci])) l=mid+eps; else r=mid-eps;
}
ans=max(ans,l-eps);
}
printf("%.7f\n",ans);
}