Editing 0591 - Firma

= Cerința =
Într-o țară sunt <code>n</code> orașe, numerotate de la <code>1</code> la <code>n</code>, unite între ele prin <code>m</code> șosele bidirecționale de lungimi cunoscute, între oricare două orașe existând drum, fie șosea directă, fie prin alte orașe. O firmă dorește să-și stabilească sediul în unul dintre orașe, astfel încât suma lungimilor drumurilor minime de la orașul în care se află sediul la toate celelaltele orașe să fie minimă. Determinați orașul care va fi ales pentru sediul firmei. Dacă sunt mai multe orașe care pot fi alese, se va alege cel cu numărul de ordine mai mic.

= Date de intrare =
Fișierul de intrare <code>firmaIN.txt</code> conține pe prima linie numerele <code>n m</code>, iar următoarele <code>m</code> linii câte trei numere <code>i j L</code>, cu semnificatia: între orașele <code>i</code> și <code>j</code> există o șosea directă de lungime <code>L</code>.

= Date de ieșire =
Fișierul de ieșire <code>firmaOUT.txt</code> va conține pe prima linie numărul <code>X</code>, reprezentând numărul de ordine al orașului care va ales ca sediu pentru firmă.

= Restricții și precizări =

* <code>1 ≤ n ≤ 100</code>

= Exemplu: =
<code>firmaIN.txt</code>
 6 9
 1 3 5
 1 4 5
 1 5 4
 2 3 5
 2 4 1
 2 5 2
 3 6 2
 4 6 4
 5 6 4
<code>firmaOUT.txt</code>
 2

== Rezolvare ==
<syntaxhighlight lang="python" line="1">
import heapq

def check_restrictions(n, m):
    if not (1 <= n <= 100):
        with open("firmaOUT.txt", "w") as fout:
            fout.write("Datele nu corespund restrictiilor impuse")
        return False
    return True

def dijkstra(nod, n, G):
    D = [float('inf')] * (n + 1)
    D[nod] = 0
    Q = [(0, nod)]
    while Q:
        x, y = heapq.heappop(Q)
        if x > D[y]:
            continue
        for nodnou, costnou in G[y]:
            if D[nodnou] > D[y] + costnou:
                D[nodnou] = D[y] + costnou
                heapq.heappush(Q, (D[nodnou], nodnou))
    return D

def main():
    with open("firmaIN.txt", "r") as fin:
        n, m = map(int, fin.readline().split())
        
        if not check_restrictions(n, m):
            return

        G = [[] for _ in range(n + 1)]
        for _ in range(m):
            x, y, w = map(int, fin.readline().split())
            G[x].append((y, w))
            G[y].append((x, w))

    minim = float('inf')
    mini = -1
    for i in range(1, n + 1):
        s = 0
        D = dijkstra(i, n, G)
        for j in range(1, n + 1):
            if D[j] != float('inf'):
                s += D[j]

        if s < minim:
            minim = s
            mini = i

    with open("firmaOUT.txt", "w") as fout:
        fout.write(str(mini))

if __name__ == "__main__":
    main()

</syntaxhighlight>