oj.uz

답안 #1114358

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1114358 2024-11-18T16:36:03 Z ShaShi Palembang Bridges (APIO15_bridge) C++17
22 / 100
 110 ms 66488 KB 
#include <bits/stdc++.h>
 
#define int long long 
 
// #pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")
 
#define F first 
#define S second
#define mp make_pair
#define pb push_back
#define all(x) x.begin(), x.end()
#define kill(x) cout << x << "\n", exit(0);
#define pii pair<int, int>
#define pll pair<long long, long long>
#define endl "\n"
 
 
 
using namespace std;
typedef long long ll;
// typedef __int128_t lll;
typedef long double ld;
 
 
const int MAXN = (int)1e6 + 7;
const int MOD = 998244353;
const ll INF = (ll)1e18 + 7;
 
 
int n, m, k, tmp, t, tmp2, tmp3, tmp4, u, v, w, flag, q, ans, N, res, pre_ans;
pii arr[MAXN], di[MAXN];
char ch, ch2;
vector<int> vec[2][MAXN], cmp;
int ps[2][MAXN], cnt[2][MAXN];



inline int id(int n) { return lower_bound(all(cmp), n)-cmp.begin(); }


vector<pii> _vec;
map<int, int> _cnt;

inline int solve1() {
    for (int i=1; i<=n; i++) {
        u = arr[i].F; v = arr[i].S;

        _cnt[u]++, _cnt[v]++, _vec.pb({u, v}), tmp2 += 2;
    }

    int res2 = 0;
 
    for (auto cur: _cnt) {
        tmp += cur.S;
 
        // cout << cur.F << " " << cur.S << endl;
 
        if (tmp+tmp >= tmp2) {
            tmp = cur.F;
            break;
        }
    }
 
    for (auto cur:_cnt) res2 += abs(cur.F-tmp)*cur.S;

    tmp = tmp2 = 0;
 
    return res2+_vec.size();
}


/* Segment Tree */
#define mid ((l+r)>>1)
#define lid (id<<1)
#define rid (lid|1)


struct D {
    int sigmaL, sigmaR, cnt;

    D () {}
} seg[MAXN<<2], emp;


inline D merge(D x, D y) {
    D res;

    res.sigmaL = x.sigmaL+y.sigmaL;
    res.sigmaR = x.sigmaR+y.sigmaR;
    res.cnt = x.cnt+y.cnt;

    return res;
}


void add(int s, int ind, int l=0, int r=N, int id=1) {
    if (l+1 == r) {
        seg[id].cnt++;
        seg[id].sigmaL += arr[ind].F;
        seg[id].sigmaR += arr[ind].S;

        return;
    }

    if (s < mid) add(s, ind, l, mid, lid);
    else add(s, ind, mid, r, rid);

    seg[id] = merge(seg[lid], seg[rid]);
}


void rem(int s, int ind, int l=0, int r=N, int id=1) {
    if (l+1 == r) {
        seg[id].cnt--;
        seg[id].sigmaL -= arr[ind].F;
        seg[id].sigmaR -= arr[ind].S;

        return;
    }

    if (s < mid) rem(s, ind, l, mid, lid);
    else rem(s, ind, mid, r, rid);

    seg[id] = merge(seg[lid], seg[rid]);
}


D get(int s, int t, int l=0, int r=N, int id=1) {
    if (s >= t) return emp;
    if (s <= l && t >= r) return seg[id];
    if (t <= mid) return get(s, t, l, mid, lid);
    if (s >= mid) return get(s, t, mid, r, rid);
    return merge(get(s, t, l, mid, lid), get(s, t, mid, r, rid));
}

/* Segment Tree */


inline int cost(int x, int y) {
    int res = ps[0][x] + ps[1][y];

    D cur;
    // cur = get(0, id(cmp[x]+cmp[y])); res += cur.sigmaL - cmp[x]*cur.cnt;
    // cur = get(id(cmp[x]+cmp[y]), N); res += cmp[x]*cur.cnt - cur.sigmaR;

    return res;
}


