Submission #161833

#	Time	Username	Problem	Language	Result	Execution time	Memory
161833		Akashi	Chase (CEOI17_chase)	C++14	0 / 100	973 ms	524292 KiB

chase

#include <bits/stdc++.h>
using namespace std;

int n, V;
int val[100005];
bool viz[100005];
long long Sol, sum[100005];
long long down[100005][105][2], up[100005][105][2];
long long up2[100005][105][2], down2[100005][105][2];
long long up3[100005][105][2];

vector <int> v[100005];

void dfs(int nod = 1){
    viz[nod] = 1;
    for(auto fiu : v[nod]){
        if(viz[fiu]) continue ;

        dfs(fiu);

//        if(nod == 7 && fiu == 6){
//            bool ok = 0;
//
//        }

        for(int k = 1; k <= V ; ++k){
            down[nod][k][0] = max(max(down[nod][k - 1][0], down[nod][k][0]), max(down[fiu][k][0], down[fiu][k][1] - val[nod]));
            down[nod][k][1] = max(max(down[nod][k - 1][1], down[nod][k][1]), max(down[fiu][k - 1][0], down[fiu][k - 1][1] - val[nod]) + sum[nod]);

            up[nod][k][0] = max(max(up[nod][k][0], up[nod][k - 1][0]), max(up[fiu][k][0], up[fiu][k][1]));
            up[nod][k][1] = max(max(up[nod][k][1], up[nod][k - 1][1]), max(up[fiu][k - 1][0] - val[fiu], up[fiu][k - 1][1] - val[fiu]) + sum[nod]);

            Sol = max(Sol, max(up[nod][k][0], up[nod][k][1]));
            Sol = max(Sol, max(down[nod][k][0], down[nod][k][1]));

            Sol = max(Sol,
                      max (
                            max(up[fiu][k][0] + down2[nod][V - k][0], up[fiu][k][1] + down2[nod][V - k][0]),
                            max(up[fiu][k][0] + down2[nod][V - k][1] - val[nod], up[fiu][k][1] + down2[nod][V - k][1] - val[nod])
                          )
                      );

            Sol = max(Sol,
                      max (
                            max(down[fiu][k][0] + up2[nod][V - k][0], down[fiu][k][1] + up2[nod][V - k][0] - val[nod]),
                            max(down[fiu][k][0] + up2[nod][V - k][1], down[fiu][k][1] + up2[nod][V - k][1] - val[nod])
                          )
                      );

            Sol = max(Sol,
                        max(
                            max(up[fiu][k - 1][0] + down2[nod][V - k][0] - val[fiu], up[fiu][k - 1][1] + down2[nod][V - k][0] - val[fiu]),
                            max(up[fiu][k - 1][0] + down2[nod][V - k][1] - val[fiu] - val[nod], up[fiu][k - 1][1] + down2[nod][V - k][1] - val[fiu] - val[nod])
                        ) + sum[nod]
                      );

            Sol = max(Sol,
                    max(
                        max(down[fiu][k - 1][0] + up3[nod][V - k][0], down[fiu][k - 1][1] + up3[nod][V - k][0] - val[nod]),
                        max(down[fiu][k - 1][0] + up3[nod][V - k][1] - val[nod], down[fiu][k - 1][1] + up2[nod][V - k][1] - val[nod])
                    ) + sum[nod]
                  );
        }


        for(int k = 1; k <= V ; ++k){
            down2[nod][k][0] = max(max(down2[nod][k - 1][0], down2[nod][k][0]), down[fiu][k][0]);
            down2[nod][k][1] = max(max(down2[nod][k - 1][1], down2[nod][k][1]), down[fiu][k][1]);

            up2[nod][k][0] = max(max(up2[nod][k][0], up2[nod][k - 1][0]), up[fiu][k][0]);
            up2[nod][k][1] = max(max(up2[nod][k][1], up2[nod][k - 1][1]), up[fiu][k][1]);

            up3[nod][k][0] = max(max(up3[nod][k][0], up3[nod][k - 1][0]), up[fiu][k][1] - val[fiu]);
            up3[nod][k][1] = max(max(up3[nod][k][1], up3[nod][k - 1][1]), up[fiu][k][1] - val[fiu]);
        }
    }

    bool ok = 0;
}

int main()
{
//    freopen("1.in", "r", stdin);

    scanf("%d%d", &n, &V);
    for(int i = 1; i <= n ; ++i)
        scanf("%d", &val[i]);

    int x, y;
    for(int i = 2; i <= n ; ++i){
        scanf("%d%d", &x, &y);

        v[x].push_back(y);
        v[y].push_back(x);

        sum[x] += val[y];
        sum[y] += val[x];
    }

    dfs();

    printf("%lld", Sol);

    return 0;
}

Compilation message (stderr)

chase.cpp: In function 'void dfs(int)':
chase.cpp:78:10: warning: unused variable 'ok' [-Wunused-variable]
     bool ok = 0;
          ^~
chase.cpp: In function 'int main()':
chase.cpp:85: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:87:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &val[i]);
         ~~~~~^~~~~~~~~~~~~~~
chase.cpp:91:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d", &x, &y);
         ~~~~~^~~~~~~~~~~~~~~~

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