Submission #101398

#	Time	Username	Problem	Language	Result	Execution time	Memory
101398		dwsc	Chase (CEOI17_chase)	C++14	0 / 100	140 ms	9436 KiB

chase

#include <bits/stdc++.h>
#define int long long
using namespace std;
int n,v;
int p[100010];
vector<int> adj[100010];
int dp(int x,int dist,int pa){
   // cout << x << " " << dist << " " << pa << '\n';
    int ans = 0;
    int sum = 0;
    for (int i = 0; i < adj[x].size(); i++){
        int nx = adj[x][i];
        if (nx == pa) continue;
        sum += p[nx];
    }
   // cout << sum << "\n";
    if (dist == 0) return sum;
    for (int i = 0; i < adj[x].size(); i++){
        int nx = adj[x][i];
        if (nx== pa) continue;
        ans = max(ans,dp(nx,dist-1,x)+sum-p[nx]);
    }
    return ans;
}

main(){
    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].push_back(b);
        adj[b].push_back(a);
    }
    if (n <= 1000){
        int ans = 0;
        for (int i = 0; i < n; i++){
          //  cout << dp(i,v,-1) << "\n";
            ans = max(ans,dp(i,v,-1));
        }
        cout << ans;
    }
    else cout << dp(0,v,-1);
}

Compilation message (stderr)

chase.cpp: In function 'long long int dp(long long int, long long int, long long int)':
chase.cpp:11:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < adj[x].size(); i++){
                     ~~^~~~~~~~~~~~~~~
chase.cpp:18:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < adj[x].size(); i++){
                     ~~^~~~~~~~~~~~~~~
chase.cpp: At global scope:
chase.cpp:26:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main(){
      ^

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