#include <bits/stdc++.h>
#define IO_OP std::ios::sync_with_stdio(0); std::cin.tie(0);
#define F first
#define S second
#define V vector
#define PB push_back
#define MP make_pair
#define EB emplace_back
#define ALL(v) (v).begin(), (v).end()
#define debug(x) cerr << "Line(" << __LINE__ << ") -> " << #x << " is " << x << endl
using namespace std;
typedef long long ll;
typedef pair<int, int> pi;
typedef V<int> vi;
const int INF = 1e9 + 7;
int n, d, m;
struct solve1 {
solve1() {
vi a(n);
for(int i = 0; i < n; i++) {
cin >> a[i];
}
sort(ALL(a));
ll ans = 0;
for(int i = 0, j = 0; i < n; i++) {
while(a[i] - a[j] > d) j++;
assert(a[i] - a[j] <= d);
ans += i - j;
}
cout << ans << '\n';
}
};
struct solve2 {
static const int N = 150000 + 7;
int bit[N];
void add(int y, int val) {
for(; y < N; y += y & -y)
bit[y] += val;
}
int qry(int y) {
int res = 0;
for(; y; y -= y & -y)
res += bit[y];
return res;
}
solve2() {
memset(bit, 0, sizeof bit);
V<pi> a(n);
for(int i = 0; i < n; i++) {
int x, y;
cin >> x >> y;
a[i] = {x - y, x + y};
}
V<array<int, 3>> ev;
ll ans = 0;
for(int i = 0; i < n; i++) {
ev.PB({a[i].F, 0, a[i].S});
ev.PB({a[i].F + d, 1, a[i].S});
}
sort(ALL(ev));
deque<pi> dk;
for(auto e:ev) {
if(e[1] == 0) { // add point
add(e[2], 1);
dk.EB(e[0], e[2]);
} else { // qry
while(dk.size() && dk[0].F < e[0] - 2 * d) {
add(dk[0].S, -1);
dk.pop_front();
}
int l = max(e[2] - d, 1), r = min(e[2] + d, N - 1);
ans += qry(r) - qry(l - 1) - 1;
}
}
cout << ans / 2 << '\n';
}
};
struct solve3 {
solve3() {
}
};
signed main()
{
IO_OP;
int b;
cin >> b >> n >> d >> m;
if(b == 1) solve1();
else if(b == 2) solve2();
else solve3();
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
364 KB |
Output is correct |
2 |
Correct |
1 ms |
364 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
14 ms |
748 KB |
Output is correct |
2 |
Correct |
14 ms |
748 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
20 ms |
768 KB |
Output is correct |
2 |
Correct |
20 ms |
748 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
20 ms |
768 KB |
Output is correct |
2 |
Correct |
20 ms |
876 KB |
Output is correct |
3 |
Correct |
19 ms |
748 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1004 KB |
Output is correct |
2 |
Correct |
2 ms |
1004 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
68 ms |
4960 KB |
Output is correct |
2 |
Correct |
61 ms |
4908 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
63 ms |
4960 KB |
Output is correct |
2 |
Correct |
59 ms |
4960 KB |
Output is correct |
3 |
Correct |
61 ms |
4960 KB |
Output is correct |
4 |
Correct |
59 ms |
4960 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
59 ms |
4960 KB |
Output is correct |
2 |
Correct |
72 ms |
4960 KB |
Output is correct |
3 |
Correct |
61 ms |
4960 KB |
Output is correct |
4 |
Correct |
63 ms |
4960 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |