playfair加密5行5列代码

以下是 Playfair 加密的 Python 代码实现，使用 5 行 5 列的矩阵进行加密：

def generate_table(key):
    # 生成 Playfair 加密所需的 5x5 矩阵
    # 先将密钥去重并转换为大写字母
    key = ''.join(sorted(set(key.upper()), key=key.upper().index))
    # 去除 J 并将 I 转换为 J
    key = key.replace('J', 'I')
    # 将剩下的字母填充到矩阵中
    table = []
    for c in key:
        if c not in table:
            table.append(c)
    for c in 'ABCDEFGHIKLMNOPQRSTUVWXYZ':
        if c not in table:
            table.append(c)
    # 将矩阵转换为 5x5 的二维列表
    table = [table[i:i+5] for i in range(0, 25, 5)]
    return table

def encrypt(plaintext, key):
    # 对明文进行 Playfair 加密
    table = generate_table(key)
    # 将明文中的 J 转换为 I
    plaintext = plaintext.upper().replace('J', 'I')
    # 在相邻重复字母之间插入 X
    plaintext = ''.join([plaintext[i] + 'X' if plaintext[i] == plaintext[i+1] else plaintext[i] for i in range(0, len(plaintext)-1, 2)] + [plaintext[-1]])
    # 如果明文长度为奇数，在末尾加一个 X
    if len(plaintext) % 2 == 1:
        plaintext += 'X'
    # 分组并加密每一组
    ciphertext = ''
    for i in range(0, len(plaintext), 2):
        a, b = plaintext[i], plaintext[i+1]
        row_a, col_a = next((r, c) for r, row in enumerate(table) for c, letter in enumerate(row) if letter == a)
        row_b, col_b = next((r, c) for r, row in enumerate(table) for c, letter in enumerate(row) if letter == b)
        if row_a == row_b:
            # 在同一行，将每个字母替换为该行中相邻的字母
            a = table[row_a][(col_a+1)%5]
            b = table[row_b][(col_b+1)%5]
        elif col_a == col_b:
            # 在同一列，将每个字母替换为该列中相邻的字母
            a = table[(row_a+1)%5][col_a]
            b = table[(row_b+1)%5][col_b]
        else:
            # 不在同一行或同一列，将每个字母替换为另一字母所在行列的字母
            a = table[row_a][col_b]
            b = table[row_b][col_a]
        ciphertext += a + b
    return ciphertext

使用示例：

key = 'playfair example'
plaintext = 'HELLO WORLD'
ciphertext = encrypt(plaintext, key)
print(ciphertext)  # 输出：DMTRQZBDMG