Closest points

#include <bits/stdc++.h>

using namespace std;

long long distance(pair<long long, long long> p1, pair<long long, long long> p2) {
    return (p1.first - p2.first)*(p1.first - p2.first) + (p1.second - p2.second)*(p1.second - p2.second);
}

bool compare_y(pair<long long, long long> a, pair<long long, long long> b) {
    return a.second <= b.second;
}

int main() {

    int n;
    cin >> n;

    vector<pair<long long, long long>> points(n);
    for (int i = 0; i < n; i++)
        cin >> points[i].first >> points[i].second;

    sort(points.begin(), points.end());

    long long d = distance(points[0], points[1]);

    for (int i = 1; i < n; i++) {

        int j = i-1;
        vector<pair<long long, long long>> around;

        while (j >= 0
            && abs(points[j].first - points[i].first)*abs(points[j].first - points[i].first) <= d) {
            around.push_back(points[j]);
            j -= 1;
        }

        sort(around.begin(), around.end(), compare_y);

        int size = (int) around.size();

        pair<long long, long long> q = {0, points[i].second - d};
        auto it = lower_bound(around.begin(), around.end(), q, compare_y);
        int k = it - around.begin();

        while (
            k < size
            && (around[k].second - points[i].second)*(around[k].second - points[i].second) <= d
        ) {
            long long dist = distance(points[i], around[k]);
            if (dist < d) 
                d = dist;
            k++;
        }

    }

    cout << d << endl;

}

/*
8
105 0
100 0
2 0
-2 0
-100 0
-105 0
-1 -1000
1 1000
*/

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: TIME LIMIT EXCEEDED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: TIME LIMIT EXCEEDED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED