oj.uz

Submission #101663

# Submission time Handle Problem Language Result Execution time Memory
101663 2019-03-19T07:51:18 Z rocketninja7 Chase (CEOI17_chase) C++14
0 / 100
 98 ms 10592 KB 
#include <cstdio>
#include <vector>
using namespace std;

vector<int> adjList[100001];

long long int p[100001];
bool vis[100001];

long long int dp(int bread, int node){
    if(bread==0){
        return 0;
    }
    vis[node]=true;
    long long int ans=0;
    long long int temp=0;
    for(int i=0;i<adjList[node].size();i++){
        temp+=p[adjList[node][i]];
    }
    for(int i=0;i<adjList[node].size();i++){
        if(!vis[adjList[node][i]]){
            ans=max(ans, max(dp(bread, adjList[node][i]), dp(bread-1, adjList[node][i])+temp));
        }
    }
    return ans;
}

int main(){
    int n, v;
    scanf("%d%d", &n, &v);
    for(int i=1;i<n+1;i++){
        scanf("%lld", &p[i]);
    }
    for(int i=0;i<n-1;i++){
        int a, b;
        scanf("%d%d", &a, &b);
        adjList[a].push_back(b);
        adjList[b].push_back(a);
    }
    long long int ans=0;
    for(int i=1;i<n+1;i++){
        for(int j=1;j<n+1;j++){
            vis[j]=false;
        }
        ans=max(ans, dp(v, i));
        if(n>1000){
            break;
        }
    }
    printf("%lld", ans);
    return 0;
}

Compilation message

chase.cpp: In function 'long long int dp(int, int)':
chase.cpp:17:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<adjList[node].size();i++){
                 ~^~~~~~~~~~~~~~~~~~~~~
chase.cpp:20:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<adjList[node].size();i++){
                 ~^~~~~~~~~~~~~~~~~~~~~
chase.cpp: In function 'int main()':
chase.cpp:30:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &n, &v);
     ~~~~~^~~~~~~~~~~~~~~~
chase.cpp:32:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld", &p[i]);
         ~~~~~^~~~~~~~~~~~~~~
chase.cpp:36:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d", &a, &b);
         ~~~~~^~~~~~~~~~~~~~~~

Subtask #1
0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 2688 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 2688 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 30.0

# Verdict Execution time Memory Grader output
1 Incorrect 98 ms 10592 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 2688 KB Output isn't correct
2 Halted 0 ms 0 KB -