Submission #230447

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
230447	2020-05-10T07:06:31 Z	VEGAnn	NLO (COCI18_nlo)	C++14	88 / 110	431 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<pis> events[N];
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[ro].PB(MP(lf, i + 1));
            events[ro].PB(MP(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[ro].PB(MP(lf, i + 1));
            events[ro].PB(MP(rt + 1, -(i + 1)));
            ro--;
        }
    }

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

        if (!sz(events[i])){
            ans += ll(k) * ll(m);
            continue;
        }

        sort(all(events[i]));

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

        for (int it = 0; it < sz(events[i]); ){
            int j = it;

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

            lst = events[i][j].ft - 1;

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

                j++;
            }

            it = j;
        }

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

    cout << ans;

    return 0;
}

Subtask #1

88.0 / 110.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	9 ms	2944 KB	Output is correct
2	Correct	20 ms	3840 KB	Output is correct
3	Correct	14 ms	3712 KB	Output is correct
4	Correct	77 ms	9600 KB	Output is correct
5	Correct	61 ms	8320 KB	Output is correct
6	Correct	431 ms	47736 KB	Output is correct
7	Correct	162 ms	20984 KB	Output is correct
8	Runtime error	193 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
9	Correct	294 ms	35192 KB	Output is correct
10	Runtime error	173 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)