牛客网python算法练习（二）电话号码分身

坚持一周五题
继MIUI8推出手机分身功能之后，MIUI9计划推出一个电话号码分身的功能：首先将电话号码中的每个数字加上8取个位，然后使用对应的大写字母代替（"ZERO", "ONE", "TWO", "THREE","FOUR", "FIVE", "SIX", "SEVEN","EIGHT", "NINE"）， 然后随机打乱这些字母，所生成的字符串即为电话号码对应的分身。

输入描述:

第一行是一个整数T（1 ≤ T ≤ 100)表示测试样例数；接下来T行，每行给定一个分身后的电话号码的分身（长度在3到10000之间）。

输出描述:

输出T行，分别对应输入中每行字符串对应的分身前的最小电话号码（允许前导0）。

示例1

输入

4
EIGHT
ZEROTWOONE
OHWETENRTEO
OHEWTIEGTHENRTEO

输出

0
234
345
0345

解题思路主要是看字母出现的个数，比如EIGHT中的G在所有数字中只会出现一次

我的python代码如下，虽然没有优化内存，但也能通过case

#coding=utf-8
'''

继MIUI8推出手机分身功能之后，MIUI9计划推出一个电话号码分身的功能：首先将电话号码中的每个数字加上8取个位，然后使用对应的大写字母代替 （"ZERO", "ONE", "TWO", "THREE", "FOUR", "FIVE", "SIX", "SEVEN", "EIGHT", "NINE"）， 然后随机打乱这些字母，所生成的字符串即为电话号码对应的分身。
输入描述:
第一行是一个整数T（1 ≤ T ≤ 100)表示测试样例数；接下来T行，每行给定一个分身后的电话号码的分身（长度在3到10000之间）。
输出描述:
输出T行，分别对应输入中每行字符串对应的分身前的最小电话号码（允许前导0）。
示例1
输入

4
EIGHT
ZEROTWOONE
OHWETENRTEO
OHEWTIEGTHENRTEO

输出

0
234
345
0345


'''

# zi = ["ZERO", "ONE", "TWO", "THREE", "FOUR", "FIVE", "SIX", "SEVEN", "EIGHT", "NINE"]
def exit(s):
        a = [0]*10
    ####发现ZERO
        a[2] = s.count('Z')
    ####发现TWO
        a[4] = s.count('W')
    ####发现FOUR
        a[6] = s.count('U')
    ####发现SiX
        a[8] = s.count('X')
    ####发现eight
        a[0] = s.count('G')

    ####发现sSEVEN，s的次数减去six
        a[9] = s.count('S') - a[8]
    ########发现three，由于H只在 THREE和EIGHT中出现，所以计算THREE可以用H出现的次数减去G出现的次数
        a[5] = s.count('H') - a[0]
    ###发现ONE，由于O在zero、one、two、four中出现，所以可以通过计算o出现的次数减去zero、two、four
        a[3] = s.count('O') - a[2] - a[4] - a[6]
    ####发现Five，由于F只出现两次，计算F出现的次数减去Four
        a[7] = s.count('F') - a[6]
    ####发现Nine,I出现四次，计算I出现的次数减去FIVE，SIX、EIGHT
        a[1] = s.count('I') - a[7] - a[8] - a[0]
        s = ''
        for i in range(10):
            s += str(i)*a[i]
        return s
n = int(input())
for i in range(n):
    a_line.append(str(raw_input()))

for i in range(n):
    print exit(a_line[i])

看了一下官网的答案，总体思路差不多，但是减少了一些变量和计算，运行时间缩减一半，参考代码如下

num=int(raw_input())
n=[0]*10
for i in range(num):
    instr=raw_input()
#统计各位数字次数
    n[0]=instr.count('Z')
    n[2]=instr.count('W')
    n[4]=instr.count('U')
    n[6]=instr.count('X')
    n[8]=instr.count('G')
    n[1]=instr.count('O')-n[0]-n[2]-n[4]
    n[3]=instr.count('T')-n[2]-n[8]
    n[5]=instr.count('F')-n[4]
    n[7]=instr.count('S')-n[6]
    n[9]=instr.count('I')-n[5]-n[6]-n[8]
#对应数字位-8
    m=n[8:]+n[0:8]
    s=''
    for i in range(len(m)):
        s += str(i)*m[i]
    print s