Editing 3565 - Furnici 1 (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# 3565 - Furnici 1
def validare(lungime1, furnici):           # functia de validare a datelor de intrare
    if lungime1 < 1 or lungime1 > 10_000_000:
        raise ValueError

    if len(furnici) < 0 or len(furnici) > 100_000:
        raise ValueError

    for pozi, directie in furnici:
        if pozi < 0 or pozi > lungime1:
            raise ValueError
        if directie not in ['S', 'D']:
            raise ValueError

    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


def last_ant_falls(lungime1, furnici):                     # functia de rezolvare
    max_time = 0
    for pozi, directie in furnici:
        if directie == 'S':
            time = pozi
        else:  # direction == 'D'
            time = lungime1 - pozi
        max_time = max(max_time, time)
    return max_time


if __name__ == '__main__':
    file_in = open("furnici1in.txt", "r")         # declararea fisierelor
    file_out = open("furnici1out.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    # din cauza datelor de intrare pot aparea 2 tipuri de erori, valueError sau IndexError pe care le tratam
    try:
        lungime, n = map(int, file_in.readline().split())
        ants = []
        for _ in range(n):
            pos, direction = file_in.readline().split()
            pos = int(pos)
            ants.append((pos, direction))

        validare(lungime, ants)                 # apelul functiei de validare
        result = last_ant_falls(lungime, ants)  # apelul functiei de rezolvare
        file_out.write(str(result))

    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")

</syntaxhighlight>