Submission #73223046


Source Code Expand

#include <bits/stdc++.h>
// #include <atcoder/all> // AtCoder
using namespace std;
// using namespace atcoder; // AtCoder
using uint = unsigned int;
using ll = long long;
using ull = unsigned long long;
using ld = long double;
// using mint = modint998244353; // AtCoder
using vll = vector<long long>;
using vvll = vector<vector<long long>>;
using vvvll = vector<vector<vector<long long>>>;
using vvvvll = vector<vector<vector<vector<long long>>>>;
using dll = deque<long long>;
using ddll = deque<deque<long long>>;
using dddll = deque<deque<deque<long long>>>;
using ddddll = deque<deque<deque<deque<long long>>>>;
using pll = pair<long long, long long>;
using vpll = vector<pair<long long, long long>>;
using vvpll = vector<vector<pair<long long, long long>>>;
#define overload2(_1, _2, name, ...) name
#define overload4(_1, _2, _3, _4, name, ...) name
#define rep1(n) for (ll i = 0; i < n; i++)
#define rep2(i, n) for (ll i = 0; i < n; i++)
#define rep3(i, a, b) for (ll i = a; i < b; i++)
#define rep4(i, a, b, c) for (ll i = a; i < b; i += c)
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)
#define reps1(n) for (ll i = 1; i <= n; i++)
#define reps2(i, n) for (ll i = 1; i <= n; i++)
#define reps(...) overload2(__VA_ARGS__, reps2, reps1)(__VA_ARGS__)
#define rrep1(n) for (int i = n - 1; i >= 0; i--)
#define rrep2(i, n) for (int i = n - 1; i >= 0; i--)
#define rrep3(i, a, b) for (int i = b - 1; i >= a; i--)
#define rrep4(i, a, b, c) for (int i = b - 1; i >= a; i -= c)
#define rrep(...) overload4(__VA_ARGS__, rrep4, rrep3, rrep2, rrep1)(__VA_ARGS__)
#define rreps1(n) for (long long i = n; i >= 1; i--)
#define rreps2(i, n) for (long long i = n; i >= 1; i--)
#define rreps(...) overload2(__VA_ARGS__, rreps2, rreps1)(__VA_ARGS__)
#define all(a) (a).begin(), (a).end()
#define rall(a) (a).rbegin(), (a).rend()
#define PI acos(-1.0)
#define YES printf("YES\n"), exit(0)
#define NO printf("NO\n"), exit(0)
#define Yes printf("Yes\n"), exit(0)
#define No printf("No\n"), exit(0)
#define MINUS printf("-1\n"), exit(0)
#define cuot cout // typo
#define endk endl // typo
constexpr char ln = '\n';
constexpr long long MOD = 998244353LL;
constexpr long long LINF = 0x1fffffffffffffff; // 4倍までOK 10^18より大きい
constexpr int INF = 0x3fffffff;				   // 2倍までOK 10^9より大きい
template <class T>
void setcout(T n) { cout << setprecision(n) << fixed; }
template <class... T>
constexpr auto min(T... a) { return min(initializer_list<common_type_t<T...>>{a...}); }
template <class... T>
constexpr auto max(T... a) { return max(initializer_list<common_type_t<T...>>{a...}); }
template <class T>
bool chmax(T &a, const T &b)
{
    if (a < b)
    {
        a = b;
        return 1;
    }
    return 0;
}
template <class T>
bool chmin(T &a, const T &b)
{
    if (b < a)
    {
        a = b;
        return 1;
    }
    return 0;
}
template <class T>
long long acc(const T &a) { return accumulate(all(a), 0LL); }
template <class T>
long double accl(const T &a) { return accumulate(all(a), 0.0L); }
template <class T>
vector<T> mrui(const vector<T> &a)
{
    vector<T> ret(a.size() + 1);
    for (int i = 0; i < a.size(); i++)
    {
        ret[i + 1] = ret[i] + a[i];
    }
    return ret;
}
template <class T>
vector<vector<T>> mrui2(const vector<vector<T>> &a)
{
    vector<vector<T>> ret(a.size() + 1, vector<T>(a.front().size() + 1));
    for (int i = 0; i < (int)a.size(); i++)
    {
        for (int j = 0; j < (int)a.front().size(); j++)
        {
            ret[i + 1][j + 1] = ret[i][j + 1] + ret[i + 1][j] - ret[i][j] + a[i][j];
        }
    }
    return ret;
}
template <class T>
vector<T> kaisa(const vector<T> &a)
{
    vector<T> ret(a.size() - 1);
    for (int i = 0; i < a.size() - 1; i++)
    {
        ret[i] = a[i + 1] - a[i];
    }
    return ret;
}

inline void in() {}
template <class Head, class... Tail>
inline void in(Head &head, Tail &...tail)
{
    cin >> head;
    in(tail...);
}

#define INT(...)     \
    int __VA_ARGS__; \
    in(__VA_ARGS__)
#define LL(...)     \
    ll __VA_ARGS__; \
    in(__VA_ARGS__)
#define ULL(...)     \
    ull __VA_ARGS__; \
    in(__VA_ARGS__)
#define STR(...)        \
    string __VA_ARGS__; \
    in(__VA_ARGS__)
#define CHR(...)      \
    char __VA_ARGS__; \
    in(__VA_ARGS__)
#define DBL(...)        \
    double __VA_ARGS__; \
    in(__VA_ARGS__)
#define LD(...)     \
    ld __VA_ARGS__; \
    in(__VA_ARGS__)
#define vec(type, name, ...) vector<type> name(__VA_ARGS__)
#define VEC(type, name, size) \
    vector<type> name(size);  \
    in(name)
#define vv(type, name, h, ...) vector<vector<type>> name(h, vector<type>(__VA_ARGS__))
#define VV(type, name, h, ...)                               \
    vector<vector<type>> name(h, vector<type>(__VA_ARGS__)); \
    in(name)
#define vvv(type, name, h, w, ...) vector<vector<vector<type>>> name(h, vector<vector<type>>(w, vector<type>(__VA_ARGS__)))

template <class T1, class T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p)
{
    os << "(" << p.first << "," << p.second << ")";
    return os;
}

template <class T1, class T2>
istream &operator>>(istream &is, pair<T1, T2> &p)
{
    is >> p.first >> p.second;
    return is;
}

template <class T>
ostream &operator<<(ostream &os, const vector<T> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << (i + 1 != (int)v.size() ? " " : "");
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, const vector<vector<T>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << endl;
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, const vector<vector<vector<T>>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << "i = " << i << endl;
        os << v[i];
    }
    return os;
}

template <class T>
istream &operator>>(istream &is, vector<T> &v)
{
    for (T &in : v)
    {
        is >> in;
    }
    return is;
}

template <class T>
ostream &operator<<(ostream &os, const deque<T> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << (i + 1 != (int)v.size() ? " " : "");
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, const deque<deque<T>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << endl;
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, const deque<deque<deque<T>>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << "i = " << i << endl;
        os << v[i];
    }
    return os;
}

template <class T>
istream &operator>>(istream &is, deque<T> &v)
{
    for (T &in : v)
    {
        is >> in;
    }
    return is;
}

