제출 #586123

#제출 시각아이디문제언어결과실행 시간메모리
586123Do_you_copyPalembang Bridges (APIO15_bridge)C++14
22 / 100
42 ms4336 KiB
#include <bits/stdc++.h>
#define taskname "test"
#define fi first
#define se second
#define pb push_back
#define faster ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;

using ll = long long;
using pii = pair <ll, ll>;

ll min(const ll &a, const ll &b){
    return (a < b) ? a : b;
}

ll max(const ll &a, const ll &b){
    return (a > b) ? a : b;
}

const int Mod = 1000000007;
const int maxN = 2e5;
const int inf = 1e9 + 1;
int k, n;
ll suf[maxN], pre[maxN];

ll cnt = 0;

struct TNode{
    TNode *l, *r;
    ll val;
    ll sum;
    TNode (const ll &i, const ll &j) : l(nullptr), r(nullptr), val(i), sum(j){}
    TNode (TNode* i, TNode* j){
        l = i; r = j;
        val = 0; sum = 0;
        if (l){
            val += l->val;
            sum += l->sum;
        }
        if (r){
            val += r->val;
            sum += r->sum;
        }
    }
};

TNode* root[3];

void create_node(TNode* id){
    if (!id->l){
        ++cnt;
        id->l = new TNode(0ll, 0ll);
    }
    if (!id->r){
        ++cnt;
        id->r = new TNode(0ll, 0ll);
    }
}

void update(TNode* id, int i, int l = 0, int r = inf){
    if (l == r){
        ++id->val; id->sum += i;
        return;
    };
    ll m = (l + r) / 2;
    create_node(id);
    if (m >= i) update(id->l, i, l, m);
    else update(id->r, i, m + 1, r);
    id->sum = id->l->sum + id->r->sum;
    id->val = id->l->val + id->r->val;
}

ll remainer;

int get(TNode* left, TNode* right, int k, int l = 0, int r = inf){
    //finding the kth largest numbers from left -> right
    if (l == r){
        remainer = right->sum - left->sum - (l * k);
        return l;
    }
    ll m = (l + r) / 2;
    int cntl = (left->l) ? left->l->val : 0;
    int cntr = (right->l) ? right->l->val : 0;
    if (cntr - cntl < k) return get((left->r) ? left->r : left, (right->r) ? right->r : right, k - cntr + cntl, m + 1, r);
    return get((left->l) ? left->l : left, (right->l) ? right->l : right, k, l, m);
}

ll getsum(TNode* left, TNode* right, int i, int j, int l = 0, int r = inf){
    //get sum of all elements ranged between i and j from left to right
    if (r < i || l > j) return 0;
    if (i <= l && r <= j){
        return right->sum - left->sum;
    }
    ll m = (l + r) / 2;
    ll x = getsum((left->l) ? left->l : left, (right->l) ? right->l : right, i, j, l, m);
    ll y = getsum((left->r) ? left->r : left, (right->r) ? right->r : right, i, j, m + 1, r);
    return x + y;
}

void Init(){
    ll ans = 0;
    cin >> k >> n;
    if (k == 1){
        vector <int> v;
        for (ll i = 1; i <= n; ++i){
            char p, q; ll s, t;
            cin >> p >> s >> q >> t;
            if (p != q){
                ++ans;
                v.pb(s);
                v.pb(t);
            }
            else{
                ans += abs(s - t);
            }
        }
        v.push_back(0);
        sort(v.begin(), v.end());
        int m = v.size();
        ll tem = LLONG_MAX;
        for (ll i = 1; i < m; ++i){
            pre[i] = pre[i - 1] + v[i];
        }
        for (ll i = m - 1; i > 0; --i){
            suf[i] = suf[i + 1] + v[i];
        }
        for (ll i = 1; i < m; ++i){
            tem = min(tem, v[i] * (i - 1) - pre[i - 1] + suf[i + 1] - v[i] * (m - i - 1));
        }
        if (m == 1) tem = 0;
        cout << ans + tem;
        return;
    }
    vector <pii> v;
    for (int i = 1; i <= n; ++i){
        char p, q; int s, t;
        cin >> p >> s >> q >> t;
        if (p != q){
            ++ans;
            v.pb({s, t});
        }
        else ans += abs(s - t);
    }
    if (v.empty()){
        cout << ans; return;
    }
    if (v.size() == 1){
        cout << ans + abs(v[0].fi - v[0].se);
        return;
    }
    sort(v.begin(), v.end(),[](const pii &i, const pii &j){
        return (i.fi + i.se) / 2 < (j.fi + j.se) / 2;
    });
    root[0] = new TNode(0, 0ll);
    root[1] = new TNode(0, 0ll);
    root[2] = new TNode(0, 0ll);
    for (ll i = 1; i <= v.size(); ++i){
        pre[i] = pre[i - 1] + v[i - 1].fi + v[i - 1].se;
        update(root[2], v[i - 1].fi);
        update(root[2], v[i - 1].se);
    }
    ll tem = LLONG_MAX;
    for (ll i = 0; i < v.size() - 1; ++i){
        update(root[1], v[i].fi);
        update(root[1], v[i].se);
        ll x = get(root[0], root[1], i + 1);
        ll l = getsum(root[0], root[1], 0, x) - remainer;
        ll y = get(root[1], root[2], v.size() - i - 1);
        ll r = getsum(root[1], root[2], 0, y) - remainer;
        tem = min(tem, pre[i + 1] - 2 * l + pre[v.size()] - pre[i + 1] - 2 * r);
    }
    //cerr << cnt << "\n";
    cout << tem + ans;
}

signed main(){
    if (fopen(taskname".txt", "r")){
        freopen(taskname".txt", "r", stdin);
        //freopen(taskname".out", "w", stdout);
    }
    faster
    Init();
}

컴파일 시 표준 에러 (stderr) 메시지

bridge.cpp: In function 'void Init()':
bridge.cpp:157:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  157 |     for (ll i = 1; i <= v.size(); ++i){
      |                    ~~^~~~~~~~~~~
bridge.cpp:163:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  163 |     for (ll i = 0; i < v.size() - 1; ++i){
      |                    ~~^~~~~~~~~~~~~~
bridge.cpp: In function 'int main()':
bridge.cpp:178:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  178 |         freopen(taskname".txt", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...