Submission #1116343

#	Time	Username	Problem	Language	Result	Execution time	Memory
1116343		gdragon	Palembang Bridges (APIO15_bridge)	C++17	0 / 100	2 ms	844 KiB

bridge

#include <bits/stdc++.h>
using namespace std;
#define TASK "long"
#define fi first
#define se second
#define ll long long
#define pb push_back
#define ALL(x) (x).begin(), (x).end()
#define GETBIT(mask, i) ((mask) >> (i) & 1)
#define MASK(i) ((1LL) << (i))
#define SZ(x) ((int)(x).size())
#define mp make_pair
#define CNTBIT(mask) __builtin_popcount(mask)
template<class X, class Y> bool maximize(X &x, Y y){ if (x < y) {x = y; return true;} return false;};
template<class X, class Y> bool minimize(X &x, Y y){ if (x > y) {x = y; return true;} return false;};
typedef pair<int, int> ii;
const int N = 1e5 + 5;
const long long INF = (long long)1e18 + 7;
const int mod = 1e9 + 7;
pair<int, int> b[N], a[N];
int n, m, k;
long long ans = 0;
void read() {
    cin >> k >> n;
    m = 0;
    for(int i = 1; i <= n; i++) {
        char c1, c2;
        cin >> c1 >> b[i].fi >> c2 >> b[i].se;
        if (c1 == c2) ans += abs(b[i].fi - b[i].se);
        else {
            if (c1 == 'B') swap(b[i].fi, b[i].se);
            a[++m] = b[i];
        }
    }
    n = m;
}
void sub1() {
    vector<int> val;
    val.push_back(-1);
    for(int i = 1; i <= n; i++) {
        val.push_back(a[i].fi);
        val.push_back(a[i].se);
    }
    sort(ALL(val));
    vector<long long> f(SZ(val) + 1, 0);
    f[0] = 0;
    for(int i = 1; i < SZ(val); i++) {
        f[i] = f[i - 1] + val[i];
    }
    auto getSum = [&](int l, int r) -> long long {
        if (l > r) return 0;
        return f[r] - f[l - 1];
    };
    long long tmp = INF;
    for(int i = 1; i < SZ(val); i++) {
        // long long lef = 1LL * val[i] * i - getSum(1, i);
        // long long rig = getSum(i, SZ(val) - 1) - 1LL * val[i] * (SZ(val) - 1 - i + 1);
        // minimize(tmp, lef + rig + n);
        long long tmp1 = 0;
        for(int j = 1; j < i; j++) tmp1 += abs(val[i] - val[j]);
        for(int j = i + 1; j < SZ(val); j++) tmp1 += abs(val[i] - val[j]);
        minimize(tmp, tmp1 + n);
        // cerr << val[i] << ' ' << lef << ' ' << rig << ' ' << lef + rig + n + ans << endl;
    }
    ans += tmp;
    cout << ans;
}
void solve() {
    if (k == 1) sub1();
    else assert(false);
}
signed main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    if (fopen(TASK".inp", "r")) {
        freopen(TASK".inp", "r", stdin);
        freopen(TASK".out", "w", stdout);
    }
    int test = 1;
    // cin >> test;
    while(test--) {
        read();
        solve();
    }
    return 0;
}

Compilation message (stderr)

bridge.cpp: In function 'void sub1()':
bridge.cpp:50:10: warning: variable 'getSum' set but not used [-Wunused-but-set-variable]
   50 |     auto getSum = [&](int l, int r) -> long long {
      |          ^~~~~~
bridge.cpp: In function 'int main()':
bridge.cpp:75:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   75 |         freopen(TASK".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
bridge.cpp:76:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   76 |         freopen(TASK".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

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