# Nicht besonders elegant geschrieben (objektorientiert wäre wohl besser).
import pygame

def initialisiere_visualisierung(n, max, einheit=5, wartezeit=300, titel=''):
    global fenster
    global fensterbreite
    global fensterhoehe
    global MAX
    global EINHEIT
    global WARTEZEIT
    MAX = max
    EINHEIT = einheit
    WARTEZEIT = wartezeit
    fensterbreite = n*einheit
    fensterhoehe = max*einheit

    pygame.init()
    pygame.display.set_caption(titel)

    fenstergroesse = (fensterbreite, fensterhoehe)
    fenster = pygame.display.set_mode(fenstergroesse)
    # Seltsam, aber ohne die nächste Zeile erscheinen
    # beim ersten Zeichnen keine Balken.
    pygame.time.wait(100)

def visualisiere(a):
    fenster.fill('white')
    for i in range(len(a)):
        rechteck = pygame.Rect(i * EINHEIT, fensterhoehe - a[i]*EINHEIT, EINHEIT-1, a[i]*EINHEIT)
        pygame.draw.rect(surface=fenster, color='black', rect=rechteck)
    pygame.display.flip()
    pygame.time.wait(WARTEZEIT)

def speichere(name):
    pygame.image.save(fenster, name)