16210 DSHS Bank

와

수학문제였다

택시거리 x / y 나눠서 생각하고 수식 풀어서 정리한 다음 부분합 + lower_bound 사용해서 풀어야 했다. 

오늘 푼 골드1보다 더 어려웠다. 

 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <deque>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>
#include <stdio.h>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using iis = pair<int, string>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;
 
int main() {
    int n;
    scanf("%d", &n);
    
    vector<ii64> v(n);
    vector<i64> vx(n);
    vector<i64> vy(n);
    for (int i = 0; i < n; i++) {
        scanf("%lld %lld", &v[i].xx, &v[i].yy);
        vx[i] = v[i].xx;
        vy[i] = v[i].yy;
    }
    
    sort(all(vx));
    sort(all(vy));
    vector<i64> sumx(n + 1);
    vector<i64> sumy(n + 1);
    for (int i = 0; i < n; i++) {
        sumx[i + 1] = sumx[i] + vx[i];
        sumy[i + 1] = sumy[i] + vy[i];
    }
    
    i64 minv = -1;
    int idx = 0;
    for (int i = 0; i < n; i++) {
        i64 xi = lower_bound(all(vx), v[i].xx) - vx.begin();
        i64 yi = lower_bound(all(vy), v[i].yy) - vy.begin();
        
        i64 sumxx = sumx[n] - sumx[xi + 1] - sumx[xi] - v[i].xx * (n - 1 - xi) + v[i].xx * (xi);
        i64 sumyy = sumy[n] - sumy[yi + 1] - sumy[yi] - v[i].yy * (n - 1 - yi) + v[i].yy * (yi);
        
        if (i == 0)
            minv = sumxx + sumyy;
        
        if (minv > sumxx + sumyy) {
            minv = sumxx + sumyy;
            idx = i;
        }
    }
    
    printf("%d\n", idx + 1);
    
    return 0;
}