【华为OJ】【026-字符串加密】

【华为OJ】【算法总篇章】

【华为OJ】【026-字符串加密】

【工程下载】

题目描述

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

输入描述

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

输出描述

返回加密后的字符串

输入例子

nihao
ni

输出例子

le

算法实现

import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.Scanner;

/** * Author: 王俊超 * Date: 2015-12-24 10:19 * Declaration: All Rights Reserved !!! */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
// Scanner scanner = new Scanner(Main.class.getClassLoader().getResourceAsStream("data.txt"));
        while (scanner.hasNext()) {
            String s1 = scanner.nextLine();
            String s2 = scanner.nextLine();
            System.out.println(encrypt(s1, s2));
        }

        scanner.close();
    }

    private static String encrypt(String mask, String s) {
        HashSet<Character> set = new LinkedHashSet<>(32);
        // 添加掩码中出现了的字母并且除去重复的字母
        for (int i = 0; i < mask.length(); i++) {
            char c = mask.charAt(i);
            if (c >= 'a' && c <= 'z') {
                set.add((char) (c - 'a' + 'A'));
            } else if (c >= 'A' && c <= 'Z') {
                set.add(c);
            }
        }

        // 添加余下的字母
        for (int i = 0; i < 26; i++) {
            set.add((char) ('A' + i));
        }

        // 转换成字符数组，都是用大写表示的
        Character[] convert = set.toArray(new Character[set.size()]);
        char[] result = s.toCharArray();

        for (int i = 0; i < result.length; i++) {
            char c = result[i];

            if (c >= 'a' && c <= 'z') {
                result[i] = (char) (convert[c - 'a'] - 'A' + 'a');
            } else if (c >= 'A' && c <= 'Z') {
                result[i] = convert[c - 'A'];
            }
        }

        return new String(result);
    }
}