oj.uz

답안 #415785

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
415785 2021-06-01T13:58:20 Z 반딧불(#7611) Road Construction (JOI21_road_construction) C++17
100 / 100
 8740 ms 159792 KB 
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

struct Query {
    int mode; /// 0: �� �߰�, 1: ���� ����, 2: ���� ��
    ll x, y1, y2; int idx;
    Query(){}
    Query(int mode, ll x, ll y1, ll y2, int idx): mode(mode), x(x), y1(y1), y2(y2), idx(idx){}

    bool operator<(const Query &r)const{
        if(x != r.x) return x<r.x;
        return mode < r.mode;
    }
};

struct Fenwick {
    int n;
    int tree[250002];

    void init(){
        memset(tree, 0, sizeof(tree));
    }

    void add(int x, int y){
        x++;
        while(x <= n){
            tree[x] += y;
            x += x&-x;
        }
    }

    int sum(int x){
        x++;
        int ret = 0;
        while(x){
            ret += tree[x];
            x -= x&-x;
        }
        return ret;
    }

    int sum(int l, int r){
        return sum(r) - sum(l-1);
    }
} tree;

struct segTree{
    vector<pair<int, ll> > tree[1000002];

    void add(int i, int l, int r, int x, int arg1, ll arg2){
        tree[i].push_back(make_pair(arg1, arg2));
        if(l==r) return;
        int m = (l+r)>>1;
        if(x<=m) add(i*2, l, m, x, arg1, arg2);
        else add(i*2+1, m+1, r, x, arg1, arg2);
    }

    void addAns(int i, int l, int r, int s, int e, ll lim, vector<int> &tVec){
        if(r<s || e<l) return;
        if(s<=l && r<=e){
            auto it = tree[i].rbegin();
            while(it != tree[i].rend() && it->second >= lim){
                tVec.push_back(it->first);
                ++it;
            }
            return;
        }
        int m = (l+r)>>1;
        addAns(i*2, l, m, s, e, lim, tVec);
        addAns(i*2+1, m+1, r, s, e, lim, tVec);
    }
} tree2;

int n, k;
vector<ll> yVal;

pair<ll, ll> arr[250002];
int tyloc[250002];

ll totalMin;
vector<ll> ans;

int main(){
    scanf("%d %d", &n, &k);
//    n = 250000, k = 250000;
    tree.n = n;
    for(int i=1; i<=n; i++){
        scanf("%lld %lld", &arr[i].first, &arr[i].second);
//        arr[i].first = rand();
//        arr[i].second = rand();
        ll tx = arr[i].first, ty = arr[i].second;
        arr[i].first = tx + ty, arr[i].second = tx - ty;
        yVal.push_back(arr[i].second);
    }

    sort(arr+1, arr+n+1);
    sort(yVal.begin(), yVal.end());
    yVal.erase(unique(yVal.begin(), yVal.end()), yVal.end());

    for(int i=1; i<=n; i++){
        tyloc[i] = lower_bound(yVal.begin(), yVal.end(), arr[i].second) - yVal.begin();
    }

    {
        ll MIN = 0, MAX = 1e10, ANS = 1e10;
        while(MIN <= MAX){
            ll MID = (MIN + MAX) / 2;
            ll sum = -n;
            tree.init();
            vector<Query> vec;
            for(int i=1; i<=n; i++){
                int y2 = upper_bound(yVal.begin(), yVal.end(), arr[i].second + MID) - yVal.begin() - 1;
                int y1 = lower_bound(yVal.begin(), yVal.end(), arr[i].second - MID) - yVal.begin();
                vec.push_back(Query(0, arr[i].first, tyloc[i], tyloc[i], i));
                vec.push_back(Query(1, arr[i].first - MID - 1, y1, y2, i));
                vec.push_back(Query(2, arr[i].first + MID, y1, y2, i));
            }
            sort(vec.begin(), vec.end());

            for(Query query: vec){
                if(query.mode == 0) tree.add(query.y1, 1);
                else if(query.mode == 1) sum -= tree.sum(query.y1, query.y2);
                else sum += tree.sum(query.y1, query.y2);
            }

            if(sum >= k*2){
                ANS = MID;
                MAX = MID-1;
            }
            else MIN = MID+1;
        }
        totalMin = ANS;
    }

    totalMin--;
    vector<Query> vec;
    for(int i=1; i<=n; i++){
        int y2 = upper_bound(yVal.begin(), yVal.end(), arr[i].second + totalMin) - yVal.begin() - 1;
        int y1 = lower_bound(yVal.begin(), yVal.end(), arr[i].second - totalMin) - yVal.begin();
        vec.push_back(Query(0, arr[i].first, tyloc[i], tyloc[i], i));
        vec.push_back(Query(2, arr[i].first + totalMin, y1, y2, i));
    }
    sort(vec.begin(), vec.end());

    for(Query query: vec){
        if(query.mode == 0){
            tree2.add(1, 0, n, query.y1, query.idx, query.x);
        }
        else{
            vector<int> tVec;
            tree2.addAns(1, 0, n, query.y1, query.y2, arr[query.idx].first - totalMin, tVec);
            for(auto x: tVec){
                if(x <= query.idx) continue;
                ans.push_back(max(abs(arr[query.idx].first - arr[x].first), abs(arr[query.idx].second - arr[x].second)));
            }
        }
    }

    sort(ans.begin(), ans.end());
    while(ans.size() < k) ans.push_back(totalMin + 1);

    for(auto x: ans) printf("%lld\n", x);
}

Compilation message

road_construction.cpp: In function 'int main()':
road_construction.cpp:163:22: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  163 |     while(ans.size() < k) ans.push_back(totalMin + 1);
      |           ~~~~~~~~~~~^~~
road_construction.cpp:87:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   87 |     scanf("%d %d", &n, &k);
      |     ~~~~~^~~~~~~~~~~~~~~~~
road_construction.cpp:91:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   91 |         scanf("%lld %lld", &arr[i].first, &arr[i].second);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 96 ms 29660 KB Output is correct
2 Correct 94 ms 29632 KB Output is correct
3 Correct 100 ms 29756 KB Output is correct
4 Correct 96 ms 29724 KB Output is correct
5 Correct 98 ms 28536 KB Output is correct
6 Correct 31 ms 25064 KB Output is correct

Subtask #2
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5215 ms 138808 KB Output is correct
2 Correct 5112 ms 138888 KB Output is correct
3 Correct 87 ms 29628 KB Output is correct
4 Correct 5088 ms 138644 KB Output is correct
5 Correct 4824 ms 138924 KB Output is correct
6 Correct 5358 ms 138932 KB Output is correct
7 Correct 5040 ms 138260 KB Output is correct

Subtask #3
7.0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7836 ms 135668 KB Output is correct
2 Correct 7222 ms 135660 KB Output is correct
3 Correct 25 ms 24652 KB Output is correct
4 Correct 5131 ms 135660 KB Output is correct
5 Correct 4528 ms 136028 KB Output is correct

Subtask #4
20.0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7836 ms 135668 KB Output is correct
2 Correct 7222 ms 135660 KB Output is correct
3 Correct 25 ms 24652 KB Output is correct
4 Correct 5131 ms 135660 KB Output is correct
5 Correct 4528 ms 136028 KB Output is correct
6 Correct 8116 ms 135560 KB Output is correct
7 Correct 7513 ms 135660 KB Output is correct
8 Correct 21 ms 24652 KB Output is correct
9 Correct 23 ms 24644 KB Output is correct
10 Correct 7470 ms 155036 KB Output is correct
11 Correct 5110 ms 135836 KB Output is correct
12 Correct 4330 ms 136304 KB Output is correct

Subtask #5
27.0 / 27.0

# 결과 실행 시간 메모리 Grader output
1 Correct 96 ms 29660 KB Output is correct
2 Correct 94 ms 29632 KB Output is correct
3 Correct 100 ms 29756 KB Output is correct
4 Correct 96 ms 29724 KB Output is correct
5 Correct 98 ms 28536 KB Output is correct
6 Correct 31 ms 25064 KB Output is correct
7 Correct 3435 ms 75604 KB Output is correct
8 Correct 3446 ms 75532 KB Output is correct
9 Correct 83 ms 29696 KB Output is correct
10 Correct 3241 ms 72292 KB Output is correct
11 Correct 2674 ms 74120 KB Output is correct
12 Correct 1828 ms 74276 KB Output is correct
13 Correct 2012 ms 72224 KB Output is correct

Subtask #6
35.0 / 35.0

# 결과 실행 시간 메모리 Grader output
1 Correct 96 ms 29660 KB Output is correct
2 Correct 94 ms 29632 KB Output is correct
3 Correct 100 ms 29756 KB Output is correct
4 Correct 96 ms 29724 KB Output is correct
5 Correct 98 ms 28536 KB Output is correct
6 Correct 31 ms 25064 KB Output is correct
7 Correct 5215 ms 138808 KB Output is correct
8 Correct 5112 ms 138888 KB Output is correct
9 Correct 87 ms 29628 KB Output is correct
10 Correct 5088 ms 138644 KB Output is correct
11 Correct 4824 ms 138924 KB Output is correct
12 Correct 5358 ms 138932 KB Output is correct
13 Correct 5040 ms 138260 KB Output is correct
14 Correct 7836 ms 135668 KB Output is correct
15 Correct 7222 ms 135660 KB Output is correct
16 Correct 25 ms 24652 KB Output is correct
17 Correct 5131 ms 135660 KB Output is correct
18 Correct 4528 ms 136028 KB Output is correct
19 Correct 8116 ms 135560 KB Output is correct
20 Correct 7513 ms 135660 KB Output is correct
21 Correct 21 ms 24652 KB Output is correct
22 Correct 23 ms 24644 KB Output is correct
23 Correct 7470 ms 155036 KB Output is correct
24 Correct 5110 ms 135836 KB Output is correct
25 Correct 4330 ms 136304 KB Output is correct
26 Correct 3435 ms 75604 KB Output is correct
27 Correct 3446 ms 75532 KB Output is correct
28 Correct 83 ms 29696 KB Output is correct
29 Correct 3241 ms 72292 KB Output is correct
30 Correct 2674 ms 74120 KB Output is correct
31 Correct 1828 ms 74276 KB Output is correct
32 Correct 2012 ms 72224 KB Output is correct
33 Correct 7723 ms 139584 KB Output is correct
34 Correct 8740 ms 139520 KB Output is correct
35 Correct 7911 ms 159792 KB Output is correct
36 Correct 4348 ms 138348 KB Output is correct
37 Correct 4516 ms 138512 KB Output is correct
38 Correct 5126 ms 144580 KB Output is correct