Submission #932466

#	Time	Username	Problem	Language	Result	Execution time	Memory
932466		VMaksimoski008	Palembang Bridges (APIO15_bridge)	C++14	31 / 100	2093 ms	2184 KiB

bridge

#include <bits/stdc++.h>

#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
//#define int long long

using namespace std;

using ll = long long;
using ull = unsigned long long;
using ld = long double;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

const int mod = 1e9 + 7;
const int LOG = 20;
const int maxn = 1e5 + 5;
const double eps = 1e-9;

int32_t main() {
    int k, n;
    cin >> k >> n;

    ll ans = 0, ans2 = 0;
    vector<int> v;
    vector<pii> p;

    for(int i=0; i<n; i++) {
        char a, b;
        int c, d;
        cin >> a >> c >> b >> d;

        if(a == b) {
            ans += abs(c - d);
            ans2 += abs(c - d);
        } else {
            v.push_back(c);
            v.push_back(d);
            p.push_back({ c, d });
        }
    }

    if(v.size() == 0) {
        cout << ans << '\n';
        return 0;
    }

    sort(all(v));

    if(k == 1) {
        ll med = (v[v.size()/2] + v[v.size()/2-1]) / 2;

        for(int &x : v) ans += abs(med - x);

        cout << ans + v.size() / 2 << '\n';
        return 0;
    } 

    ans = 1e18;
    for(int i=0; i<v.size(); i++) {
        for(int j=i+1; j<v.size(); j++) {
            ll res = 0;
            for(pii &x : p)
                res += min(abs(x.first - v[i]) + abs(x.second - v[i]), abs(x.first - v[j]) + abs(x.second - v[j]));
            ans = min(ans, res);
        }
    }

    cout << ans + ans2 + v.size() / 2 << '\n';
    return 0;
}

Compilation message (stderr)

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

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