loj#2009.「SCOI2015」小凸玩密室

题目链接

loj#2009. 「SCOI2015」小凸玩密室

题解

树高不会很高<=20

点亮灯泡x,点亮x的一个子树，再点亮x另外的子树，

然后回到x的父节点，点亮父节点之后再点亮父节点的其他子树

所以对于一个节点x，有这样两种情况

x还没有被点亮，那么下一个被点亮的是x的一个儿子

x是叶子节点，那么下一个被点亮的是它的祖先，或者是它祖先的儿子

设f[i][j]表示点亮i之后回到i的第j个祖先的最小花费

设g[i][j]表示点亮i之后回到i的第j个祖先的另一个儿子的最小花费

然后从下到上，由儿子的状态转移到父亲的状态

讨论一下儿子个数

最后模拟点亮过程统计答案

ans要开大

代码

#include<cstdio>
#include<algorithm>
using namespace std;
inline int read() {
    int x = 0,f = 1;
    char c = getchar();
    while(c < '0' || c > '9')c = getchar();
    while(c <= '9' && c >= '0') x = x * 10 + c - '0',c = getchar();
    return x * f ;
}
#define INF 2000000000000000000ll
#define LL long long
#define fa(i,j) ((1 << j - 1) <= i ? (i >> j) : -1 )
#define bro(i,j) ((i >> j - 1) ^ 1)
#define ls (i << 1)
#define rs (i << 1 | 1)
const int maxn = 200007; 

int n;
LL a[maxn],dp[maxn][27][2],dis[maxn][27];

int main() {
   	n = read();
    for(int i = 1;i <= n;++ i) a[i] = read();
    dis[1][1] = 0;
    for(int i = 2;i <= n;++ i) {
        dis[i][1] = read();
        for(int j = 2;~fa(i,j);++ j)  dis[i][j] = dis[fa(i,1)][j - 1] + dis[i][1];
    }
    for(int i = n;i >= 1;-- i) { for(int j = 1;~ fa(i,j);++ j) {
        dp[i][j][0] = dp[i][j][1] = INF;
        if(ls > n)
            dp[i][j][0] = dis[i][j] * a[fa(i,j)], dp[i][j][1] = (dis[i][j] + dis[bro(i,j)][1]) * a[bro(i,j)];
    	else if(rs > n)
            dp[i][j][0] = dp[ls][j + 1][0] + dis[ls][1] * a[ls], dp[i][j][1] = dp[ls][j + 1][1] + dis[ls][1] * a[ls];
        else
            dp[i][j][0] = min(dp[i][j][0],dp[ls][1][1] + dp[rs][j + 1][0] + dis[ls][1] * a[ls]),
            dp[i][j][0] = min(dp[i][j][0],dp[rs][1][1] + dp[ls][j + 1][0] + dis[rs][1] * a[rs]),
        	dp[i][j][1] = min(dp[i][j][1],dp[ls][1][1] + dp[rs][j + 1][1] + dis[ls][1] * a[ls]),
            dp[i][j][1] = min(dp[i][j][1],dp[rs][1][1] + dp[ls][j + 1][1] + dis[rs][1] * a[rs]);
        }
    }
    LL ans = INF;
    for(int x = 1;x <= n;x ++) {
        LL tmp = dp[x][1][0];
        for(int i = fa(x,1),last = x;~i;i = fa(i,1),last = fa(last,1)) {
            if(bro(last,1) <= n)  tmp += dis[bro(last,1)][1] * a[bro(last,1)] + dp[bro(last,1)][2][0];
            else tmp +=  dis[i][1] * a[fa(i,1)];
        }
        ans = min(ans,tmp);
    }
    printf("%lld\n",ans);
    return 0;
}