题解 | #极客杯-快乐假期-GGboy的圣诞袜#

极客杯-快乐假期-GGboy的圣诞袜

https://ac.nowcoder.com/acm/contest/73273/E

GGboy的圣诞袜

目标:使得配对的袜子能量值最小

首先我们先证明,如果第i双袜子没有丢,我们让那双袜子自己配对的操作是最优的。

证明:

如果有四只袜子

++

所以我们不用考虑成双的袜子,只需要考虑丢失了颜色的袜子即可。

因此我们的问题转换为,配对只袜子,使得能量值最小


现在我们来分情况讨论

如果K是偶数,似乎排序,然后每两个袜子进行配对即可。为什么呢?这里设排完序的袜子是A1到AK

用归纳法证明,在最优解中,总是和进行配对

假设进行配对,进行配对,且

就有+ +,因为左边的每一项都大于右边的每一项

所以我们选择去配对,而不是其他。因此偶数的情况,我们就证明了


如果k是奇数,我们可以枚举那只不参与配对的袜子,然后剩下的袜子按照偶数的情形去解决就行。但是暴力枚举是n2复杂度。如果我们可以预先处理一些值呢,比如当我们让是那只不参与配对的袜子,那么其他的和应该是+ 因此我们可以预处理的前缀和以及后缀和,就可以将时间复杂度降低为On了

具体代码实现:

def I(): return input()
def II(): return int(input())
def MI(): return map(int, input().split())
def LI(): return list(input().split())
def LII(): return list(map(int, input().split()))

N, k = MI()

arr = LII()
arr.sort()


# 奇数,枚举不参与配对的袜子,只需要枚举1,3,5,7...即可
pre = [0]*(k+1)
suf = [0]*(k+1)
for i in range(1, k+1):
    pre[i] = pre[i-1]
    if i % 2 == 0:
        pre[i] += arr[i-1] - arr[i-2]
for i in range(k-1, -1, -1):
    suf[i] = suf[i+1]
    if ((k-i)%2==0):
        suf[i] += arr[i+1] - arr[i]
ans = float("inf")
for i in range(0, k+1, 2):
    ans = min(ans, pre[i] + suf[i])
print(ans)

全部评论

相关推荐

HNU_fsq:建议直接出国,这简历太6了。自愧不如
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务