Submission #230452

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
230452	2020-05-10T07:14:06 Z	VEGAnn	NLO (COCI18_nlo)	C++14	88 / 110	757 ms	65540 KB

nlo

#include <bits/stdc++.h>
#define sz(x) ((int)x.size())
#define pii pair<int,int>
#define pis pair<int,int>
#define ft first
#define sd second
#define MP make_pair
#define PB push_back
#define all(x) x.begin(),x.end()
using namespace std;
typedef long long ll;
const int oo = 2e9;
const int N = 100100;
vector<array<int, 3> > events;
set<int> st;
int n, m, k;
ll ans = 0;

ll sqr(ll x) { return (x * x); }

ll dist(int &x1, int &x2, int &y1, int y2){
    return sqr(x1 - x2) + sqr(y1 - y2);
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);

#ifdef _LOCAL
    freopen("in.txt","r",stdin);
#endif // _LOCAL

    cin >> n >> m >> k;

    for (int i = 0; i < k; i++){
        int x, y, r;
        cin >> x >> y >> r;

        int ro = max(1, x - r), lf = y, rt = y;
        ll rr = sqr(r);

        while (ro <= x){
            while (lf > 1 && dist(x, ro, y, lf - 1) <= rr)
                lf--;

            while (rt < m && dist(x, ro, y, rt + 1) <= rr)
                rt++;

            events.PB({ro, lf, i + 1});
            events.PB({ro, rt + 1, -(i + 1)});
            ro++;
        }

        ro = min(n, x + r), lf = y, rt = y;

        while (ro > x){
            while (lf > 1 && dist(x, ro, y, lf - 1) <= rr)
                lf--;

            while (rt < m && dist(x, ro, y, rt + 1) <= rr)
                rt++;

            events.PB({ro, lf, i + 1});
            events.PB({ro, rt + 1, -(i + 1)});
            ro--;
        }
    }

    sort(all(events));

    for (int i = 1, ptr = 0; i <= n; i++){
        int lst = 0;

        if (ptr == sz(events) || events[ptr][0] > i){
            ans += ll(k) * ll(m);
            continue;
        }

        int nw_ptr = ptr;

        while (nw_ptr < sz(events) && events[nw_ptr][0] == i)
            nw_ptr++;

        st.clear();
        st.insert(0);

        for (int it = ptr; it < nw_ptr; ){
            int j = it;

            ans += (ll(events[j][1]) - ll(lst) - 1) * (ll(k) - ll(*(--st.end())));

            lst = events[j][1] - 1;

            while (j < sz(events) && events[j][1] == events[it][1]){
                if (events[j][2] < 0)
                    st.erase(-events[j][2]);
                else st.insert(events[j][2]);

                j++;
            }

            it = j;
        }

        ans += (ll(m) - ll(lst)) * ll(k);

        ptr = nw_ptr;
    }

    cout << ans;

    return 0;
}

Subtask #1

88.0 / 110.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	9 ms	832 KB	Output is correct
2	Correct	25 ms	1912 KB	Output is correct
3	Correct	19 ms	2040 KB	Output is correct
4	Correct	124 ms	12780 KB	Output is correct
5	Correct	101 ms	12780 KB	Output is correct
6	Correct	757 ms	49732 KB	Output is correct
7	Correct	294 ms	25052 KB	Output is correct
8	Runtime error	106 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
9	Correct	529 ms	49708 KB	Output is correct
10	Runtime error	111 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)