oj.uz

Submission #894956

# Submission time Handle Problem Language Result Execution time Memory
894956 2023-12-29T09:47:10 Z Pragmatism Paths (RMI21_paths) C++17
100 / 100
 220 ms 23376 KB 
//Bismillahir-Rahmanir-Rahim

#include <bits/stdc++.h>
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("O1,O2,O3,Ofast,unroll-loops")
//#pragma GCC target("sse,sse2,sse3,sse4,sse4.1,sse4.2,popcnt,avx,avx2")

#define pb push_back
#define pii pair <int, int>
#define pll pair <long long, long long>
#define pld pair <long double, long double>
#define ll long long
#define ld long double
#define x first
#define y second
#define all(v) v.begin(),v.end()
#define sz(s) (int)s.size()
#define skip continue
#define bpop(x) (ll)__builtin_popcountll(x)

using namespace std;

const int N = 1e5 + 7;
const int M = 62;
const int maxA = 1e7 + 7;
const int inf = 1e9 + 7;
const ll INF = 1e18 + 7;
const ll MOD = 998244353;
const ld eps = 1e-9;

pii dir[] = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};

//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

#define int long long

pii mx[N];
vector <pii> g[N];
int n, k, T, ans[N], dist[N];
set <pii> best, extra;
int sum_best = 0, sum_extra = 0;
void add(pii val) {
    //cout << "add: " << val.x << ' ' << val.y << '\n';
    best.insert(val), sum_best += val.x;
    if (sz(best) > k) {
        pii x = *best.begin();
        best.erase(x), extra.insert(x);
        sum_best -= x.x, sum_extra += x.x;
    }
}
void del(pii val) {
    //cout << "del: " << val.x << ' ' << val.y << '\n';
    if (best.count(val)) {
        best.erase(val), sum_best -= val.x;
        pii x = *extra.rbegin();
        extra.erase(x), sum_extra -= x.x;
        //cout << "del + ";
        add(x);
    }
    else {
        extra.erase(val);
        sum_extra -= val.x;
    }
}
void build(int v, int pr) {
    mx[v] = {dist[v], v};
    for (auto p : g[v]) {
        int to = p.x, w = p.y;
        if (to == pr)skip;
        dist[to] = dist[v] + w, build(to, v);
        mx[v] = max(mx[v], mx[to]);
    }
}
void build2(int v, int pr) {
    if (pr == 0 || mx[v].y != mx[pr].y)add({mx[v].x - dist[pr], mx[v].y});
    for (auto p : g[v]) {
        int to = p.x, w = p.y;
        if (to == pr)skip;
        build2(to, v);
    }
}
pii mxup[N];
void dfs(int v, int pr, int sub) {
    ans[v] = sum_best;
    pii mx1 = {0, 0}, mx2 = {0, 0};
    for (auto p : g[v]) {
        int to = p.x, w = p.y;
        if (to == pr)skip;
        if (mx[to].x - sub > mx1.x)mx2 = mx1, mx1 = {mx[to].x - sub, mx[to].y};
        else mx2 = max(mx2, {mx[to].x - sub, mx[to].y});
    }
    //cout << v << ":\n";
    //for (auto p : extra)cout << p.x << ' ' << p.y << '\n';
    //cout << '\n';
    //for (auto p : best)cout << p.x << ' ' << p.y << '\n';
    //cout << '\n';
    for (auto p : g[v]) {
        int to = p.x, w = p.y;
        if (to == pr)skip;
        //cout << v << "<->" << to << ":\n";
        pii val = max(((pii){mx[to].x - sub, mx[to].y} == mx1 ? mx2 : mx1), mxup[v]);
        del(val), add({val.x + w, val.y}), del({mx[to].x - sub, mx[to].y}), add({mx[to].x - w - sub, mx[to].y});
        mxup[to] = pii{val.x + w, val.y};
        dfs(to, v, sub + w);
        del({mx[to].x - w - sub, mx[to].y}), add({mx[to].x - sub, mx[to].y}), del({val.x + w, val.y}), add(val);
    }
}
void solve() {
    cin >> n >> k;
    int total = 0, cnt_leaf = 0;
    for (int i = 1;i < n;i++) {
        int x, y, w;
        cin >> x >> y >> w;
        g[x].pb({y, w}), g[y].pb({x, w});
        total += w;
    }
    for (int i = 1;i <= n;i++)cnt_leaf += (sz(g[i]) == 1);
    build(1, 0), build2(1, 0);
    if (sz(g[1]) == 1)add({0, 1});
    //k = min(k, sz(best) + sz(extra));
    //cout << "GAY:\n";
    //for (auto p : extra)cout << p.x << ' ' << p.y << '\n';
    //for (auto p : best)cout << p.x << ' ' << p.y << '\n';
    //cout << ":GAY\n";
    dfs(1, 0, 0);
    for (int i = 1;i <= n;i++)cout << ans[i] << '\n';
}
signed main() {
    //srand(time(NULL));
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    //freopen("tests.in", "r", stdin);
    //freopen("milkorder.out", "w", stdout);
    int test = 1;
    //cin >> test;
    for (int t = 1;t <= test;t++) {
        //cout << "Case " << t << ": ";
        solve();
    }
    return 0;
}

Compilation message

Main.cpp: In function 'void build2(long long int, long long int)':
Main.cpp:77:23: warning: unused variable 'w' [-Wunused-variable]
   77 |         int to = p.x, w = p.y;
      |                       ^
Main.cpp: In function 'void dfs(long long int, long long int, long long int)':
Main.cpp:87:23: warning: unused variable 'w' [-Wunused-variable]
   87 |         int to = p.x, w = p.y;
      |                       ^

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 6488 KB Output is correct
2 Correct 1 ms 6488 KB Output is correct

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 6488 KB Output is correct
2 Correct 1 ms 6488 KB Output is correct
3 Correct 1 ms 6492 KB Output is correct
4 Correct 2 ms 6744 KB Output is correct
5 Correct 1 ms 6492 KB Output is correct
6 Correct 2 ms 6492 KB Output is correct
7 Correct 2 ms 6492 KB Output is correct

Subtask #3
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 6488 KB Output is correct
2 Correct 1 ms 6488 KB Output is correct
3 Correct 1 ms 6492 KB Output is correct
4 Correct 2 ms 6744 KB Output is correct
5 Correct 1 ms 6492 KB Output is correct
6 Correct 2 ms 6492 KB Output is correct
7 Correct 2 ms 6492 KB Output is correct
8 Correct 3 ms 6748 KB Output is correct
9 Correct 2 ms 6584 KB Output is correct
10 Correct 2 ms 6744 KB Output is correct
11 Correct 2 ms 6748 KB Output is correct
12 Correct 2 ms 6748 KB Output is correct

Subtask #4
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 6488 KB Output is correct
2 Correct 1 ms 6488 KB Output is correct
3 Correct 1 ms 6492 KB Output is correct
4 Correct 2 ms 6744 KB Output is correct
5 Correct 1 ms 6492 KB Output is correct
6 Correct 2 ms 6492 KB Output is correct
7 Correct 2 ms 6492 KB Output is correct
8 Correct 3 ms 6748 KB Output is correct
9 Correct 2 ms 6584 KB Output is correct
10 Correct 2 ms 6744 KB Output is correct
11 Correct 2 ms 6748 KB Output is correct
12 Correct 2 ms 6748 KB Output is correct
13 Correct 4 ms 6748 KB Output is correct
14 Correct 4 ms 7004 KB Output is correct
15 Correct 4 ms 6748 KB Output is correct
16 Correct 4 ms 6748 KB Output is correct
17 Correct 4 ms 6748 KB Output is correct
18 Correct 3 ms 6748 KB Output is correct
19 Correct 4 ms 6864 KB Output is correct

Subtask #5
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 177 ms 18892 KB Output is correct
2 Correct 180 ms 22272 KB Output is correct
3 Correct 131 ms 16116 KB Output is correct
4 Correct 165 ms 18896 KB Output is correct
5 Correct 176 ms 20292 KB Output is correct
6 Correct 176 ms 19144 KB Output is correct

Subtask #6
32.0 / 32.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 6488 KB Output is correct
2 Correct 1 ms 6488 KB Output is correct
3 Correct 1 ms 6492 KB Output is correct
4 Correct 2 ms 6744 KB Output is correct
5 Correct 1 ms 6492 KB Output is correct
6 Correct 2 ms 6492 KB Output is correct
7 Correct 2 ms 6492 KB Output is correct
8 Correct 3 ms 6748 KB Output is correct
9 Correct 2 ms 6584 KB Output is correct
10 Correct 2 ms 6744 KB Output is correct
11 Correct 2 ms 6748 KB Output is correct
12 Correct 2 ms 6748 KB Output is correct
13 Correct 4 ms 6748 KB Output is correct
14 Correct 4 ms 7004 KB Output is correct
15 Correct 4 ms 6748 KB Output is correct
16 Correct 4 ms 6748 KB Output is correct
17 Correct 4 ms 6748 KB Output is correct
18 Correct 3 ms 6748 KB Output is correct
19 Correct 4 ms 6864 KB Output is correct
20 Correct 177 ms 18892 KB Output is correct
21 Correct 180 ms 22272 KB Output is correct
22 Correct 131 ms 16116 KB Output is correct
23 Correct 165 ms 18896 KB Output is correct
24 Correct 176 ms 20292 KB Output is correct
25 Correct 176 ms 19144 KB Output is correct
26 Correct 213 ms 19308 KB Output is correct
27 Correct 206 ms 22264 KB Output is correct
28 Correct 219 ms 23072 KB Output is correct
29 Correct 144 ms 16468 KB Output is correct
30 Correct 206 ms 19404 KB Output is correct
31 Correct 194 ms 17476 KB Output is correct
32 Correct 220 ms 20816 KB Output is correct
33 Correct 212 ms 19212 KB Output is correct
34 Correct 131 ms 16124 KB Output is correct
35 Correct 210 ms 19404 KB Output is correct
36 Correct 194 ms 23376 KB Output is correct