template <class T, class S>
ostream &operator<<(ostream &os, const map<T, S> &mp)
{
    for (auto &[key, val] : mp)
    {
        os << key << ": " << val << " ";
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, const set<T> &st)
{
    auto itr = st.begin();
    for (int i = 0; i < (int)st.size(); i++)
    {
        os << *itr << (i + 1 != (int)st.size() ? " " : "");
        itr++;
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, const multiset<T> &st)
{
    auto itr = st.begin();
    for (int i = 0; i < (int)st.size(); i++)
    {
        os << *itr << (i + 1 != (int)st.size() ? " " : "");
        itr++;
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, queue<T> q)
{
    while (q.size())
    {
        os << q.front() << " ";
        q.pop();
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, deque<T> q)
{
    while (q.size())
    {
        os << q.front() << " ";
        q.pop_front();
    }
    return os;
}

template <class T>
ostream &operator<<(ostream &os, stack<T> st)
{
    while (st.size())
    {
        os << st.top() << " ";
        st.pop();
    }
    return os;
}

template <class T, class Container, class Compare>
ostream &operator<<(ostream &os, priority_queue<T, Container, Compare> pq)
{
    while (pq.size())
    {
        os << pq.top() << " ";
        pq.pop();
    }
    return os;
}

/*ostream &operator<<(ostream &os, const mint &i) { //AtCoder
    os << i.val();
    return os;
}*/

/*ostream &operator<<(ostream &os, const vector<mint> &v) { //AtCoder
    for (int i = 0; i < (int)v.size(); i++) {
        os << v[i].val() << (i + 1 != (int)v.size() ? " " : "");
    }
    return os;
}*/

/*ostream &operator<<(ostream &os, const modint &i) { //AtCoder
    os << i.val();
    return os;
}*/

/*ostream &operator<<(ostream &os, const vector<modint> &v) { //AtCoder
    for (int i = 0; i < (int)v.size(); i++) {
        os << v[i].val() << (i + 1 != (int)v.size() ? "" : "");
    }
    return os;
}*/

int main() {
    ll n, q; cin >> n >> q;
    vll a(n); cin >> a;
    sort(all(a));
    rep(_, q) {
        ll b, k; cin >> b >> k;
        ll l = -1, r = LINF;
        while(r - l > 1) {
            ll mid = l + r >> 1;
            ll p = b - mid, q = b + mid;
            auto v = lower_bound(all(a), p);
            auto w = upper_bound(all(a), q);
            if(w - v >= k) r = mid;
            else l = mid;
        }
        cout << r << endl;
    }
}

Submission Info

Submission Time
Task G - K-th Nearest
User soshin
Language C++23 (GCC 15.2.0)
Score 425
Code Size 9558 Byte
Status AC
Exec Time 465 ms
Memory 4036 KiB

Compile Error

./Main.cpp: In function 'int main()':
./Main.cpp:362:24: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  362 |             ll mid = l + r >> 1;
      |                      ~~^~~

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 425 / 425
Status
AC × 3
AC × 21
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_random_00.txt, 01_random_01.txt, 01_random_02.txt, 01_random_03.txt, 01_random_04.txt, 01_random_05.txt, 01_random_06.txt, 01_random_07.txt, 01_random_08.txt, 01_random_09.txt, 01_random_10.txt, 01_random_11.txt, 01_random_12.txt, 01_random_13.txt, 01_random_14.txt, 01_random_15.txt, 02_minmax_00.txt, 02_minmax_01.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 1 ms 3592 KiB
00_sample_01.txt AC 1 ms 3616 KiB
00_sample_02.txt AC 1 ms 3552 KiB
01_random_00.txt AC 109 ms 3636 KiB
01_random_01.txt AC 412 ms 4036 KiB
01_random_02.txt AC 245 ms 3744 KiB
01_random_03.txt AC 413 ms 4000 KiB
01_random_04.txt AC 306 ms 3708 KiB
01_random_05.txt AC 426 ms 4032 KiB
01_random_06.txt AC 55 ms 3716 KiB
01_random_07.txt AC 414 ms 3952 KiB
01_random_08.txt AC 302 ms 3996 KiB
01_random_09.txt AC 412 ms 3888 KiB
01_random_10.txt AC 67 ms 3948 KiB
01_random_11.txt AC 419 ms 3932 KiB
01_random_12.txt AC 405 ms 3688 KiB
01_random_13.txt AC 465 ms 4004 KiB
01_random_14.txt AC 196 ms 3576 KiB
01_random_15.txt AC 287 ms 3944 KiB
02_minmax_00.txt AC 1 ms 3552 KiB
02_minmax_01.txt AC 228 ms 3888 KiB