oj.uz

답안 #157638

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
157638 2019-10-12T16:15:09 Z ismagilov Pairs (IOI07_pairs) C++14
40 / 100
 389 ms 219600 KB 
#include<bits/stdc++.h>
using namespace std;
int tip, n, d, m;
long long ans;
 
 
struct tp
{
    int x, y, z;
};
tp vv[10002];
void solvesmall()
{
    for(int i = 1; i <= n; ++i)
    {
        cin >> vv[i].x;
        if(tip >= 2)
            cin >> vv[i].y;
        if(tip >= 3)
            cin >> vv[i].z;
    }
    for(int i = 1; i <= n; ++i)
        for(int j = i+1; j <= n; ++j)
        {
            int dd = abs(vv[i].x - vv[j].x) + abs(vv[i].y - vv[j].y) + abs(vv[i].z - vv[j].z);
            if(dd <= d)
                ++ans;
        }
    cout << ans << '\n';
}
 
int v[100002];
void solve1()
{
    for(int i = 1; i <= n; ++i)
        cin >> v[i];
    sort(v+1, v+n+1);
    for(int i = 1; i <= n; ++i)
    {
        int sm = i;
        int st = 1;
        int dr = i-1;
        while(st <= dr)
        {
            int mid = (st + dr) / 2;
            if(v[i] - v[mid] <= d)
                sm = mid, dr = mid - 1;
            else
                st = mid + 1;
        }
        ans += (i - sm);
    }
    cout << ans << '\n';
}
pair<int, int> nr[100002];
int aib[400][200002];
vector<int>coord[150002];
void upd(int bu, int pz, int val)
{
    for(; pz <= 200000; pz += (pz & (-pz)))
        aib[bu][pz] += val;
}
int compute(int bu, int pz)
{
    int ans = 0;
    for(; pz > 0; pz -= (pz & (-pz)))
        ans += aib[bu][pz];
    return ans;
}
int cb(int linie, int col)
{
    int ans = 0;
    int st = 0;
    int dr = coord[linie].size() - 1;
    while(st <= dr)
    {
        int mid = (st + dr) / 2;
        if(coord[linie][mid] <= col)
            ans = (mid + 1), st = mid + 1;
        else
            dr = mid - 1;
    }
    return ans;
}
void solve2()
{
    d = d * 2;
    int buk = 400;
    for(int i = 1; i <= n; ++i)
    {
        int x, y;
        cin >> x >> y;
        nr[i].first = x + y;
        nr[i].second = x - y + 100000;
        coord[nr[i].first].push_back(nr[i].second);
        int nrbuk = nr[i].first / buk + (nr[i].first % buk > 0);
        upd(nrbuk, nr[i].second, 1);
    }
    for(int i = 0; i <= 150000; ++i)
        sort(coord[i].begin(), coord[i].end());
    for(int i = 1; i <= n; ++i)
    {
        int st = nr[i].first - d/2;
        int sf = nr[i].first + d/2;
        int nrbuk = nr[i].first / buk + (nr[i].first % buk > 0);
        upd(nrbuk, nr[i].second, -1);
        while(st <= sf && st <= nrbuk * buk)
        {
            if(st == nr[i].first)
                --ans;
            ans += cb(st, nr[i].second + d/2) - cb(st, nr[i].second - d/2 - 1), ++st;
        }
        while(st + buk - 1 <= sf)
        {
            ++nrbuk;
            ans += compute(nrbuk, nr[i].second + d/2) - compute(nrbuk, nr[i].second - d/2 - 1);
            st += buk;
        }
        while(st <= sf)
        {
            if(st == nr[i].first)
                --ans;
            ans += cb(st, nr[i].second + d/2) - cb(st, nr[i].second - d/2 - 1), ++st;
        }
        upd(nrbuk, nr[i].second, 1);
    }
    cout << ans / 2 << '\n';
}
void solve3()
{
 
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> tip >> n >> d >> m;
    if(tip == 1)
        solve1();
    else
        if(tip == 2)
            solve2();
        else
            if(n <= 10000)
                solvesmall();
            else
                if(tip == 3)
                    solve3();
    return 0;
}

Subtask #1
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 3832 KB Output is correct
2 Correct 5 ms 3832 KB Output is correct

Subtask #2
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 3836 KB Output is correct

Subtask #3
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 25 ms 4216 KB Output is correct
2 Correct 23 ms 4216 KB Output is correct

Subtask #4
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 32 ms 4448 KB Output is correct
2 Correct 30 ms 4344 KB Output is correct

Subtask #5
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 33 ms 4216 KB Output is correct
2 Correct 35 ms 4216 KB Output is correct
3 Correct 33 ms 4316 KB Output is correct

Subtask #6
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Correct 17 ms 18652 KB Output is correct
2 Runtime error 82 ms 37240 KB Execution killed with signal 11 (could be triggered by violating memory limits)

Subtask #7
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 53 ms 10360 KB Execution killed with signal 11 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -

Subtask #8
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 67 ms 10872 KB Execution killed with signal 11 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -

Subtask #9
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 389 ms 219600 KB Execution killed with signal 11 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -

Subtask #10
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 3828 KB Output is correct
2 Correct 7 ms 3832 KB Output is correct

Subtask #11
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 5 ms 3832 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #12
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 5 ms 3832 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #13
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 5 ms 3832 KB Output isn't correct
2 Halted 0 ms 0 KB -