堆棋子(网易2018校招)

题目描述

小易将n个棋子摆放在一张无限大的棋盘上。第i个棋子放在第x[i]行y[i]列。同一个格子允许放置多个棋子。每一次操作小易可以把一个棋子拿起并将其移动到原格子的上、下、左、右的任意一个格子中。小易想知道要让棋盘上出现有一个格子中至少有i(1 ≤ i ≤ n)个棋子所需要的最少操作次数.

输入描述

输入包括三行,第一行一个整数n(1 ≤ n ≤ 50),表示棋子的个数
第二行为n个棋子的横坐标x[i](1 ≤ x[i] ≤ 10^9)
第三行为n个棋子的纵坐标y[i](1 ≤ y[i] ≤ 10^9)

输出描述

输出n个整数,第i个表示棋盘上有一个格子至少有i个棋子所需要的操作数,以空格分割。行末无空格

如样例所示:
对于1个棋子: 不需要操作
对于2个棋子: 将前两个棋子放在(1, 1)中
对于3个棋子: 将前三个棋子放在(2, 1)中
对于4个棋子: 将所有棋子都放在(3, 1)中

。

示例1

输入：

4
1 2 4 9
1 1 1 1

输出：

0 1 3 10

算法分析

使dist_sum最小的格子的x坐标一定是X中的一个，y坐标一定是Y中的一个。

 

提交代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include<limits.h>

using namespace std;
vector<int> X, Y;
int n;

int get_dist_sum(int x, int y, int k)
{
    // 求出所有棋子到xy的距离
    vector<int> dist(n, 0);
    for (int i = 0; i < n; ++i)
        dist[i] = abs(X[i]-x) + abs(Y[i]-y);
    sort(dist.begin(), dist.end());
    
    int dist_sum = accumulate(dist.begin(), dist.begin()+k, 0);
    return dist_sum;
}

int min_moves(int k)
{
    int min_dist_sum = INT_MAX;
    // 最小操作数必定出现在X与Y中的组合中
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            // 求出使点X[i] Y[j]有k个棋子的的最小距离
            int dist_sum = get_dist_sum(X[i], Y[j], k);
            min_dist_sum = min(min_dist_sum, dist_sum);
        }
    }
    
    return min_dist_sum;
}

vector<int> min_move()
{
    vector<int> res(n, 0);
    // 一个格子有k个棋子
    for (int i = 2; i <= n; ++i)
        res[i-1] = min_moves(i);
    return res;
}

int main()
{
    cin >> n;
    X = vector<int>(n, 0);
    Y = vector<int>(n, 0);
    for (int i = 0; i < n; ++i) cin >> X[i];
    for (int i = 0; i < n; ++i) cin >> Y[i];
    vector<int> ans = min_move();
    for (int i = 0; i < n - 1; ++i) cout << ans[i] << " ";
    cout << ans.back();
    return 0;
}