欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【COGS2652】秘术「天文密葬法」(长链剖分,分数规划)

程序员文章站 2022-05-08 18:26:20
...

【COGS2652】秘术「天文密葬法」(长链剖分,分数规划)

题面

Cogs
上面废话真多,建议直接拉到最下面看一句话题意吧:

给个树,第i个点有两个权值ai和bi,现在求一条长度为m的路径,使得Σai/Σbi最小

题解

看到这个式子就是裸的分数规划吧。。。
二分一个答案\(k\),式子变成了找一条长度为\(m\)的路径(题目里面路径长度的定义是点数)
满足\(\sum a-k\sum b\le 0\)
首先直接把\(m=-1\)也就是没有限制的点直接判掉,这个东西没有任何意义。
(其实\(m=1\)的和没有限制是一样的。。。。)
现在把每个点的点权化为\(V_i=a_i-kb_i\),要找出一条长度为\(m\)的路径使得点权和最小。
很明显,对于一个点的子树中,我们只需要维护在相同深度中,距离它的点权和最小的那个点的点权和,距离当前根节点这个东西很不好转移,直接变成距离整棵树的根节点的距离,这样子直接减一下就好了。
那么链就只有两种情况了,一种是在一个\(LCA\)的地方转一下,这个在暴力更新轻儿子贡献的时候直接算一下。另一种是根节点向下延伸了\(m\),这个在更新完了只有直接\(check\)一下就好了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define ll long long
#define MAX 200200
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
double ans=1e18,V[MAX];
int a[MAX],b[MAX];
int n,m,md[MAX],dep[MAX],hson[MAX],len[MAX];
void dfs1(int u,int ff)
{
    md[u]=dep[u]=dep[ff]+1;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff)continue;
        dfs1(v,u);
        if(md[v]>md[hson[u]])hson[u]=v;
    }
    if(hson[u])md[u]=md[hson[u]];len[u]=md[u]-dep[u]+1;
}
double tmp[MAX],*f[MAX],*id=tmp;
void dfs2(int u,int ff)
{
    if(hson[u])f[hson[u]]=f[u]+1,dfs2(hson[u],u);
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff||v==hson[u])continue;
        f[v]=id;id+=len[v];dfs2(v,u);
    }
}
void dfs(int u,int ff)
{
    V[u]+=V[ff];f[u][0]=V[u];
    if(hson[u])dfs(hson[u],u);
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff||v==hson[u])continue;
        dfs(v,u);
        for(int j=0;j<len[v];++j)
            if(len[u]>m-j-1&&m-j-1>=0)ans=min(ans,f[u][m-j-1]+f[v][j]-V[u]-V[ff]);
        for(int j=0;j<len[v];++j)
            f[u][j+1]=min(f[u][j+1],f[v][j]);
    }
    if(len[u]>m)ans=min(ans,f[u][m]-V[ff]);
}
int main()
{
    freopen("cdcq_b.in","r",stdin);
    freopen("cdcq_b.out","w",stdout);
    n=read();m=read();
    for(int i=1;i<=n;++i)a[i]=read();
    for(int i=1;i<=n;++i)b[i]=read();
    for(int i=1;i<n;++i)
    {
        int u=read(),v=read();
        Add(u,v);Add(v,u);
    }
    if(m==-1)
    {
        for(int i=1;i<=n;++i)ans=min(ans,1.0*a[i]/b[i]);
        printf("%.2lf\n",ans);
        return 0;
    }
    dfs1(1,0);f[1]=id;id+=md[1];dfs2(1,1);
    double l=0,r=1e9;m-=1;
    while(r-l>=1e-5)
    {
        double mid=(l+r)/2;ans=1e18;
        for(int i=1;i<=n;++i)V[i]=a[i]-mid*b[i],tmp[i]=1e18;
        dfs(1,0);
        if(ans<=0)r=mid;else l=mid;
    }
    if(l>=5e8)puts("-1");else printf("%.2lf\n",l);
    return 0;
}