Editing 3745 - Oposumi (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
def read_input(file_name):
    with open(file_name, 'r') as file:
        T, N = map(int, file.readline().strip().split())
        if T == 2:
            K = int(file.readline().strip())
        else:
            K = None
    return T, N, K


def write_output(file_name, result):
    with open(file_name, 'w') as file:
        for line in result:
            if isinstance(line, list):
                file.write(' '.join(map(str, line)) + '\n')
            else:
                file.write(str(line) + '\n')


def nivel_cel_mai_de_sus(N):
    poziții = []
    valoare = 1
    for i in range(N):
        nivel = []
        for j in range(i + 1):
            nivel.append(valoare)
            valoare += 1
        poziții.append(nivel)
    return poziții


def poziția_oposumului(N, poziții, K):
    poziții_rezultat = [['-' for _ in range(N)] for _ in range(N)]
    găsit = False
    for i in range(N):
        for j in range(i + 1):
            if poziții[i][j] == K:
                poziții_rezultat[i][j] = K
                găsit = True
                break
        if găsit:
            break
    return poziții_rezultat


def main():
    input_file = 'oposumi.in'
    output_file = 'oposumi.out'

    T, N, K = read_input(input_file)

    if not (1 <= N <= 1000):
        raise ValueError("N trebuie să fie între 1 și 1000")

    poziții = nivel_cel_mai_de_sus(N)

    if T == 1:
        result = []
        for i in range(1, N * (N + 1) // 2 + 1):
            găsit = False
            for nivel in range(N):
                if găsit:
                    break
                for j in range(len(poziții[nivel])):
                    if poziții[nivel][j] == i:
                        result.append(nivel + 1)
                        găsit = True
                        break
        result = [result]  # Wrap result in a list of lists
    elif T == 2:
        result = poziția_oposumului(N, poziții, K)

    write_output(output_file, result)


if __name__ == '__main__':
    main()
</syntaxhighlight>