제출 #1246184

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1246184		tochika	Palembang Bridges (APIO15_bridge)	C++20	100 / 100	63 ms	6076 KiB

bridge

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

#define IN "iii.txt"
#define OUT "ooo.txt"
#define MULTI_TEST 0

lint l = 0, r = 0;
priority_queue<lint> under;
priority_queue<lint, vector<lint>, greater<lint>> above;

void add(lint x) {
    int med = (under.size() ? under.top() : 1000000001);

    if (x <= med) {
        under.push(x);
        l += x;
    }

    else {
        above.push(x);
        r += x;
    }

    if (above.size() + 1 < under.size()) {
        int t = under.top();
        under.pop();

        above.push(t);
        r += t;
        l -= t;
    }

    else if (under.size() < above.size()) {
        int t = above.top();
        above.pop();

        under.push(t);
        r -= t;
        l += t;
    }
}

void solve() {
    int k, n;
    cin >> k >> n;

    lint cnt = 0;
    vector<pair<lint, lint>> v = {{0LL, 0LL}};

    for (int i = 0; i < n; i++) {
        char a, b;
        lint x, y;

        cin >> a >> x >> b >> y;

        if (a == b)
            cnt += abs(x - y);

        else v.push_back({x, y});
    }

    sort(v.begin(), v.end(), [&](pair<lint, lint>& m1, pair<lint, lint>& m2){
         return m1.first + m1.second < m2.first + m2.second;
    });

    n = v.size() - 1;
    cnt += n;
    
    vector<lint> pf(n + 1, 0);

    l = r = 0;

    for (int i = 1; i <= n; i++) {
        add(v[i].first);
        add(v[i].second);

        pf[i] = r - l;
    }

    lint ans = pf[n];

    if (k == 2) {
        while (under.size()) under.pop();
        while (above.size()) above.pop();

        l = r = 0;

        for (int i = n; i > 0; i--) {
            add(v[i].first);
            add(v[i].second);

            ans = min(ans, r - l + pf[i - 1]);
        }
    }

    cout << ans + cnt;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    if (fopen(IN, "r")) {
        freopen(IN, "r", stdin);
        freopen(OUT, "w", stdout);
    }

    int t = 1;
    if (MULTI_TEST) {
        cin >> t;
    }

    for (int i = 1; i <= t; i++) {
        solve();
        cout << '\n';
    }

    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

bridge.cpp: In function 'int main()':
bridge.cpp:106:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  106 |         freopen(IN, "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~
bridge.cpp:107:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  107 |         freopen(OUT, "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~

• Subtask 1: 8 / 8
• Subtask 2: 14 / 14
• Subtask 3: 9 / 9
• Subtask 4: 32 / 32
• Subtask 5: 37 / 37