View source for 0610 - Spion

Într-un loc îndepărtat amplasat strategic, se află mai mulţi soldați identificaţi prin numere naturale nenule unice. Ordinea și disciplina militară impun aşezarea soldaților în linie la apelul de dimineață. Fiecare soldat a primit la sosirea în unitate numărul său și numerele de identificare ale camarazilor care trebuie să fie în stânga și în dreapta sa la aliniere. Excepţie fac doi soldaţi santinele care primesc doar numărul propriu de identificare şi se vor aşeza pe prima, respectiv pe ultima poziţie din linie. Uneori, unii soldaţi sunt plecați în misiuni, iar apelul de dimineaţă în unitate se face astfel: fiecare dintre cei prezenţi strigă cele trei numere primite – numărul lui, numărul camaradului din stânga lui şi al camaradului din dreapta lui, chiar dacă aceștia lipsesc.

Ei nu ştiu însă că, după un zid, stă pitită spioana Bulbuka, trimisă în recunoaștere. Ea îi ascultă şi înregistrează toate tripletele de numere care sunt strigate. După cele trei numere strigate de fiecare soldat, Bulbuka, fiind foarte inteligentă, deduce corect câte grupuri de soldaţi care și-au strigat prezența sunt până în momentul respectiv. Ea consideră că un grup este format din toţi acei soldaţi care şi-au strigat propriul număr şi care sunt aşezaţi unul lângă celălalt (dacă un soldat a strigat numărul camaradului, acesta din urmă nu este inclus în grup decât dacă a strigat şi el propriul număr).

= Cerința =
Aflați care sunt numerele de grupuri de soldați deduse de Bulbuka în fiecare moment.

= Date de intrare =
Pe prima linie a fişierului <code>input.txt</code> se află <code>N</code>, numărul de soldaţi ce îşi vor striga prezența. Pe următoarele <code>N</code> linii se află câte <code>3</code> numere <code>x[i]</code>, <code>s[i]</code> și <code>d[i]</code>, separate prin câte un spaţiu, reprezentând tripletele pe care le strigă fiecare soldat (<code>x[i]</code> numărul soldatului, <code>s[i]</code> numărul camaradului din stânga lui şi <code>d[i]</code> numărul camaradului din dreapta lui).

= Date de ieșire =
Fişierul de ieşire <code>output.txt</code> va conţine <code>N</code> linii. Pe fiecare dintre cele <code>N</code> linii se va scrie numărul de grupuri dedus de Bulbuka la momentul respectiv.

= Restricții și precizări =

* <code>1 < N  < 500000</code>

== Exemplul 1 ==
input.txt:

6

9 2 3

8 5 4

5 3 8

7 1 2

2 7 9

3 9 5

output.txt:

1

2

2

3

2

1

Explicație:

Prima dată strigă soldatul <code>9</code> care e notat prezent, deci primul grup este format doar din el. Camarazii <code>2</code> și <code>3</code> ar trebui să fie în stânga și dreapta lui dar nu și-au strigat prezența încă. În consecință, nu fac parte în acest moment din grupul lui <code>9</code>.

Strigă soldatul <code>8</code>, se formează două grupuri: <code>{9}</code> și <code>{8}</code>.

Strigă <code>5</code>, el va intra în grupul lui <code>8</code>, deci vor fi tot două grupuri: <code>{9}</code> și <code>{8,5}</code>.

Strigă <code>7</code>, dar pentru că cei doi camarazi nu s-au strigat încă prezenţi, <code>7</code> va forma un grup singur. Vor fi acum trei grupuri: <code>{7}</code>, <code>{9}</code> și @{8,5} etc.

== Exemplul 2 ==
input.txt:

9999999999

9 2 3

8 5 4

5 3 8

7 1 2

2 7 9

3 9 5

Output:

Input-ul nu convine conditiilor

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
NMAX = 1600000


def verificare(n):
    if not(2<=n<500000):
        print("Input-ul nu convine conditiilor")
        exit()

n = 0
grupuri = 0

soldati = [0] * ((NMAX + 31) // 32)

def bindex(b):
    return b // 32

def boffset(b):
    return b % 32

def set_bit(b):
    soldati[bindex(b)] |= 1 << boffset(b)

def get_bit(b):
    return (soldati[bindex(b)] & (1 << boffset(b))) != 0

with open("input.txt", "rt") as infile, open("output.txt", "wt") as outfile:
    n = int(infile.readline())
    verificare(n)
    for _ in range(n):
        x, s, d = map(int, infile.readline().split())

        prezenta = get_bit(s) + get_bit(d)
        grupuri += 1 - prezenta

        outfile.write(f"{grupuri}\n")

        set_bit(x)

</syntaxhighlight>