距离和、二维投影到一维化简

建物流中转站

http://www.nowcoder.com/questionTerminal/c82efaf9e2cc42cda0a8ad795845eceb

"""
二维投影到一维,简化时间复杂度,O(n^2)
在X,Y维求距离和,相加即为二维的距离和,在合适的地方取最小值。
"""
import sys
if __name__ == '__main__':
    # sys.stdin = open("input.txt", "r")
    n = int(input())
    matrix = []
    for _ in range(n):
        matrix.append(list(map(int, input().strip().split())))
    x, y, space = [0] * n, [0] * n, []
    for i in range(n):
        for j in range(n):
            if matrix[i][j] == 1:
                x[i] += 1
                y[j] += 1
            else:
                space.append([i, j])
    disX, disY = [], []
    for i in range(n):
        tmpX, tmpY = 0, 0
        for j in range(n):
            tmpX += abs(j - i) * x[j]
            tmpY += abs(j - i) * y[j]
        disX.append(tmpX)
        disY.append(tmpY)
    ans = sys.maxsize
    for i, j in space:
        ans = min(ans, disX[i] + disY[j])
    print(ans if ans != sys.maxsize else -1)
全部评论
大神可以解释下abs(j - i) * x[j]这一步是什么意思吗?
点赞 回复
分享
发布于 2019-10-14 17:00

相关推荐

4 收藏 评论
分享
牛客网
牛客企业服务