以菱形打印数字

时间:2014-02-26 17:46:28

标签: python

我遇到了这个问题:http://codegolf.com/numeric-diamonds要求您以这种格式打印数字:

  1
 4 2
7 5 3
 8 6
  9
           1
         7   2
      13   8   3
    19  14   9   4
  25  20  15  10   5
31  26  21  16  11   6
  32  27  22  17  12
    33  28  23  18
      34  29  24
        35  30
          36
                             1
                         11     2
                      21    12     3
                   31    22    13     4
                41    32    23    14     5
             51    42    33    24    15     6
          61    52    43    34    25    16     7
       71    62    53    44    35    26    17     8
    81    72    63    54    45    36    27    18     9
 91    82    73    64    55    46    37    28    19    10
    92    83    74    65    56    47    38    29    20
       93    84    75    66    57    48    39    30
          94    85    76    67    58    49    40
             95    86    77    68    59    50
                96    87    78    69    60
                   97    88    79    70
                      98    89    80
                         99    90
                           100

我能够找到进入特定行的数字就好了,但我安排它们的逻辑非常模糊。

for i in range(2*n - 1):
    m = <list of numbers for this row>
    row_width = (n+i)*(n/3) if i < n else row_width - (n/3)

    print ''.join(str(i).rjust(n/2 + 1) for i in m).strip().rjust(row_width))

从代码高尔夫的角度来看,这是可以的,但是有一种干净,pythonic的方式来排列这种格式的2D数组中的数字吗?

3 个答案:

答案 0 :(得分:3)

不知道“pythonic”,但这看起来很干净:

size = 10
maxlen = len(str(size * size))
m = size * 2 - 1
matrix = [[' ' * maxlen] * m for _ in range(m)]

for n in range(size * size):
    r = n // size
    c = n % size
    matrix[c + r][size - r - 1 + c] = '{0:{1}}'.format(n + 1, maxlen)

print '\n'.join(''.join(row) for row in matrix)

答案 1 :(得分:0)

因此函数M会返回每行中数字的值(例如n=10,然后是M(n) = [[1],[11,2],...]]cell函数会使每个数字占用的数量都相同空格作为最大数字(学习选定答案)。

def M(n):
    t = 2 * n - 1
    N = [range(i * n + 1, (i + 1) * n + 1) for i in range(n)]

    M = []
    for k in xrange(t):
        M.append([])
        for i in reversed(xrange(n)):
            for j in reversed(xrange(n)):
                if i + j == k:
                    M[k].append(N[i][j])
    return M



def magic_square(n):
    t = 2 * n - 1
    maxlen = len(str(n * n))
    gap = ' ' * maxlen

    cell = lambda num:  '{0:{1}}'.format(num, maxlen)

    for m in M(n):
        padding = gap  * ((t - (2 * len(m) - 1)) / 2)
        print padding + gap.join(map(cell, m)) + padding

答案 2 :(得分:0)

def print_diamond(n):
    if n == 1:
        print 1
        return
    maxlen = len(str(n*n))
    gap =  maxlen * ' '
    first = 0
    for i in xrange(2*n-1):
        if i < n:
            first = i*n + 1
            print (abs(i-n)-1)*gap + gap.join(map(lambda x: '{0:{1}}'.format(x, maxlen), xrange(first, i, 1-n)))
        else:
            first += 1
            print (abs(i-n)+1)*gap + gap.join(map(lambda x: '{0:{1}}'.format(x, maxlen), xrange(first, n*(i+2-n)-1, 1-n)))