牛客网 字符串加密

题目描述

有一种技巧可以对数据进行加密，它使用一个单词作为它的密匙。下面是它的工作原理：首先，选择一个单词作为密匙，如TRAILBLAZERS。如果单词中包含有重复的字母，只保留第1个，其余几个丢弃。现在，修改过的那个单词属于字母表的下面，如下所示：

A B C D E F G H I J K L M N O P Q R S T U V W X Y Z

T R A I L B Z E S C D F G H J K M N O P Q U V W X Y

上面其他用字母表中剩余的字母填充完整。在对信息进行加密时，信息中的每个字母被固定于顶上那行，并用下面那行的对应字母一一取代原文的字母(字母字符的大小写状态应该保留)。因此，使用这个密匙，Attack AT DAWN(黎明时攻击)就会被加密为Tpptad TP ITVH。

请实现下述接口，通过指定的密匙和明文得到密文。

详细描述：

接口说明

原型：

voidencrypt(char * key,char * data,char * encrypt);

输入参数：

char * key：密匙

char * data：明文

输出参数：

char * encrypt：密文

返回值：

void

输入描述:

先输入key和要加密的字符串

输出描述:

返回加密后的字符串

示例1

输入

nihao
ni

输出

le

def getKey(str_):　　#根据key去重
    nstr = ''
    for i in range(len(str_)):
        if(str_[i] not in list(nstr)):
            nstr += str_[i]
    return nstr

def Correspond(str_):　　#根据key得到对应的26个新key
    char = 'a'
    for i in range(26):
        if(char not in list(str_)):
            str_ += char

        char = chr(ord(char)+1)
    return str_

def getAllkeys():　　　　#得到26个顺序字母
    akey = ''
    char = 'a'
    for i in range(26):
        if(char not in list(akey)):
            akey += char
        char = chr(ord(char)+1)
    return akey

def getOut(str1,str2,str3):　　#根据输入和新key得到加密后的结果
    outstr = ''
    for i in range(len(str2)):
        index = str1.find(str2[i])
        temp = str3[index]
        outstr += temp
    return outstr



while True:
    try:
        key = raw_input()
        str_ = raw_input()
        nkey = getKey(key)
        nstr = Correspond(nkey)
        allkey = getAllkeys()
        printout = getOut(allkey,str_,nstr)
        print(printout)
    except:
        break