Submission #964638

#	Time	Username	Problem	Language	Result	Execution time	Memory
964638		Esgeer	Pairs (IOI07_pairs)	C++17	60 / 100	3932 ms	16660 KiB

pairs

#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <unistd.h>
using namespace __gnu_pbds;
template <class T>
using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
#ifndef Local
    #pragma GCC optimize("O3,unroll-loops")
#endif
#define int long long
#define vi vector<int>
#define vvi vector<vi>
#define pii pair<int, int>
#define vpi vector<pii>
#define vvpi vector<vpi>
#define vb vector<bool>
#define vvb vector<vb>
#define endl '\n'
#define sp << " " <<
#define F(i, s, n) for(int i = s; i < (int) n; i++)
#define pb push_back
#define fi first
#define se second

int mod = 1e9+7;
int inf = 1e6;
const int N = 3e5;

int t[N];

template<typename T> struct bit2d {
    vector<T> bit;
    vi xs;
    vi st;
    vi cnt;
    int n, alllen;
    void init(int N, vpi Q) {
        n = N;
        cnt.assign(N+1, 0);
        st.resize(N+1);
        sort(Q.begin(), Q.end(), [](pii &a, pii &b) { return a.se > b.se; });
        F(i, 0, Q.size()) {
            for(int j = Q[i].fi; j > 0 && j < N; j += (j & -j)) cnt[j]++;
        }
        st[0] = cnt[0]+1;
        F(i, 1, N) {
            st[i] = st[i-1] + cnt[i];
        }
        alllen = st[N-1];
        xs.assign(alllen+3, 0);
        bit.assign(alllen+3, 0);
        F(i, 0, Q.size()) {
            for(int j = Q[i].fi; j > 0 && j < N; j += (j & -j)) xs[--st[j]] = Q[i].se;
        }
    }

    int find(int y, int el) {
        return upper_bound(xs.begin() + st[y], xs.begin() + st[y] + cnt[y], el) - xs.begin() - st[y];
    }

    void update(int y, int x, T el) {
        while(y > 0 && y < n) {
            //cout << y sp x sp find(y, x) sp cnt[y] << endl;
            for(int i = find(y, x); i > 0 && i <= cnt[y]; i += (i & -i)) {
                //cout << y sp i << endl;
                bit[i + st[y]] += el;
            }
            y += (y & -y);
        }
    }

    T query(int y, int x) {
        T res = T();
        while(y > 0 && y < n) {
            //cout << y sp x sp find(y, x) sp cnt[y] << endl;
            for(int i = find(y, x); i > 0 && i <= cnt[y]; i -= (i & -i)) {
                //cout << y sp i << endl;
                res += bit[i + st[y]];
            }
            y -= (y & -y);
        }
        return res;
    }
};

void solve() {
    int d, n, k, m;
    cin >> d >> n >> k >> m;
    if(d == 1) {
        vi a(n);
        F(i, 0, n) cin >> a[i];
        sort(a.begin(), a.end());
        queue<int> last;
        int ans = 0;
        F(i, 0, n) {
            while(last.size() && last.front() < a[i] - k) last.pop();
            ans += last.size();
            last.push(a[i]);
        }
        cout << ans << endl;
    } else if(d == 2) {
        vpi a(n);
        F(i, 0, n) cin >> a[i].fi >> a[i].se;
        F(i, 0, n) {
            int y = a[i].fi, x = a[i].se;
            a[i].fi = x - y;
            a[i].se = x + y;
        }
        sort(a.begin(), a.end());
        Tree<pii> st;
        int ans = 0;
        queue<int> q;
        F(i, 0, n) {
            while(q.size() && a[q.front()].fi < a[i].fi - k) {
                st.erase({a[q.front()].se, q.front()});
                q.pop();
            }
            ans += st.order_of_key({a[i].se + k, inf}) - st.order_of_key({a[i].se - k, -1});
            st.insert({a[i].se, i});
            q.push(i);
        }
        cout << ans << endl;
    } else if(d == 3) {
        vector<pair<int, pii>> a(n);
        F(i, 0, n) {
            cin >> a[i].se.fi >> a[i].se.se >> a[i].fi;
            int y = a[i].se.fi, x = a[i].se.se;
            a[i].se.fi = x + y;
            a[i].se.se = x - y + m;
        }
        sort(a.begin(), a.end());
        vpi Q[305];
        F(i, 0, n) Q[a[i].fi].pb({a[i].se.fi, a[i].se.se});
        bit2d<int> bit[305];
        F(i, 0, 305) bit[i].init(4*m, Q[i]);
        int ans = 0;
        queue<int> q;
        F(i, 0, n) {
            //while(q.size() && a[q.front()].fi < a[i].fi - )
            F(z, 0, 305) {
                int nk = k - abs(z - a[i].fi);
                if(nk < 0) continue;
                ans += bit[z].query(a[i].se.fi + nk, a[i].se.se + nk) - bit[z].query(a[i].se.fi + nk, a[i].se.se - nk - 1) - bit[z].query(a[i].se.fi - nk - 1, a[i].se.se + nk) + bit[z].query(a[i].se.fi - nk - 1, a[i].se.se - nk - 1);
            }
            q.push(i);
            bit[a[i].fi].update(a[i].se.fi, a[i].se.se, 1);
        }
        cout << ans << endl;
    }
}

void setIO() {
    ios_base::sync_with_stdio(0); cin.tie(0);
    #ifdef Local
        freopen("local.in", "r", stdin);
        freopen("local.out", "w", stdout);
    #else 
        // freopen("friendcross.in","r",stdin);
        // freopen("friendcross.out","w",stdout);
    #endif
}

signed main() {
    setIO();
    int t = 1;
    //cin >> t;
    while(t--) solve();
}

• Subtask 1: 6 / 6
• Subtask 2: 6 / 6
• Subtask 3: 6 / 6
• Subtask 4: 6 / 6
• Subtask 5: 6 / 6
• Subtask 6: 7 / 7
• Subtask 7: 7 / 7
• Subtask 8: 8 / 8
• Subtask 9: 8 / 8
• Subtask 10: 0 / 10
• Subtask 11: 0 / 10
• Subtask 12: 0 / 10
• Subtask 13: 0 / 10