int32_t main() {
    #ifdef LOCAL
    freopen("inp.in", "r", stdin);
    freopen("res.out", "w", stdout);
    #else
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    #endif
 
    cin >> k >> n;

    N = 0;
 
    for (int i=1; i<=n; i++) {
        cin >> ch >> u >> ch2 >> v;
 
        if (ch == ch2) res += abs(u-v);
        else arr[++N] = {u, v};
    }

    n = N; pre_ans = res; res += solve1();

    if (k == 1) kill(res);

    for (int i=1; i<=n; i++) {
        u = arr[i].F; v = arr[i].S;
        if (u > v) swap(arr[i].F, arr[i].S);
        pre_ans += abs(arr[i].F-arr[i].S)+1;
        cmp.pb(u); cmp.pb(v); cmp.pb(u+v);
    }

    sort(all(cmp));
    cmp.resize(unique(all(cmp))-cmp.begin());
    N = cmp.size(); emp.sigmaL = emp.sigmaR = emp.cnt = 0;

    // cout << "@ " << endl;
    // for (int u:cmp) cout << u << " ";
    // cout << endl;


    for (int i=1; i<=n; i++) {
        u = di[i].F = id(arr[i].F);
        v = di[i].S = id(arr[i].S);

        cnt[0][u]++; cnt[1][v]++;
        vec[0][u].pb(i); vec[1][v].pb(i);
    }

    
    tmp = tmp2 = 0;
    
    for (int i=0; i<N; i++) {
        ps[0][i] = cmp[i]*tmp2-tmp;

        for (int ind:vec[1][i]) {
            tmp2++;
            tmp += arr[ind].S;
        }
    }


    tmp = tmp2 = 0;
    
    for (int i=N-1; i>=0; i--) {
        ps[1][i] = tmp-cmp[i]*tmp2;

        for (int ind:vec[0][i]) {
            tmp2++;
            tmp += arr[ind].F;
        }
    }


    // for (int i=0; i<N; i++) cout << "! " << i << " " << ps[0][i] << endl;


    // for (int i=1; i<=n; i++) add(id(arr[i].F+arr[i].S), i);

    ans = INF;
    int j = 1;

    // exit(0);

    for (int i=0; i<N; i++) {
        vector<int> temp;

        for (int j=i+1; j<N; j++) {
            // cout << "! " << i << " " << j << " : " << cmp[i] << " " << cmp[j] << " : " << cost(i, j) << " " << pre_ans+2*cost(i, j) << endl;

            ans = min(ans, pre_ans+2*cost(i, j));

            for (auto ind:vec[1][j]) {
                if (di[ind].F <= i) continue;

                temp.pb(ind); add(id(arr[ind].F+arr[ind].S), ind);
            }
        }

        for (auto ind:temp) rem(id(arr[ind].F+arr[ind].S), ind);
    }

    // cout << ans << " " << res << endl;
    cout << min(ans, res) << endl;
    
    return 0;
}

Compilation message

bridge.cpp: In function 'int32_t main()':
bridge.cpp:230:9: warning: unused variable 'j' [-Wunused-variable]
  230 |     int j = 1;
      |         ^

Subtask #1
8.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 51536 KB Output is correct
2 Correct 8 ms 49488 KB Output is correct
3 Correct 9 ms 49744 KB Output is correct
4 Correct 10 ms 49496 KB Output is correct
5 Correct 15 ms 49488 KB Output is correct
6 Correct 19 ms 49420 KB Output is correct
7 Correct 17 ms 49488 KB Output is correct
8 Correct 13 ms 49568 KB Output is correct
9 Correct 18 ms 49488 KB Output is correct
10 Correct 13 ms 49232 KB Output is correct
11 Correct 16 ms 49488 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 21 ms 49232 KB Output is correct
2 Correct 23 ms 49232 KB Output is correct
3 Correct 23 ms 49436 KB Output is correct
4 Correct 25 ms 49480 KB Output is correct
5 Correct 20 ms 49488 KB Output is correct
6 Correct 16 ms 49232 KB Output is correct
7 Correct 20 ms 49488 KB Output is correct
8 Correct 21 ms 49488 KB Output is correct
9 Correct 21 ms 49488 KB Output is correct
10 Correct 20 ms 49232 KB Output is correct
11 Correct 20 ms 49656 KB Output is correct
12 Correct 39 ms 52960 KB Output is correct
13 Correct 110 ms 64820 KB Output is correct
14 Correct 52 ms 53388 KB Output is correct
15 Correct 69 ms 58564 KB Output is correct
16 Correct 39 ms 53072 KB Output is correct
17 Correct 72 ms 64952 KB Output is correct
18 Correct 81 ms 66488 KB Output is correct
19 Correct 106 ms 64388 KB Output is correct
20 Correct 29 ms 52940 KB Output is correct
21 Correct 73 ms 64940 KB Output is correct

Subtask #3
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 16 ms 49232 KB Output is correct
2 Correct 16 ms 49232 KB Output is correct
3 Correct 14 ms 49400 KB Output is correct
4 Incorrect 14 ms 49488 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #4
0 / 32.0

# 결과 실행 시간 메모리 Grader output
1 Correct 15 ms 49244 KB Output is correct
2 Correct 18 ms 49412 KB Output is correct
3 Correct 13 ms 49232 KB Output is correct
4 Incorrect 14 ms 49484 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #5
0 / 37.0

# 결과 실행 시간 메모리 Grader output
1 Correct 17 ms 49232 KB Output is correct
2 Correct 15 ms 49232 KB Output is correct
3 Correct 16 ms 49232 KB Output is correct
4 Incorrect 14 ms 49656 KB Output isn't correct
5 Halted 0 ms 0 KB -