Fair Cut

参考: https://pastebin.com/1R543whN

直接用DP数组表示正解是无法进行状态转移的,换种思路,每次计算DP[i]的时候把当前第i位所有的计算都先算出来,因为本身就是确定的,这样DP[i]虽然不是这个子问题的解,但是最后的DP[-1]是正解

#
# Complete the fairCut function below.
# ref: https://pastebin.com/1R543whN
def fairCut(k, a):
    #
    # Write your code here.
    #
    a.sort()
    n=len(a)
    dp=[[float('inf') for _ in range(n+1)] for _ in range(n+1)]
    dp[0][0]=0
    
    for i in range(n):
        for j in range(i+1):
            if j>k or i-j>n-k: continue
            
            tmp_li = dp[i][j]+a[i]*(i-j - (n-k-(i-j)))
            dp[i+1][j+1]=min(dp[i+1][j+1], tmp_li)
            
            tmp_lu = dp[i][j]+a[i]*(j - (k-j))
            dp[i+1][j]=min(dp[i+1][j], tmp_lu)
            
    return dp[n][k]
            

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值