day11.1.py

# Teil 1, successful

import numpy as np

file = "AOC_day11_input.txt"
document = []
document2 = []
koordinaten = []
such_zeichen = 1



def einlesen(filename):
    """
    liest alle Zeilen eines Textfiles ein und speichert jede einzelne Zeile in einer Liste
    """
    
    with open(filename, "r") as file:
        # read file line by line and output the lines
        
        for line in file:
            line = line.strip()
            line2 = []
            line3 = []
            for buchstabe in line:
                line2.append(buchstabe)
                if buchstabe == "#":
                    line3.append(1)
                else:
                    line3.append(0)
                
            document.append(line2)
            document2.append(line3)

    return document, document2

def print_field(matrix):
    # danke an
    # https://plainenglish.io/blog/a-python-example-of-the-flood-fill-algorithm-bced7f96f569
    # this function will print the contents of the array
    for y in range(len(matrix)):
        for x in range(len(matrix[0])):
        # value by column and row
            print(matrix[y][x], end=' ')
            if x == len(matrix[0])-1:
                # print a new line at the end of each row
                #print('\n')
                print(" ")
    
def nullen_zeilen_verdoppeln(array):
    """
    Verdoppelt die Zeile mit nur Nullen
    gefunden in https://stackoverflow.com/questions/28663856/how-do-i-count-the-occurrence-of-a-certain-item-in-an-ndarray
    """
    zeilennr = 0
    num_zeros = 0
    i = 0
    new_array = np.copy(array)
    
    for zeile in array:
        if np.all(zeile == 0):
            new_array = np.insert(new_array, zeilennr+i, zeile, axis=0)
            i +=1
        zeilennr +=1
    
    return new_array


def nuller_zeilen_und_spalten_verdoppeln(array):
    """
    nutzt nullen zeilen verdoppeln, transponiert, zeilen again, wieder tranponieren
    """
    a1 = nullen_zeilen_verdoppeln(array)
    a2 = np.transpose(a1)
    a3 = nullen_zeilen_verdoppeln(a2)
    new_array = np.transpose(a3)
    
    return new_array                


def koordinaten_einlesen(in_array, such_zeichen):
    """
    findet alle Einsen im array und gibt die Koordinaten in Form zeile (zuerst) dann die spalte als matrix aus 
    """
    ones = np.array([])
    ones = np.where(in_array == such_zeichen)
    matrix1 = []

    i = 0

    for element in ones[0]:
        matrix1.append([ones[0][i], ones[1][i]]) 
        i +=1
    
    return matrix1


def gesamt_abstand(mat):
    """
    berechnet den x und y abstand aller koordinaten voneinander und bildet die summe der Beträge
    """
    total_dist = 0
    total_abstand = 0

    for element in mat:
        i = 0
        for i in range(len(mat)):
            distance = abs(element[0] - mat[i][0]) + abs(element[1] - mat[i][1])
    #        print(distance)
            total_dist = total_dist + distance
    total_abstand = total_dist/2
    
    return total_abstand



# This is where it all begins
#
# 

einlesen(file)
document3 = np.array(document2)
#print(document2)
print_field(document)
print(" ")
print_field(document3)
print(" ")

document4 = nuller_zeilen_und_spalten_verdoppeln(document3)
print_field(document4)
print(" ")

n = np.count_nonzero(document4)
knoten = n
kanten =  (n*(n-1))/2
print("Anzahl Knoten:", knoten, "Anzahl Kanten:", kanten)
print(" ")
koordinaten = koordinaten_einlesen(document4, such_zeichen)
print(koordinaten)
print(" ")

print(gesamt_abstand(koordinaten))