2019-08-11 拼多多笔试

第二题:

多多鸡有一串长度为L的珍珠项链,上面有N颗珍珠,分布在0到L-1这些位置上。现在多多鸡想把所有的珍珠移动到一起,并且想让所有珍珠移动的距离总和尽可能小。所有的珍珠可以看作在一个环上,珍珠可以向相邻没有珍珠的位置移动。请给出最优的方案下,所有珍珠移动的距离的总和。

笔试的时候没有做出来。这个是后来想的,不一定对。

# L = 10
# N = 5
# data = [1,3,5,7,9]

L = 1000
N = 4
data = [1,4,998,995]

data.sort()

flag = [1 if i in data else 0 for i in range(L)]
print(flag)
L= L-1
min_distance = L * N

for i in range(L): # i作为新珍珠项链的中心,其他珍珠都向i移动,放在i的左侧或者右侧
    distance = 0
    distance1 = 0
    distance2 = 0
    for num in data:  # num表示旧项链上的每个珍珠的位置

        if flag[i] == 0: #因为i是空位,所以有两种情况,一种是移动到i左侧的珍珠占用位置i,一种是移动到右侧的珍珠占用位置i。
            # (1)左侧占用位置i
            if num > i:
                d_left1 = (L - num) + 1 + i - sum(flag[num+1:]) - sum(flag[:i])
                # 如果num放在i的左侧,并且num大于i,那么移动的距离等于num先移动到旧项链尾部的距离,加上穿越尾部到旧项链头部后从头部移动
                # 到i处的距离,减去num和i之间的在num右侧和i左侧的其他珍珠的个数。因为左侧占用位置i,所以计算距离时要包括i

                d_right1 = (num - (i+1)) - sum(flag[i + 1:num])
                # 如果num放在i的右侧,并且num大于i,那么移动的距离等于num直接向左移动到i的距离,加上num左侧和i的右侧之间其他珍珠的个数。
            else:
                d_left1 = (i - num) - sum(flag[num+1:i])
                # 如果num放在i的左侧,并且num小于i,那么移动的距离等于num直接向右移动到i的距离,加上num右侧和i的左侧之间的其他珍珠的个数。

                d_right1 = num + 1 + (L - (i+1)) - sum(flag[:num]) - sum(flag[i+1:])
                # 如果num放在i的右侧,并且num小于i,那么移动的距离等于num先向左移动到旧项链头部的距离,再加上穿越到头部到旧项链尾部后从尾
                # 部向左移动到i的距离,减去在i右侧和num的左侧的其他珍珠的个数
            print("i=",i,'num=',num,'d_right1=',d_right1, 'd_left1=',d_left1, 'distance=(',distance1, distance2, ')')

            # (2) 右侧占用位置i
            if num > i:
                d_left2 = (L - num) + i - sum(flag[num+1:]) - sum(flag[:i])
                # 如果num放在i的左侧,并且num大于i,那么移动的距离等于num先移动到旧项链尾部的距离,加上穿越尾部到旧项链头部后从头部移动
                # 到i处的距离,减去num和i之间的在num右侧和i左侧的其他珍珠的个数。因为右侧占用位置i,所以计算距离时不能包括i

                d_right2 = (num - i) - sum(flag[i+1 : num])
                # 如果num放在i的右侧,并且num大于i,那么移动的距离等于num直接向左移动到i的距离,加上num左侧和i的右侧之间其他珍珠的个数。
            else:
                d_left2 = ((i-1) - num) - sum(flag[num+1:i])
                # 如果num放在i的左侧,并且num小于i,那么移动的距离等于num直接向右移动到i的距离,加上num右侧和i的左侧之间的其他珍珠的个数.
                # 并且右侧占用位置i,所以计算距离是不能包括i。

                d_right2 = num + 1 + (L - i) - sum(flag[:num]) - sum(flag[i:])
                # 如果num放在i的右侧,并且num小于i,那么移动的距离等于num先向左移动到旧项链头部的距离,再加上穿越到头部到旧项链尾部后从尾
                # 部向左移动到i的距离,减去在i右侧和num的左侧的其他珍珠的个数

            distance1 += min(d_left1, d_right1)
            distance2 += min(d_left2, d_right2)

            print("i=",i,'num=',num,'d_right2=',d_right2, 'd_left2=',d_left2, 'distance=(',distance1, distance2, ')')


        else: #如果i不是空位,这个时候就不用分是左还是右,因为都不能占用i。
            if num == i:
                d_left = 0
                d_right = 0
            elif num > i:
                d_left = (L - num) + (i-1) - sum(flag[num + 1:]) - sum(flag[:i])
                # 如果num放在i的左侧,并且num大于i,那么移动的距离等于num先移动到旧项链尾部的距离,加上穿越尾部到旧项链头部后从头部移动
                # 到i处的距离,减去num和i之间的在num右侧和i左侧的其他珍珠的个数.

                d_right = (num - (i + 1)) - sum(flag[i + 1:num])
                # 如果num放在i的右侧,并且num大于i,那么移动的距离等于num直接向左移动到i的距离,加上num左侧和i的右侧之间其他珍珠的个数。
            else:
                d_left = (i - 1 - num) - sum(flag[num + 1:i])
                # 如果num放在i的左侧,并且num小于i,那么移动的距离等于num直接向右移动到i的距离,加上num右侧和i的左侧之间的其他珍珠的个数。

                d_right = num + (L - (i + 1)) - sum(flag[:num]) - sum(flag[i + 1:])
                # 如果num放在i的右侧,并且num小于i,那么移动的距离等于num先向左移动到旧项链头部的距离,再加上穿越到头部到旧项链尾部后从尾
                # 部向左移动到i的距离,减去在i右侧和num的左侧的其他珍珠的个数
            distance += min(d_left, d_left)
            print("i=",i,'num=',num,'d_right=',d_right, 'd_left=',d_left, 'distance=',distance)


    if flag[i] == 1:
        if min_distance > distance:
            min_distance = distance
    else:
        if min_distance > min(distance1, distance2):
            min_distance = min(distance1, distance2)


    print('\n')

print(min_distance)
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 专业考题类型管理运行工作负责人一般作业考题内容选项A选项B选项C选项D选项E选项F正确答案 变电单选GYSZ本规程...
    小白兔去钓鱼阅读 12,935评论 0 13
  • 怦然心动的人生整理魔法 1.想象理想生活,先从自己心里想出最爱的样子,在慢慢打造,不仅家里焕然一新,自己内心也能得...
    小爱歌呀阅读 1,378评论 0 0
  • 青春是岁月写下最动人的字句,而结局却并非是写好的剧本,有时候人生在世,也并非只为自己而活。 2012年,春,天...
    悬海阅读 1,761评论 2 1
  • 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例...
    亮涛阅读 1,657评论 0 0