제출 #1186024

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1186024		Gray	Road Construction (JOI21_road_construction)	C++20	0 / 100	10147 ms	1842568 KiB

road_construction

#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{
    map<ll, map<ll, ll>> tr;
    ll offs, n;
    Fenwick(ll N){
        n=N*4+20; offs=N*2+10;
    }
    void add(ll _i, ll _j, ll x){
        _i+=offs; _j+=offs;
        for (ll i=_i; i<=n; i+=(-i&i)){
            for (ll j=_j; j<=n; j+=(-j&j)){
                tr[i][j]+=x;
            }
        }
    }
    ll get(ll _i, ll _j){
        ll sum=0; _i+=offs; _j+=offs;
        for (ll i=_i; i; i-=(-i&i)){
            for (ll j=_j; j; j-=(-j&j)){
                if (tr.count(i) and tr[i].count(j)){
                    sum+=tr[i][j];
                }
            }
        }
        return sum;
    }
};

struct FenwickSick{
    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 solve(){
    ll n, k; cin >> n >> k;
    Fenwick tr(2e9);
    vector<pair<ll, ll>> pts(n);
    vector<pair<ll, ll>> dpts(n);
    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};
        tr.add(dpts[i].ff, dpts[i].ss, 1);
    }
    ll l=0, r=2e9+1, dcnt=0;
    while(l+1<r){
        ll mid = (l+r)/2, cnt=0;
        for (ll i=0; i<n; i++){
            ll fi = dpts[i].ff-mid-1, fj=dpts[i].ss-mid-1, ti = dpts[i].ff+mid, tj = dpts[i].ss+mid;
            cnt+=tr.get(ti, tj)-tr.get(ti, fj)-tr.get(fi, tj)+tr.get(fi, fj)-1;
        }
        if (cnt/2<=k) {
            l=mid; dcnt=cnt/2;
        } else r=mid;
    }
    vector<ll> res;
    for (; dcnt<k; dcnt++) 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(1e9), trb(1e9);
    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