思考了很久的蓝桥杯——ALGO-1005--数字游戏

题目描述

问题描述
　　给定一个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

思路分析

拿到这个题目哦，，作为一个算法小白，第一反应？？第一反应是全排列哦。。。用python的话想到的是 itertools.permutations

或者写个回溯法
swap（）
f（I+1）
swap（）

但是，，分数分别是40 和70

在博客的帮助下，思路就是用杨辉三角形最后一行分别乘全排列。

def yh(num):
    if num == 1:
        res = [1]
    else:
        res = [[0]*num for i in range(num)]
        for i in range(num):
            for j in range(num):
                res[i][0] = 1
                if i == j:
                    res[i][j] = 1
        for i in range(2,num):
            for j in range(1,i):
                res[i][j] = res[i-1][j-1] + res[i-1][j]
    return res[num-1]
import itertools
n,s=map(int,input().split())
a=[i for i in range(1,n+1)]
for l in itertools.permutations(a,n):
    l=list(l)
    temp=l.copy()
    length=len(temp)
    fun=lambda x,y:x*y
    tmp=sum(list(map(fun,yh(n),temp)))
    if tmp==s:
        for i in l:
            print(i,end=" ")
        break

但是只有70分。。。哎

一看题目类型，，搜索，。。。呢就dfs把

代码如下，解析在注释里面：

#杨辉三角形不多说了
def yh(num):
    if num == 1:
        res = [1]
    else:
        res = [[0]*num for i in range(num)]
        for i in range(num):
            for j in range(num):
                res[i][0] = 1
                if i == j:
                    res[i][j] = 1
        for i in range(2,num):
            for j in range(1,i):
                res[i][j] = res[i-1][j-1] + res[i-1][j]
    return res[num-1]

n,sum=map(int,input().split())
yh = yh(n)
d=[]
visit=[0 for i in range(0,n+1)]
def dfs(step,s):
	#边界条件
    if s>sum:
        return False
    #说明遍历了n个元素
    if step==n:
        if s==sum:
            for i in d:
                print(i,end=" ")
            return True
        else:
            return False
    for i in range(1,n+1):
    	#如果该躺遍历的元素没有被访问过，才继续，不然直接下一轮
        if visit[i]==0:
        	#先将其设为被访问
            visit[i]=1
            d.append(i)#
            if dfs(step+1,s+yh[step]*i):#此步很关键
            #一般dfs（step+1）就行了，这里使用杨辉三角形进行剪枝。每一趟要求s不能大于s+yh[step]*i  每一步都进行了限制，大大缩减计算代价。！！！！！！！
                return True
            visit[i]=0
            d.pop()
    return False

if n==1:
    print(sum)
else:
    dfs(0,0)