Oros Ioana Diana: Pagină nouă: = Cerința = Într-o țară sunt `n` orașe, numerotate de la `1` la `n`, unite între ele prin `m` ș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ă....

2024-05-18T20:24:46Z

Pagină nouă: = 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ă....

New page

= 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>

0591 - Firma - Revision history