#!/usr/bin/env python
# -*- encoding: utf-8; py-indent-offset: 4 -*-

# Author: Sebastian Mark
# CC-BY-SA (https://creativecommons.org/licenses/by-sa/4.0/deed.de)

# pylint: disable=missing-module-docstring,missing-function-docstring,consider-using-f-string


def readinput() -> list[int]:
    reports = []
    with open("input", "r", encoding="utf-8") as file:
        for line in file:
            levels = list(map(int, line.split(" ")))
            reports.append(levels)
    return reports


def is_valid_level_difference(i: int, j: int) -> bool:
    if i == j:
        return False

    if abs(i - j) > 3:
        return False

    return True


def is_sorted(levels: list) -> bool:
    asc_sorted = levels == sorted(levels)
    desc_sorted = levels == sorted(levels, reverse=True)
    return asc_sorted or desc_sorted


def is_save(levels: list) -> bool:
    if not is_sorted(levels):
        return False

    for i in range(len(levels) - 1):
        if not is_valid_level_difference(levels[i], levels[i + 1]):
            return False

    return True


def remove_level_by_index(levels: list, index: int) -> list:
    copy = levels.copy()
    copy.pop(index)
    return copy


def is_save_with_damper(levels: list) -> bool:
    if is_save(levels):
        return True

    for i in range(len(levels)):
        if is_save(remove_level_by_index(levels, i)):
            return True

    return False


def main():
    reports = readinput()

    # part 1
    save_count = sum(is_save(report) for report in reports)
    print("Save reports: %d" % save_count)

    # part 2
    save_count = sum(is_save_with_damper(report) for report in reports)
    print("Save reports (with damper): %d" % save_count)


if __name__ == "__main__":
    main()