Editing 4068 - Grade K

= Cerinţa =
Se dă lista muchiilor unui graf neorientat și un nod <code>k</code>. Să se determine nodurile din graf care au gradul egal cu gradul nodului <code>k</code>.

= Date de intrare =
Fişierul de intrare <code>gradekIN.txt</code> conţine pe prima linie numerele <code>n k</code>, reprezentând numărul de noduri ale grafului și nodul dat. Fiecare dintre următoarele linii conține câte o pereche de numere <code>i j</code>, cu semnificația că există muchie între <code>i</code> și <code>j</code>.

= Date de ieşire =
Fişierul de ieşire <code>gradekOUT.txt</code> va conţine pe prima linie numărul <code>p</code> de noduri determinate, iar pe linia a doua cele <code>p</code> noduri determinate, în ordine crescătoare.

= Restricţii şi precizări =

* <code>1 ≤ n ≤ 100</code>
* <code>1 ≤ i , j ≤ n</code>
* muchiile se pot repeta în fișierul de intrare

= Exemplul 1 =
<code>gradekIN.txt</code>
 5 3
 1 4 
 1 3 
 3 5 
 4 5 
 2 4 
 1 2 
 4 2 
 3 4 
<code>gradekOUT.txt</code>
 2
 1 3 

= Exemplul 2 =
<code>gradekIN.txt</code>
 5 101
 1 4 
 1 3 
 3 5 
 4 5 
 2 4 
 1 2 
 4 2 
 3 4 
<code>gradekOUT.txt</code>
 Datele corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3">
def gaseste_noduri_cu_grad_egal(n, k, muchii):
    grad = [0] * (n + 1)

    for i, j in muchii:
        grad[i] += 1
        grad[j] += 1

    noduri_cu_grad_egal = [nod for nod in range(1, n + 1) if grad[nod] == grad[k]]

    return noduri_cu_grad_egal

def citeste_date_intrare(file_path):
    try:
        with open(file_path, 'r') as f:
            n, k = map(int, f.readline().split())
            if not (1 <= n <= 100):
                raise ValueError("Datele corespund restrictiilor impuse")
            if not (1 <= k <= n):
                raise ValueError("Datele corespund restrictiilor impuse")
            muchii = [list(map(int, line.split())) for line in f.readlines()]
            for i, j in muchii:
                if not (1 <= i <= n) or not (1 <= j <= n):
                    raise ValueError("Datele corespund restrictiilor impuse")
    except ValueError as e:
        with open('gradekOUT.txt', 'w') as out_file:
            out_file.write(str(e) + '\n')
        exit()

    return n, k, muchii

try:
    n, k, muchii = citeste_date_intrare('gradekIN.txt')
    rezultat = gaseste_noduri_cu_grad_egal(n, k, muchii)
    with open('gradekOUT.txt', 'w') as f:
        f.write(str(len(rezultat)) + '\n')
        f.write(' '.join(map(str, rezultat)) + '\n')
except ValueError as e:
    pass

</syntaxhighlight>