Comment lire plusieurs lignes d'entrée en python

Je trouve votre programme difficile à comprendre. Donc, je l'ai réécrit, et je pense que ma version est un peu plus facile à comprendre.

import sys
import numpy as np


line = raw_input()
max_val, num_paths = (int(n) for n in line.split())


# a will be a list of tuples of int, taken from the input.
#
# Each tuple represents a path, so this is effectively a sparse representation
# of a square matrix of possible paths.
#
# Input city numbers are 1-based, but we will treat them as 0-based, so
# subtract 1 from each value before appending to array a.

a = []
for _ in xrange(num_paths):
    line = raw_input()

    # TRICKY: subtract 1 to convert from 1-based to 0-based city numbers
    tup = tuple(int(n)-1 for n in line.split())

    if len(tup) != 2:
        raise ValueError, "input should only have two values per line"
    for n in tup:
        if not 0 <= n < max_val:
            raise ValueError, "value must be in range [1, %d]" % max_val
        if tup[0] >= tup[1]:
            #raise ValueError, "INFINITE PATHS"
            print "INFINITE PATHS"
            sys.exit(0)
    a.append(tup)


# Expand the sparse matrix representation into an actual square matrix.
# It should have a 1 anywhere a path was indicated with a tuple in list a,
# and a 0 everywhere else.
b = [ [0 for _ in xrange(max_val)] for _ in xrange(max_val)]
for i, j in a:
    b[i][j] = 1


c = 0
for i in xrange(num_paths):
    d = np.linalg.matrix_power(b, i + 1)
    c += d[0][max_val - 1]
print c

Ma version n'a imprimer 2 quand donné l'exemple d'entrée.

Voici certaines des choses que j'ai compris, j'ai travaillé sur ce:

La première ligne nous donne des constantes (N et M dans la documentation, représentant le max de valeur juridique et le nombre de chemins respectivement). Vous devez enregistrer ces valeurs dans des variables avec des noms, plutôt que de les mettre dans une liste et de se référer à eux par l'indice de la liste. J'ai utilisé les noms de max_val et num_paths. Vous avez vous-même fait une erreur: vous êtes censé trouver les chemins de la ville de 1 pour la ville de N, de sorte que le chèque à la fin devrait être d[0][max_val - 1]; vous avez utilisé l[1] qui est num_paths plutôt que l[0].

b doit être une matrice carrée. Votre code a été réglage de la largeur en fonction de la longueur de a, mais max_val et num_paths peut-être pas toujours égale, de sorte que, d'une manière dangereuse pour le faire.

Il est étrange de faire une boucle sur chaque point de la matrice carrée et vérifier pour voir si elle doit être définie en tant que 1 ou pas. Il est également très inefficace, en particulier parce que les in test est O(n) où n est la longueur du tableau a. Au lieu de cela, construire la place vide de la matrice, et puis tout simplement en boucle sur les chemins d'accès et définissez les valeurs 1 par chemin.

De même, il est étrange de valider les valeurs d'entrée dans la boucle qui initialise la matrice carrée; il est préférable de valider les valeurs d'entrée lorsqu'elles sont lues dans la boucle d'entrée. Et encore, il est dangereux, parce que num_paths peut-être rien à voir avec max_val. Aussi il est inefficace, parce que vous avez le contrôle a[i-1][0] contre a[i-1][1] une fois par colonne dans b; que la comparaison n'utilise pas la valeur j à tous. Vous faisiez chaque case à cinq fois; il suffit de faire chaque case à la fois.

Il y a un Python, un langage que j'ai utilisé, où vous pouvez utiliser _ (un seul trait de soulignement) comme le nom d'une variable lorsque vous n'avez pas de soins sur la valeur de cette variable. Lorsque nous sommes juste faire quelque chose d'un certain nombre de fois avec une boucle, et nous ne serons pas à l'aide de la boucle de contre-valeur à quelque chose, j'ai utilisé un _ que la variable de compteur de boucle. Ce n'est pas essentiel, bien sûr.

Pour répondre à votre question: je ne vois pas de moyen possible pour que votre programme de ne pas produire de sortie. Je soupçonne qu'il pourrait y avoir un problème sur le serveur qui exécute ce test de problème. Votre programme doit toujours imprimer "INFINI CHEMINS" ou bien une sorte de valeur de type entier.

P. S. je n'ai pas vraiment à comprendre comment le programme fonctionne; la description du problème, vous devez fournir un certain nombre de chemins modulo 1e9, et je ne vois rien de la faire respecter.

OriginalL'auteur steveha

si vous avez l'entrée dans un fichier input.txt dans le même dossier que le script:

with open("input.txt") as f:
    l = [int(i) for i in f.readline().split(" ")]
    a = []
    for line in f.readlines():
        a.append([int(i) for i in line.strip().split(" ")])
print(l, a)

si l'entrée est passée comme argument de ligne de commande:

import sys
input_string = sys.argv[1]
print(input_string) # test if this prints the input
...

OriginalL'auteur Aprillion