POJ 1723 中位数

题意

传送门 POJ 1723

题解

设索引为 $i$ 的士兵最终位置为 $(x+j,y),j\in [0,N)$，目标是最小化 $$\sum _{i=0}^{N-1}|Y[i]-y|+\sum _{i=0}^{N-1}|X[i]-(x+j)|$$ 观察到横轴与纵轴的最小步数是独立的，于是可以分别求解。对于 $$\sum _{i=0}^{N-1}|Y[i]-y|$$ 答案是 $Y[i]$ 的中位数。考虑 $\sum _{i=0}^{N-1}|X[i]-(x+j)|$ 中 $i,j$ 的关系，为了使步数最小，那么满足 $X_i<X_{i^{\prime }}$ 的士兵的最终位置应该满足 $x+j<x+j^{\prime }$，那么排序 $X[i]$，目标是最小化 $$\sum _{i=0}^{N-1}|(X[i]-i)-x|$$ 答案是 $X[i]-i$ 的中位数。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxn 10005
int N, X[maxn], Y[maxn];

int main()
{
    scanf("%d", &N);
    for (int i = 0; i < N; ++i)
        scanf("%d%d", X + i, Y + i);
    sort(X, X + N);
    for (int i = 0; i < N; ++i)
        X[i] -= i;
    sort(X, X + N);
    sort(Y, Y + N);
    int m = N >> 1, x = X[m], y = Y[m];
    int res = 0;
    for (int i = 0; i < N; ++i)
        res += abs(Y[i] - y) + abs(X[i] - x);
    printf("%d\n", res);
    return 0;
}