題解

神仙dp啊QAQ

我們發現我們需要枚舉一個起點，遍歷完它所有的兒子然後向上爬

設\(f[i][j]\)表示第i個點的子樹全部處理完之後到達i深度為j的祖先的兄弟處

我們只需要對葉子節點和只有一個兒子的點特殊討論，因為所有的向上爬都是從葉子爬的

轉移的時候只要枚舉從兩個兒子裏哪個爬上取就好了

設\(g[i][j]\)表示第i個點的子樹全部處理完之後到達i深度為j的祖先處

轉移個f數組類似，但是要用到f數組

枚舉每個點當起點的時候，遵循x的父親->x的兄弟->x的父親的父親->x的父親的兄弟這樣順序來覆蓋即可

代碼

#include <bits/stdc++.h>
 

//#define ivorysi
#define enter putchar(‘\n‘)
#define space putchar(‘ ‘)
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define eps 1e-8
#define mo 974711
#define MAXN 200005
#define pii pair<int,int>
using namespace std;
typedef long long int64;
typedef double db;
template
 
<class T>
void read(T &res) {
    res = 0;char c = getchar();T f = 1;
    while(c < ‘0‘ || c > ‘9‘) {
    if(c == ‘-‘) f = -1;
    c = getchar();
    }
    while(c >= ‘0‘ && c <= ‘9‘) {
    res = res * 10 + c - ‘0‘;
    c = getchar();
    }
    res *= f;
}
template<class T>
void
 
 out(T x) {
    if(x < 0) {putchar(‘-‘);x = -x;}
    if(x >= 10) {
    out(x / 10);
    }
    putchar(‘0‘ + x % 10);
}
int N,dep[MAXN];
int64 a[MAXN],b[MAXN],dis[MAXN],f[MAXN][18],g[MAXN][18];
void Solve() {
    read(N);
    for(int i = 1 ; i <= N ; ++i) read(a[i]);
    for(int i = 2 ; i <= N ; ++i) read(b[i]);
    for(int i = 1 ; i <= N ; ++i) {
    dep[i] = dep[i >> 1] + 1;
    dis[i] = dis[i >> 1] + b[i];
    }
    for(int u = N ; u >= 1 ; --u) {
    if((u << 1) > N) {
        for(int j = dep[u]; j >= 2 ; --j) {
        int fa = u >> (dep[u] - j + 1),x = (u >> (dep[u] - j)) ^ 1;
        f[u][j] = (dis[u] + dis[x] - 2 * dis[fa]) * a[x];
        }
    }
    else if((u << 1) == N) {
        for(int j = dep[u] ; j >= 2 ; --j) {
        f[u][j] = f[u << 1][j] + a[u << 1] * b[u << 1];
        }
    }
    else {
        int lc = u << 1,rc = u << 1 | 1;
        for(int j = dep[u] ; j >= 2 ; --j) {
        f[u][j] = min(a[lc] * b[lc] + f[lc][dep[lc]] + f[rc][j],a[rc] * b[rc] + f[rc][dep[rc]] + f[lc][j]);
        }
    }
    }
    for(int u = N ; u >= 1 ; --u) {
    if((u << 1) > N) {
        for(int j = dep[u] - 1; j >= 0 ; --j) {
        int fa = u >> (dep[u] - j);
        g[u][j] = (dis[u] - dis[fa]) * a[fa];
        }
    }
    else if((u << 1) == N) {
        for(int j = dep[u] - 1 ; j >= 0 ; --j) {
        g[u][j] = g[u << 1][j] + a[u << 1] * b[u << 1];
        }
    }
    else {
        int lc = u << 1,rc = u << 1 | 1;
        for(int j = dep[u] - 1 ; j >= 0 ; --j) {
        g[u][j] = min(a[lc] * b[lc] + f[lc][dep[lc]] + g[rc][j],a[rc] * b[rc] + f[rc][dep[rc]] + g[lc][j]);
        }
    }
    }
    int64 ans = 1e18;
    for(int u = N ; u >= 1 ; --u) {
    int64 res = g[u][dep[u] - 1];
    for(int x = u ; x > 1 ; x >>= 1) {
        int t = x ^ 1;
        if(t > N) res += a[x >> 2] * b[x >> 1];
        else res += a[t] * b[t] + g[t][dep[t] - 2];
    }
    ans = min(ans,res);
    }
    out(ans);enter;
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Solve();
    return 0;
}

【LOJ】 #2009. 「SCOI2015」小凸玩密室