用Python做50道ACM之《Hay Points》

7.Hay Points

http://acm.fzu.edu.cn/problem.php?pid=1008

import sys

def hay_points(discribtions,haypoint_dict):
    money=0
    for i in range(0,len(discribtions)):
        disc=discribtions[i]
        if disc in haypoint_dict.keys():
            money+=haypoint_dict[disc]
    print(money)
    return None

line_at=0
while True:
    read_in=sys.stdin.readline()
    if not read_in:
        break
    if line_at==0:
        disc_num,posi_num=tuple([int(x) for x in read_in.rstrip().split(' ')])
        hp_dict={}
        disc_text_list=[]
        line_at+=1
    elif line_at<=disc_num:
        key,value=tuple([x for x in read_in.rstrip().split(' ')])
        hp_dict[key]=eval(value)
        line_at+=1
    else:
        if read_in.rstrip() != '.':
            disc_text_list.extend([x for x in read_in.rstrip().split(' ')])
        else:
            # process
            hay_points(disc_text_list,hp_dict)
            posi_num-=1
            disc_text_list=[] # 重新初始化描述文本单词列表
            if posi_num==0:
                break
            else:
                continue