oj.uz

Submission #851093

# Submission time Handle Problem Language Result Execution time Memory
851093 2023-09-18T13:51:40 Z Itamar Paths (RMI21_paths) C++14
36 / 100
 600 ms 856 KB 
// greedy cat.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <iostream>
using namespace std;
#include <vector>
#define ll long long
#define pll pair<ll,ll>
#define vll vector<ll>
#include <queue>
const int siz = 2000;
vector<pll> fr[siz];
ll ans[siz];
int N,K;
#include <set>
pll dfs(int i, vector<pll>&pad,set<pll>&s) {
    pll ansi={0,i};
    for (pll j: fr[i]) {
        if (j.first != pad[i].first) {
            pad[j.first] = { i,j.second };
            pll k =dfs(j.first, pad, s);
            s.insert({ -k.first - j.second,k.second });
            ansi = max(ansi, { k.first + j.second,k.second });

        }
    }
    return ansi;
}

void calc(int r) {
    set<pll> s;
    vector<pll> pad(N,{-1,-1});
    dfs(r,pad,s);
    vector<bool> vis(N);
    vis[r] = 1;
    for (int i = 0; i < K; i++) {
        if (s.empty())break;
        pll p = *s.begin();
        ans[r] -= p.first;
        ll cur = 0;
        int v = p.second;
        while (!vis[v]) {
            cur += pad[v].second;
            s.erase({ -cur, p.second });
            vis[v] = 1;
            v = pad[v].first;
        }
    }
}

int main()
{
    int n,k;
    cin >> n >> k;
    N = n, K =k;
    for (int i = 0; i < n - 1; i++) {
        int x, y, c;
        cin >> x >> y >> c;
        x--, y--;
        fr[x].push_back({ y,c });
        fr[y].push_back({ x,c });
    }
    for (int i = 0; i < n; i++) {
        calc(i);
    }
    for (int i = 0; i < n; i++)cout << ans[i] << "\n";
}

// Run program: Ctrl + F5 or Debug > Start Without Debugging menu
// Debug program: F5 or Debug > Start Debugging menu

// Tips for Getting Started: 
//   1. Use the Solution Explorer window to add/manage files
//   2. Use the Team Explorer window to connect to source control
//   3. Use the Output window to see build output and other messages
//   4. Use the Error List window to view errors
//   5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project
//   6. In the future, to open this project again, go to File > Open > Project and select the .sln file

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 6 ms 344 KB Output is correct
4 Correct 9 ms 348 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Correct 8 ms 348 KB Output is correct
7 Correct 7 ms 344 KB Output is correct

Subtask #3
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 6 ms 344 KB Output is correct
4 Correct 9 ms 348 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Correct 8 ms 348 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 177 ms 600 KB Output is correct
9 Correct 166 ms 696 KB Output is correct
10 Correct 211 ms 636 KB Output is correct
11 Correct 156 ms 600 KB Output is correct
12 Correct 176 ms 856 KB Output is correct

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 6 ms 344 KB Output is correct
4 Correct 9 ms 348 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Correct 8 ms 348 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 177 ms 600 KB Output is correct
9 Correct 166 ms 696 KB Output is correct
10 Correct 211 ms 636 KB Output is correct
11 Correct 156 ms 600 KB Output is correct
12 Correct 176 ms 856 KB Output is correct
13 Execution timed out 871 ms 728 KB Time limit exceeded
14 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 600 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #6
0 / 32.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 6 ms 344 KB Output is correct
4 Correct 9 ms 348 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Correct 8 ms 348 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 177 ms 600 KB Output is correct
9 Correct 166 ms 696 KB Output is correct
10 Correct 211 ms 636 KB Output is correct
11 Correct 156 ms 600 KB Output is correct
12 Correct 176 ms 856 KB Output is correct
13 Execution timed out 871 ms 728 KB Time limit exceeded
14 Halted 0 ms 0 KB -