递归求解一组元素的所有排列情况&包含m个元素的组合情况&网易笔试题-合唱团

#encoding:utf-8

#打印排列组合-递归：
#每次取序列中的一个元素作为头，每个头引导包含的排列情况数  等于剩下其余元素的所有排列情况
#因此递归执行这个过程，当前序列只剩一个元素时，则排列就是它自己
def perm_sub(all):
result=[]
if len(all)==1:
result.append(all[0])
else:
for i in xrange(0,len(all)):
oldFirst=all[0] #交换当前的头元素
all[0]=all[i]
all[i]=oldFirst
preResult=perm(all[1:])
for j in xrange(0,len(preResult)):
result.append(all[0]+preResult[j])
all[i]=all[0]  #交换后要换回去
all[0]=oldFirst
return result

#视元素为字符串列出排列情况，入口函数
def perm(all):
for i in xrange(0,len(all)):
all[i]=str(all[i])
return perm_sub(all)

#打印所有的组合情况
def comb(all):
for i in xrange(0,len(all)):
all[i]=str(all[i])
result=[]
for i in xrange(1,len(all)+1):
result.extend( comb_sub(all,i) )
return result

#打印包含m个元素的组合情况，m个元素的组合数=第一个元素+剩下序列元素的m-个组合数
#如1234，m=2个元素的组合数等于1+[2,3,4](234的m-1组合),2+[3,4](34的m-1组合)，3+[4]
#如1234，m=3个元素的组合数等于1+[23,24,34](234的m-1组合),2+[34](34的m-1组合)
def comb_sub(all,m):
result=[]
if m==1:
result=all
else:#每次循环将起始位置往后移1，并且保证剩下的元素数大于m-1组合数
for i in xrange(0,len(all)-(m-1) ):#最后至少要剩下m-1个数，如果剩下的数量都少于要组合的数量就无法组合了
preResult=comb_sub(all[i+1:],m-1)
for j in xrange(0,len(preResult)):
result.append( all[i]+ preResult[j])

return result

#求all中的m个元素的所有组合情况，要求元素之间的位置间隔不超过d
#按组合情况的求解方法求解，添加组合时与前一个元素的间隔不超过d的条件
def wangyi_sub(all,m,pre,d):#pre=0表示当前是第一个头,=1表示不是第一个头
result=[]
if m==1:
result=all[:d]
else:
if pre==0:
maxStart=len(all)-(m-1)
else:
maxStart=d if d < len(all)-(m-1) else len(all)-(m-1)
for i in xrange(0,maxStart):
preResult=wangyi_sub(all[i+1:],m-1,1,d)
for j in xrange(0,len(preResult)):
result.append( all[i] * preResult[j])
return result

# print perm(all)
# print comb(all)

#print comb_sub(all,2)

all=[7,4,7]
print wangyi_sub(all,2,0,50)
print max(wangyi_sub(all,2,0,50))

# all=['1','2','3','4','5','6','7','8','9']
# print wangyi_sub(all,3,0,2)
# print max(wangyi_sub(all,3,0,2))

def wangyiniuke():
import sys
n=sys.stdin.readline()
s=sys.stdin.readline().split(' ')
for i in xrange(0,len(s)):
s[i]=int(s[i])
kd=sys.stdin.readline().split(' ')
k=int(kd[0])
d=int(kd[1])
i=0
allResult=wangyi_sub(s,k,0,d)
print max(allResult)