QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#646328#8938. Crawling on a Treeucup-team173RE 1ms3800kbC++202.0kb2024-10-16 22:15:502024-10-16 22:15:51

Judging History

This is the latest submission verdict.

  • [2024-10-16 22:15:51]
  • Judged
  • Verdict: RE
  • Time: 1ms
  • Memory: 3800kb
  • [2024-10-16 22:15:50]
  • Submitted

answer

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

#define int ll
using ll = long long;
constexpr int inf = 1e18;

signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    int n, m;
    cin >> n >> m;
    vector G(n + 1, vector<array<int, 3>>());
    for(int i = 1; i < n; i++) {
        int u, v, l, k;
        cin >> u >> v >> l >> k;
        G[u].push_back({v, l, k});
        G[v].push_back({u, l, k});
    }
    vector<int> c(n + 1), ek(n + 1), el(n + 1), fa(n + 1);
    for(int i = 2; i <= n; i++) {
        cin >> c[i];
    }
    vector f(n + 1, vector(m + 1, 0ll));
    auto merge = [&](vector<int> f, vector<int> g, int l, int r) -> vector<int> {
        vector<int> res(m + 1, inf);
        int i = 0, j = l;
        while(i < m && f[i] == inf) i++;
        while(j < r && g[j] == inf) j++;
        while(i + j <= m) {
            res[i + j] = f[i] + g[j];
            if(j < r && f[i] + g[j + 1] < f[i + 1] + g[j]) j++;
            else i++;
        }
        return res;
    };
    auto dfs = [&](auto self, int x, int fz) -> void {
        fa[x] = fz;
        for(auto [y, l, k] : G[x]) if(y != fz) {
            ek[y] = k, el[y] = l;
            self(self, y, x);
            c[x] = max(c[x], c[y]);
            for(int i = 1; i <= m; i++) {
                f[y][i] += (2 * max(c[y], i) - i) * el[y];
            }
            if(ek[y] < c[y]) {
                f[x] = vector(m + 1, inf);
            } else {
                f[x] = merge(f[x], f[y], 2 * c[y] - ek[y], ek[y]);
            }
        }
    };
    dfs(dfs, 1, 0);
    // for(int i = 1; i <= n; i++) {
    //     cerr << i << ": ";
    //     for(int j = 0; j <= m; j++) {
    //         cerr << f[i][j] << " \n"[j == m];
    //     }
    // }
    vector<int> ans(m + 1, inf);
    for(int i = 0; i <= m; i++) {
        ans[max(c[1], i)] = min(ans[max(c[1], i)], f[1][i]);
    }
    for(int i = 1; i <= m; i++) {
        if(ans[i] == inf) ans[i] = -1;
        cout << ans[i] << '\n';
    }
    return 0;
}

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 3608kb

input:

4 2
1 2 3 2
2 3 2 1
2 4 5 1
1 1 1

output:

-1
13

result:

ok 2 number(s): "-1 13"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3484kb

input:

4 2
1 2 3 2
2 3 2 1
2 4 5 1
2 2 2

output:

-1
-1

result:

ok 2 number(s): "-1 -1"

Test #3:

score: 0
Accepted
time: 1ms
memory: 3548kb

input:

2 1
2 1 1 1
1

output:

1

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3576kb

input:

2 50
2 1 1 1
50

output:

-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1

result:

ok 50 numbers

Test #5:

score: 0
Accepted
time: 0ms
memory: 3800kb

input:

2 50
2 1 1 50
50

output:

-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
-1
50

result:

ok 50 numbers

Test #6:

score: -100
Runtime Error

input:

2 50
1 2 1 100000
50

output:


result: