Submission #1186087

#	Time	Username	Problem	Language	Result	Execution time	Memory
1186087		Gray	Road Construction (JOI21_road_construction)	C++20	0 / 100	10092 ms	88988 KiB

road_construction

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

#define ll long long
#define ull unsigned long long
#define ld long double
#define ff first
#define ss second
#define ln "\n"
#define mp make_pair
#define pb push_back
#define INF (ll)2e18
#define MOD (ll)(1e9+7)

struct Fenwick{
    unordered_map<int, int> tr;
    ll offs, n;
    Fenwick(ll N){
        n=N*4+20; offs=2*N+10;
    }
    void add(ll _i, ll x){
        _i+=offs; for (ll i=_i; i<=n; i+=(-i&i)){
            tr[i]+=x;
        }
    }
    ll get(ll _i){
        ll sum=0; _i+=offs;
        for (ll i=_i; i; i-=(-i&i)){
            if (tr.count(i)){
                sum+=tr[i];
            }
        }
        return sum;
    }
    void clear(){
        tr.clear();
    }
};

struct FenwickSick{
    unordered_map<ll, multiset<ll>> tr;
    ll n, offs;
    FenwickSick(ll N){
        n=N*4+20; offs=N*2+10;
    }
    void add(ll i, ll x){
        i+=offs; for (; i<=n; i+=(-i&i)){
            tr[i].insert(x);
        }
    }
    void collect(ll i, ll lim, vector<ll> &col){
        i+=offs; for (; i; i-=(-i&i)){
            if (tr[i].empty()) continue;
            auto iter = tr[i].upper_bound(lim);
            if (iter!=tr[i].begin()){
                do{
                    iter--;
                    col.push_back(*iter);
                }while (iter!=tr[i].begin());
            }
        }
    }
    void clear(){
        tr.clear();
    }
};

void solve(){
    ll n, k; cin >> n >> k;
    vector<pair<ll, ll>> pts(n);
    vector<pair<ll, ll>> dpts(n);
    // vector<ll> dxs, dys;
    for (ll i=0; i<n; i++){
        cin >> pts[i].ff >> pts[i].ss;
        dpts[i] = {pts[i].ff+pts[i].ss, pts[i].ff-pts[i].ss};
        // dxs.push_back(dpts[i].ff); dys.push_back(dpts[i].ss);
    }
    // sort(dxs.begin(), dxs.end()); dxs.erase(unique(dxs.begin(), dxs.end()), dxs.end());
    // sort(dys.begin(), dys.end()); dys.erase(unique(dys.begin(), dys.end()), dys.end());
    sort(dpts.begin(), dpts.end());
    ll l=0, r=1e10, dcnt=0;
    Fenwick tr(1e10);
    while(l+1<r){
        ll mid = (l+r)/2, cnt=0;
        ll lo=0, hi=-1; tr.clear();
        for (ll i=0; i<n; i++){
            while (dpts[lo].ff<dpts[i].ff-mid) {
                tr.add(dpts[lo].ss, -1); lo++;
            }
            while (hi+1<n and dpts[hi+1].ff<=dpts[i].ff+mid){
                hi++; tr.add(dpts[hi].ss, 1);
            }
            cnt+=tr.get(dpts[i].ss+mid)-tr.get(dpts[i].ss-mid-1)-1;
        }
        if (cnt/2<=k) {
            l=mid; dcnt=cnt/2;
        } else r=mid;
    }
    // cout << l << " " << dcnt << ln;
    // return;
    vector<ll> res;
    for (ll i=dcnt; i<k; i++) res.push_back(l+1);
    // for (ll i=0; i<n; i++){
    //     for (ll j=i+1; j<n; j++){
    //         if (abs(pts[i].ff-pts[j].ff)+abs(pts[i].ss-pts[j].ss)<=l) {
    //             res.push_back(abs(pts[i].ff-pts[j].ff)+abs(pts[i].ss-pts[j].ss));
    //         }
    //     }
    // }
    // sort(res.begin(), res.end());
    // for (auto ch:res) cout << ch << ln;
    // return;
    map<ll, vector<ll>> sweep;
    for (auto [x, y]:pts){
        sweep[x].push_back(y);
    }
    FenwickSick trs(1e10), trb(1e10);
    vector<ll> add;
    for (auto &[x, ys]:sweep){

        sort(ys.begin(), ys.end());
        for (auto y:ys){
            ll val = l-x-y; add.clear();
            trs.collect(y, val, add);
            for (auto ch:add) res.push_back(ch+x+y);
            val = l-x+y; add.clear();
            trb.collect(-y-1, val, add);
            for (auto ch:add) res.push_back(ch+x-y);
        }
        for (ll i=0; i<(ll)ys.size(); i++){
            for (ll j=i-1; j>=0 and ys[i]-ys[j]<=l; j--){
                res.push_back(ys[i]-ys[j]);
            }
        }
        for (auto y:ys){
            trs.add(y, -x-y);
            trb.add(-y, -x+y);
        }
    }
    sort(res.begin(), res.end());
    for (auto ch:res) cout << ch << ln;
}

int main(){
    ios_base::sync_with_stdio(false); cin.tie(nullptr);
    #ifdef LOCAL
    auto start = chrono::high_resolution_clock::now();
    #endif
    ll t=1;
    // cin >> t;
    for (ll __c=1; __c<=t; __c++) solve();
    #ifdef LOCAL
    auto duration = chrono::duration_cast<chrono::microseconds>(chrono::high_resolution_clock::now() - start);
    cout << setprecision(0) << fixed << "time: " << (double)duration.count()/1000.0 << " milliseconds" << endl;
    #endif
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 6
• Subtask 3: 0 / 7
• Subtask 4: 0 / 20
• Subtask 5: 0 / 27
• Subtask 6: 0 / 35