算法/编程练习:找出若干个数使其和最接近于M

找出若干个数使其和最接近于M


1. 题目

给定一个由正数组成的列表alts,一个目标数M
需要从alts中选取若干个备选数,使其和为M
若找不到和刚好与M相等的备选数列表,则返回和与M最接近的备选数列表
若有多个结果,返回一个即可
eg1.
	输入: 
		alts = [10, 9, 8, 7, 6, 5]
		M = 22
	输出: [10, 7, 5] 或 [9, 8, 5]

eg2.
	输入: 
		alts = [10, 7, 6, 3]
		M = 18
	输出: [10, 7]

拓展:
	若alts中同时有正数和负数呢

2. 思路:

先考虑alts中只有正数的情况

由于只需要返回一个可行结果或最优结果,那么可以从大到小在alts中搜索备选数,
当备选数之和等于M时结束搜索,否则新加入或修改已经加入的备选数

假设已经有备选数列表A,其和为N,与M之差为d=M-N,
则选择下一个备选数时应选择小于等于d且与d最接近的数

若新加入一个数之后,发现从剩下的数中肯定找不到可行解了,则说明前面加入的那个数时错误的,应换掉

例如eg1.的搜索过程
	step1: A = [10], N = 10, d = 12 (22-10), 剩下的数中9与12最接近,下一步加入9
	setp2: A = [10, 9], N = 19, d = 3, 剩下的数中没有小于等于3的,说明9的加入是错误的,
		   应换一个比9小的,最接近的是8,换8
	step3: A = [10, 8], N = 18, d = 4, 剩下的数中没有小于等于4的,说明8的加入是错误的,
		   应换一个比8小的,最接近的是7,换7
	step4: A = [10, 7], N = 17, d = 4, 剩下的数中与5最接近的是5,加入
	step5: A = [10, 7, 5], N = 22, d = 0, 结束


当alts中有负数时,不能根据差d选最接近的下一个备选数,
因为中间可能有值满足可行解但是被跳过了,
解决办法是不用d选择下一个备选数,直接选跟上小于等于上一个备选数且与之最接近的数即可

3. Python代码:

# -*- coding: utf-8 -*-


def FindAddends_BigFirst(M, alts, tol=0.0, max_loop=1000000): 
    '''
    从给定的列表alts(可以有负数)中选取若干个数,使其和最接近M
    
    思路: 大的备选数优先,从大到小依次加入备选数
         若加入新值之后找不到理想解,则删除最后加入的值,继续添加下一个更小的备选数
         下一个备选数确定方式:
             当alts中只有正数时,剩下的数中与剩余和(目标和减去当前和)最接近的数
             当alts中有负数时,直接取比最后加进去的数更小的数(搜索速度会变慢很多)
    
    Args:
        M: 目标和
        alts: list,备选数列表
        max_loop: 搜索次数限制
        
    Returns:
        choseds_best: list,最优备选数列表
    '''
    
    alts.sort() # 升序(大的数优先进入)
    all_postive = all([x >= 0 for x in alts])
    
    # 初始化
    # idx_last记录最新进入的数的索引号
    # chosed_idxs记录备选中的数的索引号
    # chosed_addends保存被选中的数
    # choseds_best保存最优结果
    # now_sum为chosed_addends求和
    idx_last = len(alts) - 1
    chosed_idxs = [idx_last]
    chosed_addends = [alts[idx_last]]
    choseds_best = []
    now_sum = alts[idx_last]
    
    # 搜索过程
    loop_count = 0
    while loop_count < max_loop:
        loop_count += 1
        
        # 更新最优解
        if abs(M - now_sum) < abs(M - sum(choseds_best)):
            choseds_best = chosed_addends.copy()
        
        # 结束条件
        if M == now_sum:
            print('找到最优解,结束搜索。')
            break
        
        # 无最优解(搜索到最小值且只剩它一个备选数),结束搜索
        if idx_last == 0 and len(chosed_idxs) == 1:
            print('无最优解,结束搜索。')
            break
        
        # 刚好搜索到最小值且不是最优解,此时去掉最小值并更改最小值前面一个进去的值
        if idx_last == 0:
            idx_last = chosed_idxs[-2] - 1
            del chosed_idxs[-2:]
            chosed_idxs.append(idx_last)
            del chosed_addends[-2:]
            chosed_addends.append(alts[idx_last])
            now_sum = sum(chosed_addends)
            continue
        
        # 下一个备选数
        if not all_postive:
            idx_last -= 1
        else:
            idx_last = backfind_sml1st_idx(M-now_sum, alts[0:idx_last])
        
        # 保留最后一个加进去的数情况下找不到最优解,更改最后进去的那个数
        if idx_last < 0:
            idx_last = chosed_idxs[-1] - 1
            chosed_idxs[-1] = idx_last
            chosed_addends[-1] = alts[idx_last]
            now_sum = sum(chosed_addends)
            continue
        
        chosed_idxs.append(idx_last)
        chosed_addends.append(alts[idx_last])
        now_sum += alts[idx_last]
            
    return choseds_best


def backfind_sml1st_idx(M, alts):
    '''
    alts(已升序排列)从后往前搜索,返回第一个小于等于M的数的索引
    '''
    if len(alts) == 0:
        return -1    
    idx = len(alts) - 1
    while idx >= 1 and alts[idx] > M:
        idx -= 1
    return idx
    

if __name__ == '__main__':    
    alts = [22, 15, 14, 13, 7, 6.1, 5, 21.5, 100]
    M = 22 + 21 + 4.1
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [200, 107, 100, 99, 98, 6, 5, 3, -1, -20, -25]
    M = 100 + 6 - 25
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [100, -100, -105, -102, -25, -30, -1]
    M = -26
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [10, 9, 8, 7, 6, 5]
    M = 17
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [10, 9, 8, -7, -6, -5]
    M = 3
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [10, 7, 6, 3]
    M = 18
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [10, 7, 6, 3]
    M = 12
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [10, 9, 8, 7, 6, 5, 3]
    M = 14
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))
    
    alts = [10, 9, 8, 7, 6, 5]
    M = 22
    print(FindAddends_BigFirst(M, alts, max_loop=10000000))

欢迎关注公众号:一本正经d胡说
Genlovy562

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值