loj2009. 「SCOI2015」小凸玩密室

「SCOI2015」小凸玩密室

小凸和小方相约玩密室逃脱，这个密室是一棵有 $ n $ 个节点的完全二叉树，每个节点有一个灯泡。点亮所有灯泡即可逃出密室。每个灯泡有个权值 $ A_i $，每条边也有个权值 $ b_i $。

点亮第 $ 1 $ 个灯泡不需要花费，之后每点亮一个新的灯泡 $ V $ 的花费，等于上一个被点亮的灯泡 $ U $ 到这个点 $ V $ 的距离 $ D(u, v) $，乘以这个点的权值 $ A_v $。

在点灯的过程中，要保证任意时刻所有被点亮的灯泡必须连通，在点亮一个灯泡后必须先点亮其子树所有灯泡才能点亮其他灯泡。请告诉他们，逃出密室的最少花费是多少。

---

Sol

f[i][j]表示点完以i为根的子树，下一个点他的深度为j的祖先的另一个儿子所需要的最小代价。

g[i][j]表示点完以i为根的子树，下一个点他深度为j的祖先。

转移时分讨他没有孩子，只有一个孩子，有两个孩子三种情况。

这样设计状态是nlogn的

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 200005
#define ll long long
#define ls (x<<1)
#define rs (x<<1|1)
using namespace std;
int n,a[maxn],deep[maxn],w[maxn];
ll f[maxn][],g[maxn][],sw[maxn],ans=1e16;
int main()
{
    cin>>n;
    for(int i=;i<=n;i++)scanf("%d",&a[i]);
    for(int i=;i<=n;i++)scanf("%d",&w[i]);
    for(int i=;i<=n;i++)sw[i]=sw[i/]+w[i];
    for(int i=;i<=n;i++)deep[i]=deep[i/]+;
    for(int x=n;x>=;x--){
        int y=x^,fa=x/;
        for(int i=deep[x]-;i>=;i--){
            if(ls>n){
                f[x][i]=(sw[x]+sw[y]-*sw[fa])*a[y];
                g[x][i]=(sw[x]-sw[fa])*a[fa];
            }
            else {
                if(rs>n){
                    f[x][i]=f[ls][i]+w[ls]*a[ls];
                    g[x][i]=g[ls][i]+w[ls]*a[ls];
                }
                else {
                    f[x][i]=min(
                    f[ls][deep[x]]+w[ls]*a[ls]+f[rs][i],
                    f[rs][deep[x]]+w[rs]*a[rs]+f[ls][i]
                    );
                    g[x][i]=min(
                    f[ls][deep[x]]+w[ls]*a[ls]+g[rs][i],
                    f[rs][deep[x]]+w[rs]*a[rs]+g[ls][i]
                    );
                }
            }
            y=(y/)^;fa=fa/;
        }
    }

    for(int x=;x<=n;x++){
        int y=x;ll sum=g[y][deep[y]-];
        while(deep[y]>){
            if((y^)<=n)sum+=w[y^]*a[y^]+g[y^][deep[y]-];
            else sum+=w[y/]*a[y/];
            y/=;
        }
        ans=min(ans,sum);
    }
    cout<<ans<<endl;
    return ;
}