球与盒子的分类问题

shadowrocker

定义一个函数f（n，r）,希望实现这样的功能：列出所有可能的n个自然数之和为r的组合。
比如说，输入f（3,4），希望输出的是
4,0,0
3,1,0
2,2,0
2,1,1

望大神们给些思路！

小火木

1. def f(box,ball):
   
2.     lis0=[0]*box
   
3.     lis1=[lis0]
   
4.     lis2=[]
   
5.     for i in range(ball):
   
6.        for each in lis1:
   
7.            for j in range(box):
   
8.                lis3=each[:]
   
9.                lis3[j]+=1
   
10.                lis4=lis3[:]
    
11.                lis4.sort()
    
12.                if lis4 not in lis2:
    
13.                    lis2.append(lis4)
    
14.                    lis5=lis2[:]
    
15.        lis1=lis2[:]
    
16.        lis2=[]   
    
17.     for each in lis5:
    
18.         print(each)
    
19. 
    
20. f(3,4)
    

复制代码

相当于r个球分别放到n个框，
每次放一个球，列出所有可能的情况，
去重，
然后再放一个球，重复
直到所有的球放完

holdme

这样应该就可以了：

1. from itertools import combinations
   
2. def f(n, r):
   
3.     for i in combinations(n*[j for j in range(r+1)],n):
   
4.         if sum(i) == r:
   
5.             yield i
   
6. 
   
7. if __name__ == '__main__':
   
8.     print(set(list(f(3,4))))

复制代码