字符串加密

字符串加密

描述
有一种技巧可以对数据进行加密，它使用一个单词作为它的密匙。下面是它的工作原理：首先，选择一个单词作为密匙，如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。

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

本题有多组输入数据。

数据范围：1<=n<=100 ，保证输入的字符串中仅包含小写字母

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

输出描述：
返回加密后的字符串

示例1
输入：
nihao
ni
输出：
le

#include<iostream>
#include<set>
using namespace std;

int main(){
    string str;
    while(cin>>str){
        string str1;
        cin>>str1;
        set<char> st;
        string key="";
        const int k='a'-'A';
        for(int i=0;i<str.size();++i){
            char tmp=str[i];
            if(tmp>='A' && tmp<='Z') tmp+=k;
            if(st.find(tmp)==st.end()){
                st.insert(tmp);
                key+=tmp;
            }
        }
        for(int i=0;i<26;++i){
            char tmp1='a'+i;
            if(st.find(tmp1)==st.end()){
                key+=tmp1;
            }
        }
        for(int i=0;i<str1.size();++i){
            if(str1[i]>='a' && str1[i]<='z'){
                int N=str1[i]-'a';
                str1[i]=key[N];
            }
            else if(str1[i]>='A' && str1[i]<='Z'){
                int N=str1[i]-'A';
                str1[i]=key[N]-k;
            }
        }
        cout<<str1<<endl;
    }
    return 0;
}