[USACO] 1.2.3 Name That Number 命名那个数字 Python

1.2.3 Name That Number 命名那个数字

(namenum.pas/c/cpp)

在威斯康辛州牛大农场经营者之中，都习惯于请会计部门用连续数字给母牛打上烙印。但是,母牛本身并没感到这个系统的便利,它们更喜欢用它们喜欢的名字来呼叫它们的同伴，而不是用像这个的语句"C'mon, #4734, get along."。请写一个程序来帮助可怜的牧牛工将一只母牛的烙印编号翻译成一个可能的名字。因为母牛们现在都有手机了，使用标准的按键的排布来把将数目翻译为文字:( 除了 "Q" 和 "Z")

        2: A,B,C     5: J,K,L    8: T,U,V
        3: D,E,F     6: M,N,O    9: W,X,Y
        4: G,H,I     7: P,R,S

可接受的名字都被放在这样一个叫作"dict.dic" 的文件中，它包含一连串的少于 5,000个（准确地说是4617个）可被接受的牛的名字。 (所有的名字都是大写的且已按字典序排列) 请读入母牛的编号并返回那些能从编号翻译出来并且在字典中的名字。举例来说,编号 4734 能产生的下列各项名字: GPDG GPDH GPDI GPEG GPEH GPEI GPFG GPFH GPFI GRDG GRDH GRDI GREG GREH GREI GRFG GRFH GRFI GSDG GSDH GSDI GSEG GSEH GSEI GSFG GSFH GSFI HPDG HPDH HPDI HPEG HPEH HPEI HPFG HPFH HPFI HRDG HRDH HRDI HREG HREH HREI HRFG HRFH HRFI HSDG HSDH HSDI HSEG HSEH HSEI HSFG HSFH HSFI IPDG IPDH IPDI IPEG IPEH IPEI IPFG IPFH IPFI IRDG IRDH IRDI IREG IREH IREI IRFG IRFH IRFI ISDG ISDH ISDI ISEG ISEH ISEI ISFG ISFH ISFI

碰巧，81个中只有一个"GREG"是有效的(在字典中)。

写一个程序来对给出的编号打印出所有的有效名字，如果没有则输出NONE。编号可能有12位数字。

格式

PROGRAM NAME: namenum

INPUT FORMAT:

(file namenum.in)

单独的一行包含一个编号(长度可能从1到12)。

OUTPUT FORMAT:

(file namenum.out)

以字典顺序输出一个有效名字的不重复列表，一行一个名字。 如果没有有效名字，输出'NONE'。

SAMPLE INPUT

4734 

SAMPLE OUTPUT

GREG

时间限制: 1000ms
空间限制: 128MB

=====解析=====

⨂"dict.txt" 请在USACO training网站查看
USACO Training Gateway

思路：
先把所有的名字组合全部组合出来，再到 "dict.txt"中去寻找

这个问题用Python做的话很容易时间超时（例如使用 A in B 的语法）

但是这里有一个更简单的方式，就是使用 set

把组合名字和所有名字都转换为set，然后找交集（A∩B）

后面又转会list的原因是：
程序在我的电脑是正常运行，但在USACO的网站会输出先后顺序出错
将 set 转换回 list 即可解决这个问题

=====Code=====

fin= open('namenum.in','r')
fout= open('namenum.out','w')
lines=fin.read().splitlines()
num=[x for x in lines[0]]
name_dic={'2':['A','B','C'],
          '3':['D','E','F'],
          '4':['G','H','I'],
          '5':['J','K','L'],
          '6':['M','N','O'],
          '7':['P','R','S'],
          '8':['T','U','V'],
          '9':['W','X','Y']}

def num_to_name(num):
    name=[]
    name.extend(name_dic[num[0]])
    counter=1
    while counter<len(num) and name:
        number=name[:] 
        name.clear()
        for i in number:
            for j in name_dic[num[counter]]:
                name.append(f'{i}{j}')
        counter+=1
    return name

def find_name(num):
    name=set(num_to_name(num))
    with open('dict.txt','r') as f:
        content = [line.strip() for line in f]
    content = set(content)
    same = list(name.intersection(content))
    same.sort()
    if same:
        for i in same:
            fout.write(str(i)+'\n')
    else:
        fout.write('NONE\n')
    fout.close()

find_name(num)