Submission #961650

#	Time	Username	Problem	Language	Result	Execution time	Memory
961650		LucaIlie	Chase (CEOI17_chase)	C++17	0 / 100	317 ms	252528 KiB

chase

#include <bits/stdc++.h>

using namespace std;

const int MAX_N = 1e5;
const int MAX_K = 100;
const long long INF = 1e15;
int k;
int a[MAX_N + 1];
long long b[MAX_N + 1];
long long dp[MAX_N + 1][MAX_K + 1][3];
vector<int> adj[MAX_N + 1];

long long maxCost = 0;
void dfs( int u, int p ) {
    for ( int v: adj[u] ) {
        if ( v == p )
            continue;
        dfs( v, u );
    }

    for ( int x = 0; x <= k; x++ )
        dp[u][x][0] = b[u];
    for ( int v: adj[u] ) {
        if ( v == p )
            continue;

        for ( int x = 0; x <= k; x++ ) {
            ///0 0
            maxCost = max( maxCost, dp[u][x][0] + dp[v][k - x][0] - a[u] - a[v] );
            ///0 1
            maxCost = max( maxCost, dp[u][x][0] + dp[v][k - x][1] - a[u] );
            ///0 2
            maxCost = max( maxCost, dp[u][x][0] + dp[v][k - x][2] );
            ///1 0
            maxCost = max( maxCost, dp[u][x][1] + dp[v][k - x][0] - a[u] );
            ///2 0
            maxCost = max( maxCost, dp[u][x][2] + dp[v][k - x][0] );
            ///1 1
            maxCost = max( maxCost, dp[u][x][1] + dp[v][k - x][1] );
            ///2 1
            maxCost = max( maxCost, dp[u][x][2] + dp[v][k - x][1] );
            ///1 2
            maxCost = max( maxCost, dp[u][x][1] + dp[v][k - x][2] );
            ///2 2
            maxCost = max( maxCost, dp[u][x][2] + dp[v][k - x][2] );
        }

        for ( int x = 1; x <= k; x++ ) {
            dp[u][x][0] = max( dp[u][x][0], max( max( dp[v][x - 1][0] + b[u] - a[u] - a[v], dp[v][x - 1][1] + b[u] - a[v] ), dp[v][x - 1][2] + b[u] ) );
            dp[u][x][1] = max( dp[u][x][1], max( max( dp[v][x][0], dp[v][x][1] ), dp[v][x][2] ) );
            dp[u][x][2] = max( dp[u][x][2], max( dp[v][x][1], dp[v][x][2] ) );
        }
    }

    /*printf( "nod %d\n", u );
    for ( int x = 0; x <= k; x++ )
        printf( "%d %d %d\n", dp[u][x][0], dp[u][x][1], dp[u][x][2] );
    printf( "\n" );*/
}

int main() {
    int n;

    cin >> n >> k;
    for ( int v = 1; v <= n; v++ )
        cin >> a[v];
    for ( int i = 0; i < n - 1; i++ ) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back( v );
        adj[v].push_back( u );
        b[u] += a[v];
        b[v] += a[u];
    }

    dfs( 1, 0 );

    cout << maxCost << "\n";

    return 0;
}

• Subtask 1: 0 / 20
• Subtask 2: 0 / 20
• Subtask 3: 0 / 30
• Subtask 4: 0 / 30