Submission #1284933

#TimeUsernameProblemLanguageResultExecution timeMemory
1284933tntPairs (IOI07_pairs)C++20
47 / 100
4096 ms3516 KiB
#include <bits/stdc++.h>
using namespace std;

#define pb push_back                    
#define ll long long 
#define s second
#define f first
#define int long long
#define all(v) v.begin(),v.end()
const long long inf = 2e9 + 7;
const int N = 1e5 + 10;
void solve(){
    int b,n,m,d;
    cin >> b >> n >> d >> m;
    int ans = 0;
    if(b == 1){
        vector <int> v(n + 1);
        for(int i = 1; i <= n; i++){
            cin >> v[i];
        }
        sort(v.begin() + 1,v.end());
        for(int i = 1; i <= n; i++){
            int it = upper_bound(v.begin(),v.end(),v[i] + d) - v.begin() - 1 - i;
            ans += it;
        }
    }
    else if(b == 2){
        vector <array<int,2>> v;
        for(int i = 1; i <= n; i++){
            int x,y;
            cin >> x >> y;
            v.pb({x,y});
        }
        for(int i = 0; i < n; i++){
            for(int j = 0; j < i; j++){
                int sum = 0;
                for(int k = 0; k < 2; k++){
                    sum += abs(v[j][k] - v[i][k]);
                }
                if(sum <= d) ans++;
            }
        }
    }
    else{
        vector <array<int,3>> v;
        for(int i = 1; i <= n; i++){
            int x,y,z;
            cin >> x >> y >> z;
            v.pb({x,y,z});
        }
        for(int i = 0; i < n; i++){
            for(int j = 0; j < i; j++){
                int sum = 0;
                for(int k = 0; k < 3; k++){
                    sum += abs(v[j][k] - v[i][k]);
                }
                if(sum <= d) ans++;
            }
        }
    }
    cout << ans;
}
signed main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    //freopen("promote.in", "r", stdin);
    //freopen("promote.out", "w", stdout);
    int t1 = 1;
    while(t1--){
     	solve();
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...