oj.uz

답안 #842154

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
842154 2023-09-02T12:53:39 Z I_love_Hoang_Yen 봉쇄 시간 (IOI23_closing) C++17
8 / 100
 133 ms 40896 KB 
#include "closing.h"
#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using Graph = vector<vector<pair<int,ll>>>;

void dfs(int u, int fu, const Graph& g, vector<ll>& dists) {
    for (auto [v, w] : g[u]) {
        if (v == fu) continue;
        assert(dists[v] < 0);

        dists[v] = dists[u] + w;
        dfs(v, u, g, dists);
    }
}

vector<ll> get_dists(int u, const Graph& g) {
    int n = g.size();
    vector<ll> dists(n, -1);
    dists[u] = 0;

    dfs(u, -1, g, dists);
    return dists;
}

int no_common(int n, ll k, const vector<ll>& dx, const vector<ll>& dy) {
    vector<ll> all;
    for (auto d : dx) all.push_back(d);
    for (auto d : dy) all.push_back(d);
    sort(all.begin(), all.end());
    int cnt = 0;
    for (auto d : all) {
        if (d <= k) {
            k -= d;
            ++cnt;
        } else break;
    }
    return cnt;
}

// linear graph
int sub4(int n, ll k, int x, int y,
        const vector<ll>& dx, const vector<ll>& dy) { // x < y
    int res = no_common(n, k, dx, dy);

    // 0 .. [xl .. x .. xr]
    //                 [yl .. y .. yr] .. n-1
    // xr >= yl
    for (int yl = x; yl <= y; ++yl) {
        ll can = k;
        int cur = 0;
        // 0. for both x and y: visit yl
        can -= max(dx[yl], dy[yl]);
        cur += 2;

        // 1. for x:
        // 1.1. visit all of [x, yl)
        for (int i = x; i < yl; ++i) {
            can -= dx[i];
            cur++;
        }
        // 1.2. option to visit others
        vector<ll> rem_dx;
        for (int i = 0; i < x; ++i) {
            rem_dx.push_back(dx[i]);
        }
        for (int i = yl + 1; i < n; ++i) {
            rem_dx.push_back(max(0ll, dx[i] - dy[i]));
        }

        // 2. for y:
        // 2.1. visit all of (yl, y]
        for (int i = yl+1; i <= y; i++) {
            can -= dy[i];
            cur++;
        }
        // 2.2. option to visit others
        vector<ll> rem_dy;
        for (int i = y+1; i < n; ++i) {
            rem_dy.push_back(dy[i]);
        }
        if (yl == x) {
            for (int i = 0; i < x; ++i) {
                rem_dy.push_back(max(dy[i] - dx[i], 0ll));
            }
        }

        if (can >= 0) {
            res = max(res, no_common(n, can, rem_dx, rem_dy) + cur);
        }
    }
    return res;
}

int max_score(int n, int X, int Y, long long K,
              vector<int> U, vector<int> V, vector<int> W) {
    Graph g(n);
    assert(int(U.size()) == n-1);
    assert(int(V.size()) == n-1);
    assert(int(W.size()) == n-1);
    for (int i = 0; i < n - 1; i++) {
        int u = U[i];
        int v = V[i];
        int w = W[i];
        g[u].emplace_back(v, w);
        g[v].emplace_back(u, w);
    }
    if (X > Y) swap(X, Y);

    auto distsX = get_dists(X, g);
    auto distsY = get_dists(Y, g);
    
    return no_common(n, K, distsX, distsY);
}

Subtask #0
0.0 / 0.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #1
8.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 132 ms 33120 KB Output is correct
2 Correct 133 ms 40896 KB Output is correct
3 Correct 75 ms 2896 KB Output is correct

Subtask #2
0.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB Output is correct
2 Incorrect 1 ms 344 KB 1st lines differ - on the 1st token, expected: '30', found: '24'
3 Halted 0 ms 0 KB -

Subtask #3
0.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB Output is correct
2 Incorrect 1 ms 344 KB 1st lines differ - on the 1st token, expected: '30', found: '24'
3 Halted 0 ms 0 KB -

Subtask #4
0.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB Output is correct
2 Incorrect 1 ms 344 KB 1st lines differ - on the 1st token, expected: '30', found: '24'
3 Halted 0 ms 0 KB -

Subtask #5
0.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #6
0.0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #7
0.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #8
0.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #9
0.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -