playfair密码加密pycharm

以下是使用 Playfair 密码进行加密的 Python 代码示例：

def generate_playfair_table(key):
    # 生成 Playfair 表
    key = key.replace(" ", "").upper()
    alphabet = "ABCDEFGHIKLMNOPQRSTUVWXYZ"
    table = ""
    for char in key:
        if char not in table:
            table += char
    for char in alphabet:
        if char not in table:
            table += char
    return table

def clean_message(message):
    # 清理消息中的无效字符
    message = message.upper().replace(" ", "").replace("J", "I")
    cleaned_message = ""
    i = 0
    while i < len(message):
        cleaned_message += message[i]
        if i + 1 < len(message) and message[i] == message[i + 1]:
            cleaned_message += "X"
        i += 1
    if len(cleaned_message) % 2 == 1:
        cleaned_message += "X"
    return cleaned_message

def encrypt(message, key):
    # 加密消息
    table = generate_playfair_table(key)
    message = clean_message(message)
    ciphertext = ""
    for i in range(0, len(message), 2):
        a = message[i]
        b = message[i + 1]
        a_index = table.index(a)
        b_index = table.index(b)
        a_row = a_index // 5
        a_col = a_index % 5
        b_row = b_index // 5
        b_col = b_index % 5
        if a_row == b_row:
            a_col = (a_col + 1) % 5
            b_col = (b_col + 1) % 5
        elif a_col == b_col:
            a_row = (a_row + 1) % 5
            b_row = (b_row + 1) % 5
        else:
            a_col, b_col = b_col, a_col
        ciphertext += table[a_row * 5 + a_col]
        ciphertext += table[b_row * 5 + b_col]
    return ciphertext

你可以使用以下代码测试加密函数：

key = "MYKEY"
message = "HELLO WORLD"
ciphertext = encrypt(message, key)
print(ciphertext)

输出应该是 "DMTRTQSPHT"。