oj.uz

답안 #1114361

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1114361 2024-11-18T16:48:10 Z ShaShi Palembang Bridges (APIO15_bridge) C++17
63 / 100
 2000 ms 105288 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 (id == 1) cout << "+ " << arr[ind].F << " " << arr[ind].S << " " << s << endl;

    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 (id == 1) cout << "- " << arr[ind].F << " " << arr[ind].S << endl;

    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[y]*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=1; i<=n; i++) add(id(arr[i].F+arr[i].S), i);

    ans = INF;
    int j = 1;

    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 9 ms 49488 KB Output is correct
2 Correct 9 ms 49232 KB Output is correct
3 Correct 14 ms 49232 KB Output is correct
4 Correct 13 ms 49488 KB Output is correct
5 Correct 14 ms 49488 KB Output is correct
6 Correct 17 ms 49436 KB Output is correct
7 Correct 15 ms 49512 KB Output is correct
8 Correct 18 ms 49488 KB Output is correct
9 Correct 21 ms 49488 KB Output is correct
10 Correct 21 ms 49232 KB Output is correct
11 Correct 20 ms 49500 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 22 ms 49232 KB Output is correct
2 Correct 22 ms 49240 KB Output is correct
3 Correct 23 ms 49232 KB Output is correct
4 Correct 21 ms 49500 KB Output is correct
5 Correct 19 ms 49512 KB Output is correct
6 Correct 24 ms 49232 KB Output is correct
7 Correct 27 ms 49504 KB Output is correct
8 Correct 27 ms 49548 KB Output is correct
9 Correct 22 ms 49488 KB Output is correct
10 Correct 21 ms 49232 KB Output is correct
11 Correct 22 ms 49488 KB Output is correct
12 Correct 32 ms 52932 KB Output is correct
13 Correct 108 ms 64844 KB Output is correct
14 Correct 48 ms 53436 KB Output is correct
15 Correct 70 ms 58412 KB Output is correct
16 Correct 37 ms 52940 KB Output is correct
17 Correct 73 ms 64912 KB Output is correct
18 Correct 80 ms 64952 KB Output is correct
19 Correct 108 ms 64444 KB Output is correct
20 Correct 40 ms 53112 KB Output is correct
21 Correct 85 ms 64964 KB Output is correct

Subtask #3
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 19 ms 49404 KB Output is correct
2 Correct 22 ms 49244 KB Output is correct
3 Correct 26 ms 49232 KB Output is correct
4 Correct 28 ms 49488 KB Output is correct
5 Correct 26 ms 49232 KB Output is correct
6 Correct 25 ms 49232 KB Output is correct
7 Correct 25 ms 49368 KB Output is correct
8 Correct 28 ms 49488 KB Output is correct
9 Correct 26 ms 49232 KB Output is correct
10 Correct 25 ms 49480 KB Output is correct
11 Correct 23 ms 49328 KB Output is correct
12 Correct 27 ms 49488 KB Output is correct

Subtask #4
32.0 / 32.0

# 결과 실행 시간 메모리 Grader output
1 Correct 23 ms 49224 KB Output is correct
2 Correct 26 ms 49244 KB Output is correct
3 Correct 27 ms 49232 KB Output is correct
4 Correct 28 ms 49232 KB Output is correct
5 Correct 29 ms 49456 KB Output is correct
6 Correct 24 ms 49232 KB Output is correct
7 Correct 24 ms 49224 KB Output is correct
8 Correct 25 ms 49488 KB Output is correct
9 Correct 24 ms 49232 KB Output is correct
10 Correct 27 ms 49488 KB Output is correct
11 Correct 27 ms 49212 KB Output is correct
12 Correct 34 ms 49232 KB Output is correct
13 Correct 25 ms 49480 KB Output is correct
14 Correct 29 ms 49488 KB Output is correct
15 Correct 785 ms 50000 KB Output is correct
16 Correct 24 ms 49232 KB Output is correct
17 Correct 31 ms 49480 KB Output is correct
18 Correct 114 ms 49680 KB Output is correct
19 Correct 27 ms 49500 KB Output is correct
20 Correct 931 ms 50004 KB Output is correct
21 Correct 225 ms 49744 KB Output is correct
22 Correct 814 ms 49988 KB Output is correct
23 Correct 21 ms 49488 KB Output is correct
24 Correct 701 ms 54024 KB Output is correct

Subtask #5
0 / 37.0

# 결과 실행 시간 메모리 Grader output
1 Correct 21 ms 49232 KB Output is correct
2 Correct 18 ms 49232 KB Output is correct
3 Correct 15 ms 49532 KB Output is correct
4 Correct 19 ms 49312 KB Output is correct
5 Correct 19 ms 49376 KB Output is correct
6 Correct 16 ms 49232 KB Output is correct
7 Correct 18 ms 49232 KB Output is correct
8 Correct 20 ms 49300 KB Output is correct
9 Correct 16 ms 49232 KB Output is correct
10 Correct 18 ms 49500 KB Output is correct
11 Correct 13 ms 49232 KB Output is correct
12 Correct 21 ms 49488 KB Output is correct
13 Correct 18 ms 49656 KB Output is correct
14 Correct 20 ms 49576 KB Output is correct
15 Correct 755 ms 49996 KB Output is correct
16 Correct 12 ms 49232 KB Output is correct
17 Correct 15 ms 49488 KB Output is correct
18 Correct 99 ms 49624 KB Output is correct
19 Correct 15 ms 49488 KB Output is correct
20 Correct 919 ms 50000 KB Output is correct
21 Correct 226 ms 49744 KB Output is correct
22 Correct 788 ms 49744 KB Output is correct
23 Correct 14 ms 49488 KB Output is correct
24 Correct 687 ms 49972 KB Output is correct
25 Correct 53 ms 60404 KB Output is correct
26 Correct 587 ms 61488 KB Output is correct
27 Execution timed out 2055 ms 105288 KB Time limit exceeded
28 Halted 0 ms 0 KB -