oj.uz

Submission #786489

# Submission time Handle Problem Language Result Execution time Memory
786489 2023-07-18T08:21:18 Z danikoynov Pairs (IOI07_pairs) C++14
60 / 100
 55 ms 6356 KB 
#include<bits/stdc++.h>
#define endl '\n'

using namespace std;
typedef long long ll;

void speed()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
}

const int maxn = 1e5 + 10;

struct point
{
    int x, y, z;

    point(int _x = 0, int _y = 0, int _z = 0)
    {
        x = _x;
        y = _y;
        z = _z;
    }
} p[maxn], s[maxn];

bool cmp(point p1, point p2)
{
    if (p1.x != p2.x)
        return p1.x < p2.x;
    if (p1.y != p2.y)
        return p1.y < p2.y;
    return p1.z < p2.z;
}
const int max_size = 150010;
struct fenwick_1d
{

    int fen[max_size];

    fenwick_1d()
    {
        memset(fen, 0, sizeof(fen));
    }
    void update(int pos, int val)
    {
        for (int i = pos; i < max_size; i += (i & -i))
            fen[i] += val;
    }

    int query(int pos)
    {
        if (pos >= max_size)
            pos = max_size - 1;
        if (pos < 0)
            pos = 0;
            ///cout << pos << endl;
        int sum = 0;
        for (int i = pos; i > 0; i -= (i & -i))
            sum += fen[i]; ///cout << "step " << i << " " << fen[i] << endl;
            ///cout << "value " << sum << endl;
        return sum;
    }
};
int b, n, d, m;
void solve()
{
    cin >> b >> n >> d >> m;
    for (int i = 1; i <= n; i ++)
    {
        cin >> p[i].x;
        if (b >= 2)
            cin >> p[i].y;
        if (b == 3)
            cin >> p[i].z;
    }

    if (b == 1)
    {
        sort(p + 1, p + n + 1, cmp);
        int pt = 1;
        ll ans = 0;
        for (int i = 1; i <= n; i ++)
        {
            while(p[i].x - p[pt].x > d)
                pt ++;
            ///cout << i << " : " << p[i].x << " " << pt << endl;
            ans += (ll)(i - pt);
        }
        cout << ans << endl;
    }
    else if (b == 2)
    {
        for (int i = 1; i <= n; i ++)
        {
            s[i].x = p[i].x - p[i].y;
            s[i].y = p[i].x + p[i].y;
        }

        ll ans = 0;
        /**for (int i = 1; i <= n; i ++)
            for (int j = i + 1; j <= n; j ++)
            if (abs(s[i].x - s[j].x) <= d && abs(s[i].y - s[j].y) <= d)
                ans ++;*/

        fenwick_1d tree;
        sort(s + 1, s + n + 1, cmp);
        int pt = 1;
        for (int i = 1; i <= n; i ++)
        {
            tree.update(s[i].y, 1);
            while(s[i].x - s[pt].x > d)
            {
                tree.update(s[pt].y, -1);
                pt ++;
            }

            ll len = tree.query(s[i].y + d) - tree.query(s[i].y - d - 1);

            ans = ans + (len - 1);
        }
        cout << ans << endl;
    }
    else

    {
        assert(false);
    }
}

int main()
{
    solve();
    return 0;
}

Subtask #1
6.0 / 6.0

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

Subtask #2
6.0 / 6.0

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

Subtask #3
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 3156 KB Output is correct
2 Correct 27 ms 3232 KB Output is correct

Subtask #4
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 36 ms 3232 KB Output is correct
2 Correct 39 ms 3208 KB Output is correct

Subtask #5
6.0 / 6.0

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

Subtask #6
7.0 / 7.0

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

Subtask #7
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 42 ms 3212 KB Output is correct
2 Correct 39 ms 3128 KB Output is correct

Subtask #8
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 46 ms 3156 KB Output is correct
2 Correct 46 ms 3212 KB Output is correct
3 Correct 54 ms 3156 KB Output is correct
4 Correct 47 ms 3156 KB Output is correct

Subtask #9
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 55 ms 3156 KB Output is correct
2 Correct 54 ms 3208 KB Output is correct
3 Correct 55 ms 4348 KB Output is correct
4 Correct 55 ms 4324 KB Output is correct

Subtask #10
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 6356 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #11
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 39 ms 6288 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #12
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 44 ms 6284 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #13
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 40 ms 6284 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -