各位数字之和为5的数，救救孩子，求一个代码

输入一个1000以内的正整数n，在同一行内输入[0，n]之间各位数字之和为5的数，数字之间用空格分开(行末一个空格）。

s = input()

res = []

for i in range(int(s)):
    a = str(i)
    if sum(int(j) for j in a) == 5:
        res.append(i)
print(res)

import more_itertools as mi 


n = int(input(">>>"))
result = []
for i in range(1, n + 1):
    x = mi.distinct_permutations(range(n + 1), i)
    result.extend([j for j in x if sum(j) == 5])

print('\n'.join(map(lambda x:' '.join( map(str, x)), result)))