HDU 4311 Meeting point-1(曼哈顿距离优化枚举)

HDU4311

题意

给你n个点,再n个点中找一个点,这个点到其他点的曼哈顿距离之和最小

思路

首先说一下曼哈顿距离
$$\left|x-x_i\right|+\left|y-y_i\right|$$
对于这个式子,我们可以先对所有点以$x$坐标从小到大进行排序,并求解前缀和,得到$sumx[]$数组,将以$x$排完序的点用一个$id$进行记录,然后再对$y$进行排序,同样的对$y$求解前缀和,得到$sumy[]$数组,然后我们再枚举一遍点,这个时候因为是对$y$排完序的,当我们枚举到$i$这个点时,$\left|y-y_i\right|$可以利用前缀和数组求出,分成$y$坐标比$i$大的部分和比$i$小的两部分即
$$sumy[n]-sumy[i]-(n-i)\ast a[i].y+i\ast a[i].y-sumy[i]$$
$a$为记录点的结构体数组$sumy[n]-sumy[i]-(n-i)\ast a[i].y$为比$i$大的部分$i\ast a[i].y-sumy[i]$为比$i$小的部分同样的对于$x$我们之前存了$id$所以我们可以利用$id$得到同样的式子即
$$a[i].id\ast a[i].x-sumx[a[i].id]+sumx[n]-sumx[a[i].id]-(n-a[i].id)\ast a[i].x$$
所以曼哈顿距离之和的最小值可以枚举一遍得出时间复杂度是$O(nlogn)$

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
#include <math.h>
using namespace std;
const int maxn=100005;
struct Point
{
    long long x,y,id;
}a[maxn];
long long sumx[maxn],sumy[maxn];
bool cmpx(Point a,Point b)
{
    return a.x<b.x;
}
bool cmpy(Point a,Point b)
{
    return a.y<b.y;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(sumx,0,sizeof(sumx));
        memset(sumy,0,sizeof(sumy));
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%lld%lld",&a[i].x,&a[i].y);
        }
        sort(a+1,a+n+1,cmpx);
        sumx[1]=a[1].x;
        a[1].id=1;
        for(int i=2;i<=n;i++)
        {
            sumx[i]=sumx[i-1]+a[i].x;
            a[i].id=i;
        }
        sort(a+1,a+n+1,cmpy);
        sumy[1]=a[1].y;
        for(int i=2;i<=n;i++)
            sumy[i]=sumy[i-1]+a[i].y;
        long long ans=1LL<<62;
        for(int i=1;i<=n;i++)
            ans=min(ans,sumy[n]-sumy[i]-(n-i)*a[i].y+i*a[i].y-sumy[i]+a[i].id*a[i].x-sumx[a[i].id]+sumx[n]-sumx[a[i].id]-(n-a[i].id)*a[i].x);
        printf("%lld\n",ans);
    }
    return 0;
}