求一個(gè)多元一次函數(shù)的可能解

NO_Getway

cu的大神們，小弟學(xué)藝不精，請(qǐng)教個(gè)問(wèn)題

對(duì)于給定的元祖（或者列表）a，求元祖（或者列表）中各個(gè)元素乘一個(gè)整數(shù)然后在求和等于一個(gè)常數(shù)b

例子：a=(1,2,3)，b=100,求1*x + 2 * y + 3 * z = 100中x,y,z的可能解對(duì)應(yīng)的表達(dá)式。
解有：
1*100 +2 *0 +3*0=100
1*98 + 2 *1 +3 *0 =100
等

那元祖（或者列表）如果為定長(zhǎng)，那么尚可把每個(gè)可能遍歷求解，那么如果元祖（或者列表）是未知長(zhǎng)度呢？

求給個(gè)思路或者給段核心代碼，謝謝

bikong0411

for x in xrange(100):
    for y in xrange(50):
        for z in xrange(33):
            if x + 2*y + 3 * z == 100:
                print("x=%d,y=%d,z=%d" % (x,y,z))

ssfjhh

1. #一條直線方程，解的個(gè)數(shù)是無(wú)限的，我暫且假定，樓主想要的是只有非負(fù)整數(shù)解。
   
2. a = (1, 2, 3)
   
3. b = 100
   
4. 
   
5. for x in range(0, b // a[0]):
   
6.     for y in range(0, b // a[1]):
   
7.         for z in range(0, b // a[2]):
   
8.             if a[0] * x + a[1] * y + a[2] * z == b:
   
9.                 print('{i} * {x} + {j} * {y} + {k} * {z} = '.format(
   
10.                     i = a[0], j = a[1], k = a[2], x = x, y = y, z = z, b = b))
    

復(fù)制代碼

NO_Getway

回復(fù) 3# ssfjhh

就是你理解的意思

那如果 a這個(gè)元祖 長(zhǎng)度不定呢？
總不能這樣子遍歷啊 我提問(wèn)上說(shuō)道了

麻煩解答下呢

   

NO_Getway

回復(fù) 2# bikong0411

這個(gè)是簡(jiǎn)單，如果a的元素長(zhǎng)度不定呢
   

ssfjhh

回復(fù) 5# NO_Getway


    顯而易見，遞規(guī)解決，代碼只能晚上給你。

icymirror

回復(fù) 1# NO_Getway
你可以嘗試使用SymPy的Solver，這個(gè)可以的。

Hadron74

回復(fù) 6# ssfjhh

寫了個(gè)遞歸的代碼，按照你的思路，還能改進(jìn)嗎？
、

1. def printResult(A,a,b):
   
2.     result = []
   
3.     for ll,ai in zip(A,a):
   
4.         result.append( ' {} * {} '.format(ll,ai))
   
5.     print "+".join(result)+" = {} ".format(b)
   
6. 
   
7. def solveEq(a,b,P,A,i):
   
8.     if i == len(P):
   
9.         s = sum([ai*Ai for Ai,ai in zip(A,a)])
   
10.         if s == b :
    
11.             printResult(A,a,b)
    
12.         return
    
13. 
    
14.     for j in range(P[i]):
    
15.         A[i] = j
    
16.         solveEq(a,b,P,A,i+1)
    
17. 
    
18. 
    
19. a = (1,2,3)
    
20. b = 10
    
21. P = [b/ai+1 for ai in a ]
    
22. A = [0] * len(a)
    
23. 
    
24. solveEq(a,b,P,A,0)
    

復(fù)制代碼

結(jié)果：

1. 0 * 1 + 2 * 2 + 2 * 3  = 10
   
2. 0 * 1 + 5 * 2 + 0 * 3  = 10
   
3. 1 * 1 + 0 * 2 + 3 * 3  = 10
   
4. 1 * 1 + 3 * 2 + 1 * 3  = 10
   
5. 2 * 1 + 1 * 2 + 2 * 3  = 10
   
6. 2 * 1 + 4 * 2 + 0 * 3  = 10
   
7. 3 * 1 + 2 * 2 + 1 * 3  = 10
   
8. 4 * 1 + 0 * 2 + 2 * 3  = 10
   
9. 4 * 1 + 3 * 2 + 0 * 3  = 10
   
10. 5 * 1 + 1 * 2 + 1 * 3  = 10
    
11. 6 * 1 + 2 * 2 + 0 * 3  = 10
    
12. 7 * 1 + 0 * 2 + 1 * 3  = 10
    
13. 8 * 1 + 1 * 2 + 0 * 3  = 10
    
14. 10 * 1 + 0 * 2 + 0 * 3  = 10

復(fù)制代碼

NO_Getway

回復(fù) 6# ssfjhh

先謝謝了
   

ssfjhh

回復(fù) 8# Hadron74


    Nice。

@NO_Getway