字符串加密

題目描述

有一種技巧能夠對數據進行加密，它使用一個單詞做爲它的密匙。下面是它的工做原理：
首先，選擇一個單詞做爲密匙，如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;

/**
 * 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);
    }
}