HJ36 字符串加密

描述

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

AC code

/// HJ36 字符串加密
#include <bits/stdc++.h>
using namespace std;

int main()
{
    string input;
    getline(cin, input);
    
    // 建立密文字母表key
    int cnt[26] = {0};
    string key;
    
    // 去重，加入密文字母表表头
    for(int i=0; i<input.size(); i++)
    {
        if(cnt[input[i]-'a']==0)
        {
            cnt[input[i]-'a'] += 1;
            key.push_back(input[i]);
        }
    }
    
    // 按顺序加入表头没有的字母
    for(int i=0; i<26; i++)
    {
        if(cnt[i]==0)
            key.push_back((char)('a'+i));
    }
    
    // 明文转换
    string str;
    getline(cin, str);
    for(int i=0; i<str.size(); i++)
    {
        cout<<key[str[i]-'a'];
    }

    return 0;
}