Editing 4078 - Distanta

= Cerința =
Matei, care locuiește in orașul <code>p</code>, vrea să ajungă in orașul <code>q</code> să il viziteze. El are o hartă pe care se află <code>n</code> orașe și <code>m</code> drumuri prin care poate să treacă pentru a ajunge la destinație. Pe hartă apare și durata de timp <code>td</code> pentru fiecare drum, care reprezintă numărul de minute în care este parcurs drumul și <code>to</code> pentru traversarea orașelor (unele orașe sunt mai aglomerate, iar altele nu). Matei vrea să ajungă cat mai rapid la destinație, dar dacă există mai multe trasee de timp minim, il va alege pe acela care trece prin mai multe orașe. Nu vor exista trasee de timp minim cu același număr de orașe. Ajutati-l pe Matei să gasească drumul cerut.

= Date de intrare =
Fișierul de intrare <code>distantaIN.txt</code> conține pe prima linie numărul <code>n</code>, iar pe următoarele <code>n</code> linii se citesc <code>nume oraș</code> și <code>to</code> al acestuia. Se citește <code>m</code>, iar pe următoarele <code>m</code> linii <code>i</code>, <code>j</code>, <code>td</code> unde <code>i</code> și <code>j</code> reprezintă orașele care sunt legate prin drumul care se parcurge în <code>td</code>. Pe ultima linie se citesc <code>p</code> și <code>q</code>.

= Date de ieșire =
Fișierul de ieșire <code>distantaOUT.txt</code> va conține pe prima linie numărul <code>S</code>, reprezentând durata de timp în care se parcurge traseul cerut, iar pe următoarele linii se afișează traseul. Dacă nu există nici un traseu se va afișa <code>NU EXISTA TRASEU</code>.În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

= Restricții și precizări =

* <code>1 < n < 21</code>
* <code>0 ≤ m ≤ 190</code>
* <code>0 < numele orașelor (număr caractere) < 21</code>
* <code>0 ≤ numărul de drumuri dintre oricare 2 orașe ≤ 1</code>
* <code>0 ≤ S ≤ 2.000.000.001</code>
* Nu există drum de la un oraș la el însuși
* Dacă nu există nici un traseu se va afișa <code>NU EXISTA TRASEU</code>.

= Exemplul 1: =
<code>distantaIN.txt</code>
 7 
 Bistrita 14
 Oradea 80
 Beius 7
 Galatii-Bistritei 1
 Reghin 3
 Sibiu 15
 Cluj-Napoca 30
 10
 Bistrita Oradea 223
 Cluj-Napoca Sibiu 119
 Bistrita Cluj-Napoca 100
 Beius Oradea 50
 Bistrita Galatii-Bistritei 15
 Reghin Galatii-Bistritei 35
 Oradea Cluj-Napoca 100
 Sibiu Reghin 154
 Beius Sibiu 130
 Reghin Cluj-Napoca 133
 Bistrita Beius
<code>distantaOUT.txt</code>
 374
 Bistrita
 Galatii-Bistritei
 Reghin
 Sibiu
 Beius

= Exemplul 2: =
<code>distantaIN.txt</code>
 22
 Bistrita 14
 Oradea 80
 Beius 7
 Galatii-Bistritei 1
 Reghin 3
 Sibiu 15
 Cluj-Napoca 30
 10
 Bistrita Oradea 223
 Cluj-Napoca Sibiu 119
 Bistrita Cluj-Napoca 100
 Beius Oradea 50
 Bistrita Galatii-Bistritei 15
 Reghin Galatii-Bistritei 35
 Oradea Cluj-Napoca 100
 Sibiu Reghin 154
 Beius Sibiu 130
 Reghin Cluj-Napoca 133
 Bistrita Beius
<code>distantaOUT.txt</code>
 Datele nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
from queue import Queue

Inf = 0x3f3f3f3f

def check_restrictions(n, m, codif):
    if not (1 < n < 21) or not (0 <= m <= 190):
        return False

    for _, _, s in codif:
        if not (0 < len(s) < 21):
            return False

    return True
if __name__ == "__main__":
    with open("distantaIN.txt", "r") as infile, open("distantaOUT.txt", "w") as outfile:
        n = int(infile.readline().strip())
        codif = []
        G = [[] for _ in range(51)]
        t = [0] * 51
        timp = [0] * 51

        if not (1 < n < 21):
            outfile.write("Datele nu corespund restrictiilor impuse")
        else:
            for i in range(1, n + 1):
                line = infile.readline().split()
                if len(line) != 2:
                    outfile.write("Datele nu corespund restrictiilor impuse")
                    break

                s, cost = line
                cost = int(cost)
                codif.append((i, cost, s))

            m = int(infile.readline().strip())

            if not (0 <= m <= 190):
                outfile.write("Datele nu corespund restrictiilor impuse")
            else:
                for i in range(1, m + 1):
                    line = infile.readline().split()
                    if len(line) != 3:
                        outfile.write("Datele nu corespund restrictiilor impuse")
                        break

                    unu, doi, cost = line
                    cost = int(cost)
                    x = next(item[0] for item in codif if item[2] == unu)
                    y = next(item[0] for item in codif if item[2] == doi)
                    G[x].append((y, cost))
                    G[y].append((x, cost))

                unu, doi = infile.readline().split()
                start_node = next(item[0] for item in codif if item[2] == doi)

                def decodif(oras):
                    for x, y, s in codif:
                        if s == oras:
                            return x

                def costnod(nod):
                    for x, y, s in codif:
                        if nod == x:
                            return y

                def afisare(nod):
                    for x, y, s in codif:
                        if nod == x:
                            return s

                def bfs(nod):
                    t[nod] = nod
                    timp[nod] = costnod(nod)
                    q = Queue()
                    q.put(nod)

                    while not q.empty():
                        x = q.get()

                        for i, c in G[x]:
                            costnou = timp[x] + c + costnod(i)

                            if not timp[i] or costnou <= timp[i]:
                                timp[i] = costnou
                                t[i] = x
                                q.put(i)

                bfs(start_node)

                if timp[decodif(unu)] != 0:
                    outfile.write(str(timp[decodif(unu)]) + '\n')

                    plec = decodif(unu)
                    while plec != t[plec]:
                        outfile.write(afisare(plec) + '\n')
                        plec = t[plec]

                    outfile.write(afisare(decodif(doi)))
                else:
                    outfile.write("NU EXISTA TRASEU")

</syntaxhighlight>