Editing 0841 - Bomber (section)

==Rezolvare==
<syntaxhighlight lang="python" line="1">
 
def validare(n, i, bombe):
    # Verifica daca valorile de intrare sunt valide
    if not 1 <= n <= 100:
        return False
    if not 1 <= i <= n:
        return False
    for x, y, p in bombe:
        if not -200 <= x <= 200 or not -200 <= y <= 200:
            return False
        if not 1 <= p <= 30:
            return False
    return True

def rezolvare(n, i, bombe):
    # Initializeaza lista de bombe explodate cu bomba i
    explodate = [i]
    while True:
        # Gaseste bombele care trebuie sa explodeze in runda curenta
        de_explodat = []
        for j, (x, y, p) in enumerate(bombe, start=1):
            if j not in explodate and any((x-x0)**2 + (y-y0)**2 <= p0**2 for x0, y0, p0 in (bombe[k-1] for k in explodate)):
                de_explodat.append(j)
        # Iesirea din bucla while daca nu mai exista bombe de explodat
        if not de_explodat:
            break
        # Adauga bombele care trebuie sa explodeze la lista de bombe explodate
        explodate += de_explodat
    # Returneaza numarul de bombe neexplodate
    return n - len(explodate)

def main():
    # Citeste valorile de intrare din fisierul bomber.in
    with open('bomber.in') as f:
        n, i = map(int, f.readline().split())
        bombe = [tuple(map(int, line.split())) for line in f]

    # Valideaza datele de intrare
    if not validare(n, i, bombe):
        print('Date de intrare invalide')
        return

    # Rezolva problema si afiseaza rezultatul in fisierul bomber.out
    with open('bomber.out', 'w') as f:
        f.write(str(rezolvare(n, i, bombe)) + '\n')

if __name__ == '__main__':
    main()

</syntaxhighlight>