python - python3中将整数除以 "as symmetric array as possible"

Question

我试图在 python3 中将整数拆分为“尽可能对称的数组”，但似乎不太正确。

所需的输入/输出可以通过此数据模式表进行描述。

# data pattern TABLE
# INPUT (int) => OUTPUT (len(array), array )
# 1 => 1 1
# 2 => 1 2
# 3 => 2 2,1
# 4 => 2 2,2
# 5 => 2 3,2
# 6 => 2 3,3
# 7 => 3 3,2,2
# 8 => 3 3,3,2
# 9 => 3 3,3,3
# 10 => 3 4,3,3
# 11 => 3 4,4,3
# 12 => 3 4,4,4
# 13 => 4 4,3,3,3
# 14 => 4 4,4,3,3
# 15 => 4 4,4,4,3
# 16 => 4 4,4,4,4
# 17 => 4 5,4,4,4
# 18 => 4 5,5,4,4
# 19 => 4 5,5,5,4
# 20 => 4 5,5,5,5

到目前为止，我通过在 stackoverflow 上查看一些内容得到了这段代码。

def splitnum(a, n):
    num, div = a, n
    return (num // div + (1 if x < num % div else 0)  for x in range (div))

x =20
print(list(splitnum(x, int(x/2))))

但是它会导致数组被分成两部分。我真的只是想正确理解该模式，如果有人知道一些巧妙的技巧来解决这个问题，那将非常有帮助!

感谢任何帮助!

干杯!

Answer 1

我认为这是一个很好的提议:

import math
def squarit(n: int):
  size = int(math.floor(math.sqrt(n)))
  ans = [n//size]*size
  for i in range(n-n//size*size):
    ans[i] += 1
  return size, ans

for i in range(4, 25):
  print(f'{i}: {squarit(i)}')

结果:

4: (2, [2, 2])
5: (2, [3, 2])
6: (2, [3, 3])
7: (2, [4, 3])
8: (2, [4, 4])
9: (3, [3, 3, 3])
10: (3, [4, 3, 3])
11: (3, [4, 4, 3])
12: (3, [4, 4, 4])
13: (3, [5, 4, 4])
14: (3, [5, 5, 4])
15: (3, [5, 5, 5])
16: (4, [4, 4, 4, 4])
17: (4, [5, 4, 4, 4])
18: (4, [5, 5, 4, 4])
19: (4, [5, 5, 5, 4])
20: (4, [5, 5, 5, 5])
21: (4, [6, 5, 5, 5])
22: (4, [6, 6, 5, 5])
23: (4, [6, 6, 6, 5])
24: (4, [6, 6, 6, 6])