Python: 在不确定需要输入行数的情况下,逐行读取输入数据

HACKER RANK的一个编程题
import sys 

# Read input and assemble Phone Book
n = int(input())
phoneBook = {}
for i in range(n):
    contact = input().split(' ')
    phoneBook[contact[0]] = contact[1]

# Process Queries
lines = sys.stdin.readlines()
for i in lines:
    name = i.strip()
    if name in phoneBook:
        print(name + '=' + str( phoneBook[name] ))
    else:
        print('Not found')

使用sys模块的stdin.readlines可以读取屏幕上输入的所有行,在使用ctrl+d的时候结束。

 

你可能感兴趣的:(python)