Submission #1354043

#TimeUsernameProblemLanguageResultExecution timeMemory
1354043cnam9Pairs (IOI07_pairs)C++20
100 / 100
477 ms8188 KiB
#include <iostream>
#include <algorithm>

using namespace std;

constexpr int N = 1e5 + 5;

namespace subtask1 {
    int points[N];

    void solve(int n, int D) {
        for (int i = 0; i < n; i++) {
            cin >> points[i];
        }
        sort(points, points + n);

        long long res = 0;
        int j = 0;
        for (int i = 0; i < n; i++) {
            while (points[i] - points[j] > D) j++;
            res += i - j;
        }
        cout << res;
    }
}

namespace subtask2 {
    struct Point {
        int x, y;

        auto operator<=>(const Point &) const = default;
    } points[N];

    constexpr int M = 75000 + 5;
    int m;
    int tree[M * 2];

    template <int w>
    void update(int i) {
        for (; i <= m; i += i & -i) {
            tree[i] += w;
        }
    }

    int query(int i) {
        int s = 0;
        for (; i; i &= i - 1) {
            s += tree[i];
        }
        return s;
    }

    void solve(int n, int D, int magnitude) {
        m = magnitude * 2 - 1;

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

            points[i] = {x + y - 1, x - y + magnitude};
        }

        sort(points, points + n);

        long long res = 0;
        int j = 0;
        for (int i = 0; i < n; i++) {
            auto [x, y] = points[i];
            while (x - points[j].x > D) {
                update<-1>(points[j++].y);
            }
            res += query(min(m, y + D)) - query(max(y - D - 1, 0));
            update<1>(y);
        }
        cout << res;
    }
}

namespace subtask3 {
    constexpr int M = 75;

    int m;
    struct {
        int tree[2 * M][2 * M];

        template <int w>
        void update(int x, int y) {
            for (int i = x; i <= m; i += i & -i) {
                for (int j = y; j <= m; j += j & -j)
                    tree[i][j] += w;
            }
        }

        int query(int x, int y) {
            int s = 0;
            for (int i = x; i; i &= i - 1) {
                for (int j = y; j; j &= j - 1)
                    s += tree[i][j];
            }
            return s;
        }
    } trees[2 * M];

    struct Point {
        int x, y, z;

        auto operator<=>(const Point &) const = default;
    } points[N];

    void solve(int n, int D, int magnitude) {
        m = magnitude * 2 - 1;

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

            points[i] = {z, x + y - 1, x - y + magnitude};
        }

        sort(points, points + n);

        long long res = 0;
        for (int i = 0; i < n; i++) {
            auto [x, y, z] = points[i];

            for (int xp = 1; xp <= m; xp++) {
                int d = D - abs(xp - x);
                if (d < 0) continue;

                int yl = max(0, y - d - 1);
                int yr = min(m, y + d);
                int zl = max(0, z - d - 1);
                int zr = min(m, z + d);
                res += trees[xp].query(yr, zr) - trees[xp].query(yr, zl) - trees[xp].query(yl, zr) + trees[xp].query(yl, zl);
            }
            trees[x].update<1>(y, z);
        }
        cout << res;
    }
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    // freopen("input.txt", "r", stdin);

    int dimension, n, D, magnitude;
    cin >> dimension >> n >> D >> magnitude;

    if (dimension == 1) {
        subtask1::solve(n, D);
        return 0;
    }
    if (dimension == 2) {
        subtask2::solve(n, D, magnitude);
        return 0;
    }
    subtask3::solve(n, D, magnitude);
    return 0;
}
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...