#! /usr/bin/env python
#
def ytb_print ( n, a, title ):

#*****************************************************************************80
#
## YTB_PRINT prints a Young tableau.
#
#  Licensing:
#
#    This code is distributed under the GNU LGPL license.
#
#  Modified:
#
#    07 May 2015
#
#  Author:
#
#    John Burkardt
#
#  Parameters:
#
#    Input, integer N, the integer that is partitioned.
#
#    Input, integer A(N), describes the Young tableau.
#    A[I-1] is the row of the tableau on which I occurs.
#
#    Input, string TITLE, an optional title.
#
  print ( '' )
  if ( 0 < len ( title ) ):
    print ( title )

  row = 0

  while ( True ):

    row = row + 1

    row_length = 0

    for jm1 in range ( 0, n ):

      j = jm1 + 1

      if ( a[jm1] == row ):
        row_length = row_length + 1
        print ( '%4d  ' % ( j ), end = '' )

    if ( row_length <= 0 ):
      break
    else:
      print ( '' )

  return

def ytb_print_test ( ):

#*****************************************************************************80
#
## YTB_PRINT_TEST tests YTB_PRINT.
#
#  Licensing:
#
#    This code is distributed under the GNU LGPL license.
#
#  Modified:
#
#    07 May 2015
#
#  Author:
#
#    John Burkardt
#
  import numpy as np
  import platform

  print ( '' )
  print ( 'YTB_PRINT_TEST' )
  print ( '  Python version: %s' % ( platform.python_version ( ) ) )
  print ( '  YTB_PRINT prints a Young tableau.' )

  n = 6
  a = np.array ( [ 1, 1, 2, 2, 3, 1 ] )
  ytb_print ( n, a, '  A Young tableau:' )
#
#  Terminate.
#
  print ( '' )
  print ( 'YTB_PRINT_TEST:' )
  print ( '  Normal end of execution.' )
  return

if ( __name__ == '__main__' ):
  from timestamp import timestamp
  timestamp ( )
  ytb_print_test ( )
  timestamp ( )