Submission #1045566

#	Time	Username	Problem	Language	Result	Execution time	Memory
1045566		phoenix0423	Chase (CEOI17_chase)	C++17	0 / 100	452 ms	9636 KiB

chase

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
#define fastio ios::sync_with_stdio(false), cin.tie(0)
// #pragma GCC optimize("Ofast")
#define pb push_back
#define eb emplace_back
#define f first
#define s second
#define int long long
#define lowbit(x) x&-x
const int maxn = 1e5 + 5;
const int INF = 4e18;
const int N = 998244353;
vector<int> adj[maxn];
int p[maxn], deg[maxn], dep[maxn], dp[maxn];
int n, v, ans = 0;

void dfs(int pos, int prev){
    dp[pos] += deg[pos];
    ans = max(ans, dp[pos]);
    for(auto x : adj[pos]){
        if(x == prev) continue;
        dep[x] = dep[pos] + 1;
        if(dep[x] <= v){
            dp[x] = dp[pos] - p[pos];
            dfs(x, pos);
        }
    }
}

signed main(void){
    fastio;
    cin>>n>>v;
    for(int i = 0; i < n; i++) cin>>p[i];
    for(int i = 0; i < n - 1; i++){
        int a, b;
        cin>>a>>b;
        a--, b--;
        adj[a].pb(b), adj[b].pb(a);
        deg[a] += p[b], deg[b] += p[a];
    }
    if(!v){
        cout<<0<<"\n";
        return 0;
    }
    for(int i = 0; i < n; i++){
        dp[i] = 0;
        dep[i] = 1;
        dfs(i, -1);
    }
    cout<<ans<<"\n";
}

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