蓝桥杯算法训练数字游戏-杨辉三角-dfs-python题解

数字游戏（杨辉三角+dfs）

问题描述
　　给定一个1～N的排列a[i]，每次将相邻两个数相加，得到新序列，再对新序列重复这样的操作，显然每次得到的序列都比上一次的序列长度少1，最终只剩一个数字。
　　例如:
　　3 1 2 4
　　4 3 6
　　7 9
　　16
　　现在如果知道N和最后得到的数字sum，请求出最初序列a[i]，为1～N的一个排列。若有多种答案，则输出字典序最小的那一个。数据保证有解。
输入格式
　　第1行为两个正整数n，sum
输出格式
　　一个1～N的一个排列
样例输入
4 16
样例输出
3 1 2 4
数据规模和约定
　　0<n<=10

def yh_tri(n):
    if n==1:
        result=[1]
    else:
        result=[[0 for i in range(n)]for j in range(n)]
        for i in range(n):
            for j in range(n):
                result[i][0]=1
                if i==j:
                    result[i][j]=1
        for x in range(2,n):
            for y in range(1,x):
                result[x][y]=result[x-1][y-1]+result[x-1][y]
    return result[n-1]

n,s=map(int,input().split())
y=yh_tri(n)

res=[]
used=[0 for i in range(s)]
flag=0
def dfs(num,tem_sum):
    global flag
    if flag==1:
        return
    if num==n:
        if tem_sum==s:
            for x in res:
                print(x,end=' ')
                flag=1

    else:
        for i in range(1,n+1):
            if used[i]==0:
                used[i]=1
                res.append(i)
                tem_sum=tem_sum+y[num]*i
                if tem_sum<=s:
                    dfs(num+1,tem_sum)
                used[i]=0
                res.pop()
                tem_sum=tem_sum-y[num]*i
                
if n==1:
    print(s)
else:
    dfs(0,0)