oj.uz

Submission #797047

# Submission time Handle Problem Language Result Execution time Memory
797047 2023-07-29T05:18:30 Z vjudge1 Road Construction (JOI21_road_construction) C++17
0 / 100
 10000 ms 353200 KB 
#include <bits/stdc++.h>

#define fi first
#define se second

const int N = 500500;
const int mod = 1e9 + 7;

using namespace std;

int n;
int id[N];
int t[N];
pair<long long, long long> a[N];
vector<pair<long long, int>> v;

void upd(int x, int g)
{
        while(x < N) {
                t[x] += g;
                x += x & -x;
        }
}

int get_c(int x)
{
        int res = 0;
        while(x > 0) {
                res += t[x];
                x -= x & -x;
        }
        return res;
}

int get_f(int k)
{
        int res = 0;
        for (int i = 20; i >= 0; i--) {
                if (res + (1 << i) >= N) {
                        continue;
                } else if (t[res + (1 << i)] < k) {
                        res += (1 << i);
                        k -= t[res];
                }
        }
        return res + 1;
}

long long get(long long D)
{
        long long res = 0;
        for(int i = 0; i < N; i++) t[i] = 0;

        for(int i = 1, j = 1; i <= n; i++) {
                if(i > 1){
                        upd(id[i - 1], 1);
                }
                while(j < i && a[i].fi - a[j].fi > D) {
                        upd(id[j], -1);
                        j += 1;
                }
                int l = lower_bound(v.begin(), v.end(), make_pair(a[i].se - D, 0)) - v.begin();
                int r = lower_bound(v.begin(), v.end(), make_pair(a[i].se + D + 1, 0)) - v.begin() - 1;
                if(l <= r) {
                        res += get_c(v[r].se) - get_c(v[l].se - 1);
                }
        }
        return res;
}

queue<int> Q[N];

void add_res(vector<long long> &res, queue<int> q, int i)
{
        while (!q.empty()) {
                int j = q.front();
                q.pop();
                res.push_back(max(abs(a[i].fi - a[j].fi), abs(a[i].se - a[j].se)));
        }
}

vector<long long> solve(long long D)
{
        vector<long long> res;
        for(int i = 0; i < N; i++) t[i] = 0;

        for(int i = 1, j = 1; i <= n; i++) {
                if(i > 1){
                        upd(id[i - 1], 1);
                        Q[id[i - 1]].push(i - 1);
                }
                while(j < i && a[i].fi - a[j].fi > D) {
                        upd(id[j], -1);
                        Q[id[j]].pop();
                        j += 1;
                }
                int l = lower_bound(v.begin(), v.end(), make_pair(a[i].se - D, 0)) - v.begin();
                int r = lower_bound(v.begin(), v.end(), make_pair(a[i].se + D + 1, 0)) - v.begin() - 1;
                if(l <= r) {
                        int tr = get_c(v[r].se);
                        int tl = get_c(v[l].se - 1);
                        while (tl < tr) {
                                int pos = get_f(tl + 1);
                                add_res(res, Q[pos], i);
                                tl += Q[pos].size();
                        }
                }
        }
        return res;
}

int main() {
        #ifdef zxc
            freopen("input.txt", "r", stdin);
            freopen("output.txt", "w", stdout);
        #endif
        ios_base::sync_with_stdio(0);

        long long k;
        cin >> n >> k;
        for(int i = 1; i <= n; i++) {
                int x, y;
                cin >> x >> y;

                a[i] = {x + y, x - y};
        }
        sort(a + 1, a + n + 1);

        v.push_back({-1e10, 0});
        v.push_back({1e10, 0});

        for(int i = 1; i <= n; i++) {
                v.push_back({a[i].se, i});
        }
        sort(v.begin(), v.end());
        for(int i = 0, j = 0; i < v.size(); i++) {
                if(i > 0 && v[i].fi > v[i - 1].fi) {
                        j += 1;
                }
                id[v[i].se] = j;
                v[i].se = j;
        }

        long long l = 0, r = 3e9;
        while(l < r) {
                int m = (l + r) / 2;
                if(get(m + 1) < k) {
                        l = m + 1;
                } else {
                        r = m;
                }
        }

        auto res = solve(l);
        while (res.size() < k) {
                res.push_back(l + 1);
        }
        sort(res.begin(), res.end());
        for (int x: res) {
                cout << x << "\n";
        }
}

Compilation message

road_construction.cpp: In function 'int main()':
road_construction.cpp:136:33: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  136 |         for(int i = 0, j = 0; i < v.size(); i++) {
      |                               ~~^~~~~~~~~~
road_construction.cpp:155:27: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
  155 |         while (res.size() < k) {
      |                ~~~~~~~~~~~^~~

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 235 ms 343876 KB Output is correct
2 Correct 234 ms 343848 KB Output is correct
3 Execution timed out 10109 ms 339148 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #2
0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 1532 ms 353200 KB Output is correct
2 Correct 1539 ms 353148 KB Output is correct
3 Execution timed out 10081 ms 339152 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #3
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 3126 ms 348164 KB Output is correct
2 Correct 2899 ms 348068 KB Output is correct
3 Execution timed out 10077 ms 339128 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 3126 ms 348164 KB Output is correct
2 Correct 2899 ms 348068 KB Output is correct
3 Execution timed out 10077 ms 339128 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #5
0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 235 ms 343876 KB Output is correct
2 Correct 234 ms 343848 KB Output is correct
3 Execution timed out 10109 ms 339148 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #6
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 235 ms 343876 KB Output is correct
2 Correct 234 ms 343848 KB Output is correct
3 Execution timed out 10109 ms 339148 KB Time limit exceeded
4 Halted 0 ms 0 KB -