Submission #922161

#	Time	Username	Problem	Language	Result	Execution time	Memory
922161		asdasdqwer	Palembang Bridges (APIO15_bridge)	C++14	22 / 100	84 ms	7356 KiB

bridge

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

#define int int64_t
#define pii array<int,2>

struct cmp {
    bool operator()(const pii &x, const pii &y) {
        return (x[0] + x[1]) < (y[0] + y[1]);
    }
};

signed main() {
    int k, n;cin>>k>>n;
    vector<pii> dif;
    int dis = 0;
    for (int i=0;i<n;i++) {
        char c1,c2;
        int p1,p2;
        cin>>c1>>p1>>c2>>p2;
        if (c1 == c2) {
            dis += abs(p1-p2);
        }
        else {
            dif.push_back({p1,p2});
        }
    }

    if (dif.size() == 0) {
        cout<<dis<<"\n";
        return 0;
    }

    if (k == 1 || n == 1) {

        vector<int> al;
        for (auto x:dif) {
            al.push_back(x[0]);
            al.push_back(x[1]);
        }

        sort(al.begin(),al.end());

        int med = al[al.size() / 2];

        for (int x:al) {
            dis += abs(x - med);
        }

        dis += dif.size();

        cout<<dis<<"\n";

        return 0;
    }

    sort(dif.begin(),dif.end(), cmp());

    multiset<int> m1, m2;
    int s1 = 0, s2 = 0;

    function<void(int)> insert=[&](int num) {
        m1.insert(num);
        s1 += num;
        while ((m1.size() > m2.size() && ((m1.size() + m2.size()) % 2 == 0)) ||
                (m1.size() - 1 > m2.size() && ((m1.size() + m2.size()) % 2 == 1))) {
            num = *m1.rbegin();
            m1.erase(m1.find(num));
            m2.insert(num);
            s1 -= num;
            s2 += num;
        }
    };

    vector<int> cst1;
    vector<int> tmp1;
    for (auto x:dif) {
        insert(x[0]);
        insert(x[1]);
        int med = *m1.rbegin();
        tmp1.push_back(med);
        cst1.push_back((m1.size() * med - s1) + (s2 - m2.size() * med));
    }

    reverse(dif.begin(),dif.end());
    vector<int> cst2;
    vector<int> tmp2;
    m1.clear();m2.clear();
    s1=s2=0;
    for (auto x:dif) {
        insert(x[0]);
        insert(x[1]);
        int med = *m1.rbegin();
        tmp2.push_back(med);
        cst2.push_back((m1.size() * med - s1) + (s2 - m2.size() * med));
    }

    reverse(cst2.begin(),cst2.end());
    reverse(tmp2.begin(), tmp2.end());
    int res = 1e17;

    for (int i=1;i<cst1.size();i++) {
        res = min(res, cst1[i-1] + cst2[i]);
    }

    res += dif.size();

    cout<<res+dis<<"\n";
}

Compilation message (stderr)

bridge.cpp: In function 'int main()':
bridge.cpp:102:19: warning: comparison of integer expressions of different signedness: 'int64_t' {aka 'long int'} and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  102 |     for (int i=1;i<cst1.size();i++) {
      |                  ~^~~~~~~~~~~~

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