Submission #101543

#	Time	Username	Problem	Language	Result	Execution time	Memory
101543		shenxy	Chase (CEOI17_chase)	C++11	20 / 100	502 ms	525312 KiB

chase

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
int n, v;
vector<long long int> p;
vector< vector<int> > adjlist;
vector< vector<int> > children;
vector< vector< vector<long long int> > > dptable;
long long int smalldp(int v, int pa, int crumbs) {
    if (dptable[v][pa][crumbs] != -1) return dptable[v][pa][crumbs];
    if (crumbs == 0) return 0;
    long long int ans = 0, maxdp = 0, maxnocrumb = 0;
    for (int i = 0; i < adjlist[v].size(); i++) {
        if (adjlist[v][i] == pa) continue;
        ans += p[adjlist[v][i]];
        maxdp = max(maxdp, smalldp(adjlist[v][i], v, crumbs - 1));
        maxnocrumb = max(maxnocrumb, smalldp(adjlist[v][i], v, crumbs));
    }
    return dptable[v][pa][crumbs] = max(maxdp + ans, maxnocrumb);
}
int main() {
    scanf("%d %d", &n, &v);
    long long int nextint;
    for (int i = 0; i < n; i++) {
        scanf("%lld", &nextint);
        p.push_back(nextint);
        adjlist.push_back(vector<int>());
    }
    int a, b;
    for (int i = 1; i < n; i++) {
        scanf("%d %d", &a, &b);
        adjlist[a - 1].push_back(b - 1);
        adjlist[b - 1].push_back(a - 1);
    }
    if (n <= 1000) {
        for (int i = 0; i < n; i++) {
            dptable.push_back(vector< vector<long long int> >(n + 1, vector<long long int>(v + 1, -1)));
        }
        long long int ans = 0;
        for (int i = 0; i < n; i++) {
            ans = max(ans, smalldp(i, n, v));
        }
        printf("%lld\n", ans);
        return 0;
    } else {
        printf("HIHIHIHIHIHIWRONGANSWER");
        return 0;
        for (int i = 0; i < n; i++) {
            dptable.push_back(vector< vector<long long int> >(v, {-1}));
            children.push_back(vector<int>());
        }
    }
}

Compilation message (stderr)

chase.cpp: In function 'long long int smalldp(int, int, int)':
chase.cpp:14:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < adjlist[v].size(); i++) {
                     ~~^~~~~~~~~~~~~~~~~~~
chase.cpp: In function 'int main()':
chase.cpp:23: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:26:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld", &nextint);
         ~~~~~^~~~~~~~~~~~~~~~~~
chase.cpp:32: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: 20 / 20
• Subtask 2: 0 / 20
• Subtask 3: 0 / 30
• Subtask 4: 0 / 30