oj.uz

Submission #78023

# Submission time Handle Problem Language Result Execution time Memory
78023 2018-10-01T20:15:33 Z triplem5ds Sails (IOI07_sails) C++14
100 / 100
 102 ms 1772 KB 
#include "bits/stdc++.h"
using namespace std;

typedef pair<int,int> ii;
typedef long long ll;

const int N = 1e5 + 5;

/*
    IDEA :
    the answer is sum function for number of sails
    on this height meaning if there are x sails at height h
    ans += x * (x - 1) / 2;
    so the order of placement of sails is irrelevant
    so we sort them according to height
    now i know that it's optimal to place a sail first at the index with smallest value
    if there is a tie place it at the biggest one
    i can do that with line sweep and bit
    so i grab the biggest k range which is h-k+1 -> h
    color it if h==k or if the current element i am standing on is 0
    so i either binary search on the position of the last element >= query(l)
    now all of these are optimal to put in and i am sure they are less than k
    then decrease the k and do the same for k + 1 i am sure that they will end up with sum == k
*/

int bit[N];

void upd(int idx, int val){
    for(; idx < N ; idx += idx & -idx)
        bit[idx] += val;
}
int qry(int idx){
    int ret = 0;
    for(;idx; idx -= idx & -idx)
        ret += bit[idx];
    return ret;
}
ii a[N];
int getIdx(int x){
    int lo = 0, hi = N;

    while(lo < hi){

        int md = lo + (hi - lo + 1) / 2;
        if(qry(md) >= x)
            lo=md;
        else
            hi=md-1;

    }

    return lo;
}
int main(){

    int n;

    scanf("%d", &n);
    for(int i = 0; i < n; i++)
        scanf("%d%d", &a[i].first, &a[i].second);

    sort(a,a+n);

    for(int i = 0; i < n; i++){

        int h, k;
        tie(h,k) = a[i];
        int l = h-k+1;
        if(l == 1 || qry(l) != qry(l-1)){
            upd(l,1);
            upd(h+1,-1);
        }   else {
            int idx = getIdx(qry(l));
            if(qry(l)){
                upd(idx+1,1);
                upd(h+1,-1);
                k -= (h - idx);
            }
            idx = getIdx(qry(l) + 1);
            upd(idx+1,1);
            upd(idx+k+1,-1);
        }
    }

    ll ans = 0;

    for(int i = 1; i < N; i++){
        ll tmp = qry(i);
        ans += tmp * (tmp-1) / 2;
    }

    cout << ans << "\n";

    return 0;
}

Compilation message

sails.cpp: In function 'int main()':
sails.cpp:58:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
sails.cpp:60:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d", &a[i].first, &a[i].second);
         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 376 KB Output is correct
2 Correct 3 ms 380 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 440 KB Output is correct
2 Correct 3 ms 628 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 628 KB Output is correct
2 Correct 3 ms 628 KB Output is correct

Subtask #4
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 628 KB Output is correct
2 Correct 4 ms 628 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 628 KB Output is correct
2 Correct 4 ms 968 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 968 KB Output is correct
2 Correct 29 ms 968 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 25 ms 968 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 52 ms 1160 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 81 ms 1516 KB Output is correct
2 Correct 76 ms 1624 KB Output is correct

Subtask #10
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 84 ms 1696 KB Output is correct
2 Correct 68 ms 1696 KB Output is correct

Subtask #11
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 102 ms 1772 KB Output is correct
2 Correct 84 ms 1772 KB Output is correct