2018网易校招堆棋子(python)

题目描述

小易将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)中

输入

4
1 2 4 9
1 1 1 1

输出

0 1 3 10

重点:

要解出这道题目,必须要提前知道一个数学前提:放置棋子的格子的横坐标来自所有棋子的横坐标,纵坐标来自所有其次的纵坐标,也就是说这道题目有n*n个可能的点,我们要做的就是遍历这n*n个点,找出最优解

代码:

n = int(input().strip())
xlist = list(map(int,input().strip().split()))
ylist = list(map(int,input().strip().split()))

ans=[float("inf") for _ in range(n)]
for x in range(n):
    for y in range(n):
        sum = 0
        res = []
        for i in range(n):
            res.append(abs(xlist[x] - xlist[i])+abs(ylist[y] - ylist[i]))
        res.sort()
        for i in range(n):
            sum += res[i]
            ans[i] = min(ans[i],sum)
print(" ".join(map(str,ans)))
 

猜你喜欢

转载自blog.csdn.net/qq_37676370/article/details/88774188