Submission #386745

#	Time	Username	Problem	Language	Result	Execution time	Memory
386745		ParsaAlizadeh	Teams (IOI15_teams)	C++17	100 / 100	1710 ms	451708 KiB

teams

#include <bits/stdc++.h>
#include "teams.h"
using namespace std;

typedef pair<int, int> pii;

struct Node {
    int cnt;
    Node *lc, *rc;

    Node() : cnt(0), lc(this), rc(this) {}

    Node* add(int l, int r, int ind) {
        if (ind < l || r <= ind)
            return this;
        Node* cur = new Node();
        if (r - l == 1) {
            cur->cnt = cnt + 1;
            return cur;
        }
        int mid = (l + r) >> 1;
        cur->lc = lc->add(l, mid, ind);
        cur->rc = rc->add(mid, r, ind);
        cur->cnt = cnt + 1;
        return cur; 
    }

    int get(Node* prv, int l, int r, int qr) {
        if (qr <= l)
            return 0;
        if (r <= qr)
            return cnt - prv->cnt;
        int mid = (l + r) >> 1;
        return lc->get(prv->lc, l, mid, qr) + rc->get(prv->rc, mid, r, qr);
    }

    int query(Node* prv, int l, int r, int t) {
        if (cnt - prv->cnt < t)
            return r;
        if (r - l == 1)
            return l;
        int mid = (l + r) >> 1;
        int lft = lc->cnt - prv->lc->cnt;
        if (lft >= t)
            return lc->query(prv->lc, l, mid, t);
        return rc->query(prv->rc, mid, r, t - lft);
    }

    Node* sub(Node* prv, int l, int r, int qr) {
        if (qr <= l)
            return prv;
        if (r <= qr)
            return this;
        int mid = (l + r) >> 1;
        Node* cur = new Node();
        cur->lc = lc->sub(prv->lc, l, mid, qr);
        cur->rc = rc->sub(prv->rc, mid, r, qr);
        cur->cnt = cur->lc->cnt + cur->rc->cnt;
        return cur;
    }
};

int n;
vector<Node*> root;
vector<int> R;
vector<pii> vec;

void init(int n, int A[], int B[]) {
    ::n = n;
    for (int i = 0; i < n; i++) {
        vec.emplace_back(B[i], A[i]);
    }
    sort(begin(vec), end(vec));
    vector<int> L[n];
    for (int i = 0; i < n; i++) {
        L[vec[i].second - 1].push_back(i);
    }
    root.resize(n + 1);
    R.resize(n + 1);
    root[0] = new Node();
    for (int i = 0; i < n; i++) {
        root[i + 1] = root[i];
        for (int j : L[i]) {
            root[i + 1] = root[i + 1]->add(0, n, j);
        }
        R[i + 1] = lower_bound(begin(vec), end(vec), pii{i + 1, -1}) - begin(vec);
    }
}

int can(int M, int K[]) {
    sort(K, K + M);
    Node* cur = new Node();
    for (int i = 0; i < M; i++) {
        int t = root[K[i]]->get(cur, 0, n, R[K[i]]);
        int qr = root[K[i]]->query(cur, 0, n, t + K[i]);
        if (qr == n || vec[qr].first < K[i])
            return 0;
        cur = root[K[i]]->sub(cur, 0, n, qr + 1);
    }
    return 1;
}

Compilation message (stderr)

teams.cpp: In function 'void init(int, int*, int*)':
teams.cpp:68:34: warning: declaration of 'n' shadows a global declaration [-Wshadow]
   68 | void init(int n, int A[], int B[]) {
      |                                  ^
teams.cpp:63:5: note: shadowed declaration is here
   63 | int n;
      |     ^
teams.cpp:86:70: warning: conversion from '__gnu_cxx::__normal_iterator<std::pair<int, int>*, std::vector<std::pair<int, int> > >::difference_type' {aka 'long int'} to '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'} may change value [-Wconversion]
   86 |         R[i + 1] = lower_bound(begin(vec), end(vec), pii{i + 1, -1}) - begin(vec);
      |                    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~

• Subtask 1: 21 / 21
• Subtask 2: 13 / 13
• Subtask 3: 43 / 43
• Subtask 4: 23 / 23