算法训练—数字游戏

资源限制

时间限制：1.0s 内存限制：256.0MB

问题描述

　　给定一个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

import itertools
n,summ = map(int,input().split( ))
dp=[]
for i in range(1,n+1):#或者用dp=list(range(1,n+1))，此时dp=[]可省略
    dp.append(i)
for x in itertools.permutations(dp):#此时x已经全排列
     x = list(x)
     x1=x[:]#x[:]是x的副本，即改变x不会改变x1
     for i in range(n-1):
        for j in range(n-i-1):
           x[j]=(x[j]+x[j+1])
     if x[0] ==summ:
       for i in x1:#若不循环输出，直接输出x1则是列表形式而不是数字形式
           print(i,end=' ')
       break

运行超时，70分。

import itertools
import math
import numpy as np
n,summ =map(int,input().split( ))
dp = list(range(1,n+1))
p=[]
for i in range(1,n+1):
    p.append(math.factorial(n-1) / (math.factorial(i-1) * math.factorial(n - i)))
#print(p) 可知此处p为list
for x in itertools.permutations(dp):
    x=list(x)
    #print(x)
    m=sum(np.multiply(x,p))
    if m ==summ:
       #print(x)此处不能直接输出，因为直接输出是list形式
       for i in x:
           print(i,end=' ')
       break

python测试成功，放到系统里错误，0分。改不动了，摆烂。