Сдача лабы номер 2

[DNPotapov]

#a=97
#z=122

def asci_num(a, b = None):
    if b:
        return ord(a) - 96 + ord(b) - 96
    return ord(a) - 96

def sort_(lst):
    count_mass = [0] * 676
    res = []
    for i in range(676):
        count_mass[i] = []
    for item in lst:
        if len(item) == 1:
            count_mass[asci_num(item[0])].append(item)
        else:
            count_mass[asci_num(item[0], item[1])].append(item)

    for i in range(len(count_mass)):
        for item in count_mass[i]:
            res.append(item)
    print(' '.join(res))

if __name__ == '__main__':
    a = 'london is a capital of great britan aa zz az z'
    sort_(a.split())