python实现正整数N等于M个正整数之和

系统 201 0

#_*_ coding: UTF-8 _*_
#正整数N等于M个正整数之和

import itertools
import sys
import os 
import copy

#参数说明
#total:表示有多少个数,M
#sum:表示正整数的和,N
#comb_list:存储最终输出的所有列表组合
#tmp_comb_list:临时最终输出的所有列表组合
#max_num:当前组合中的最大取值
#num_pos:当前求出的数值在当前list中保存的位置
def doFindComb(total, sum, comb_list, tmp_comb_list, max_num=1, num_pos=0):
    tmp_result = copy.deepcopy(tmp_comb_list)
    if(total > sum):
        return
    elif(total == sum):
        for i in range(num_pos, total+num_pos):
            tmp_result[i]=1
        result_data = list(itertools.permutations(tmp_result))
        list_len = len(result_data)
        for i in range(0, list_len):
            result_data[i] = list(result_data[i])
        for item in result_data:
            if item not in comb_list:
                comb_list.append(item)
        return
    elif(total == 1):
        if(sum <= max_num):
            tmp_result[num_pos]=sum
            result_data = list(itertools.permutations(tmp_result))
            list_len = len(result_data)
            for i in range(0, list_len):
                result_data[i] = list(result_data[i])
            for item in result_data:
                if item not in comb_list:
                    comb_list.append(item)
        return
    
    min = int(sum/total)
    if(sum % total != 0):
        min = int(sum/total) +1
    max = sum-total+1
    tmp_list = [0 for x in range(0, total+num_pos)]
    for i in range(min, max+1):
        tmp_list = copy.deepcopy(tmp_comb_list)
        tmp_list[num_pos]=i
        doFindComb(total-1, sum-i, comb_list, tmp_list, i, num_pos+1)


'''
comb_list = []
tmp_comb_list = [0 for x in range(0, total)]
total = 5
sum = 20
doFindComb(total, sum, comb_list, tmp_comb_list)
'''
if __name__ == "__main__":
    if (len(sys.argv) != 3):
        print("parameter error")
        print("usage: python xxx.py m n")
    else:
        total = int(sys.argv[1])
        sum = int(sys.argv[2])
        comb_list = []
        tmp_comb_list = [0 for x in range(0, total)]
        doFindComb(total, sum, comb_list, tmp_comb_list)
        print(comb_list)


更多文章、技术交流、商务合作、联系博主

微信扫码或搜索:z360901061

微信扫一扫加我为好友

QQ号联系: 360901061

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请请扫描上面二维码支持博主1元、2元、5元等您想捐的金额吧,狠狠点击下面给点支持吧

发表我的评论
最新评论 总共0条评论