【秋招笔试题】矿脉开采（树形DP）

此题直接按树形dp做即可，每次从0枚举到k转移状态

#include <iostream>
#include <cstring>
#include <algorithm>


using namespace std;
#define endl '\n'
#define lson node << 1
#define rson node << 1 | 1
const int maxn = 1e5 + 5;
const int maxk = 100;
int dp[maxn][maxk],n,k,res;
int a[maxn];
int check(int node){
    if (node > n || a[node] == -1) return false;
    return true;
}
void dfs(int node){
    dp[node][1] = a[node];
    int maxx = 0;
    if (check(lson)){
        dfs(lson);
        maxx = max(maxx,dp[lson][0]);
        for (int i = 1; i <= k; ++i) {
            maxx = max(maxx,dp[lson][i]);
            dp[node][i] = max(dp[node][i],dp[lson][i - 1] + a[i]);
        }
    }
    if (check(rson)){
        dfs(rson);
        maxx = max(maxx,dp[rson][1]);
        for (int i = 1; i <= k; ++i) {
            maxx = max(maxx,dp[rson][i]);
            dp[node][i] = max(dp[node][i],dp[rson][i - 1] + a[i]);
        }
    }
    dp[node][0] = maxx;
    for (int i = 0; i <= k; ++i) {
        for (int j = 1; i + j + 1<= k; ++j) {
            dp[node][i + j + 1] = max(dp[node][i + j + 1],dp[lson][i] + dp[rson][j] + a[node]);
        }
    }
}

void solve(){
    cin >> k >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
    }
    dfs(1);
    for (int i = 0; i <= k; ++i) {
        res = max(res,dp[1][i]);
    }
    cout << res << endl;
}
signed main(){
    std::ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    solve();

}