[Codewars] 098: Vigenère Cipher Helper

题目

The Vigenère cipher is a classic cipher originally developed by Italian cryptographer Giovan Battista Bellaso and published in 1553. It is named after a later French cryptographer Blaise de Vigenère, who had developed a stronger autokey cipher (a cipher that incorporates the message of the text into the key).

The cipher is easy to understand and implement, but survived three centuries of attempts to break it, earning it the nickname "le chiffre indéchiffrable" or "the indecipherable cipher."

From Wikipedia:

The Vigenère cipher is a method of encrypting alphabetic text by using a series of different Caesar ciphers based on the letters of a keyword. It is a simple form of polyalphabetic substitution.

. . .

In a Caesar cipher, each letter of the alphabet is shifted along some number of places; for example, in a Caesar cipher of shift 3, A would become D, B would become E, Y would become B and so on. The Vigenère cipher consists of several Caesar ciphers in sequence with different shift values.

Assume the key is repeated for the length of the text, character by character. Note that some implementations repeat the key over characters only if they are part of the alphabet -- this is not the case here.

The shift is derived by applying a Caesar shift to a character with the corresponding index of the key in the alphabet.

Visual representation:

"my secret code i want to secure"  // message
"passwordpasswordpasswordpasswor"  // key

Write a class that, when given a key and an alphabet, can be used to encode and decode from the cipher.

Example

var alphabet = 'abcdefghijklmnopqrstuvwxyz';
var key = 'password';

// creates a cipher helper with each letter substituted
// by the corresponding character in the key
var c = new VigenèreCipher(key, alphabet);

c.encode('codewars'); // returns 'rovwsoiv'
c.decode('laxxhsj');  // returns 'waffles'

Any character not in the alphabet must be left as is. For example (following from above):

c.encode('CODEWARS'); // returns 'CODEWARS'

维吉尼亚密码的加密和解密程序

我的答案

# coding: utf-8
class VigenereCipher(object):
    def __init__(self, key, alphabet):
        self.key = key
        self.n = len(key)
        self.N = len(alphabet)
        self.dic1 = dict(zip(alphabet, list(range(self.N))))
        self.dic2 = dict(zip(list(range(self.N)), alphabet))
    
    def encode(self, text):
        if self.N == 0 or self.n == 0:
            return text
        else:
            res = ''
            text = text
            for i in range(len(text)):
                if text[i] in self.dic1:
                    shift = self.dic1[self.key[i%self.n]]
                    res += self.dic2[(self.dic1[text[i]] + shift) % self.N]
                else:
                    res += text[i]
            return res
    
    def decode(self, text):
        if self.N == 0 or self.n == 0:
            return text
        else:
            res = ''
            text = text
            for i in range(len(text)):
                if text[i] in self.dic1:
                    shift = self.dic1[self.key[i%self.n]]
                    res += self.dic2[(self.dic1[text[i]] - shift + self.N) % self.N]
                else:
                    res += text[i]
            return res

我的答案

from itertools import cycle

class VigenereCipher (object):
    def __init__(self, key, alphabet):
        self.key = key.decode('utf-8')
        self.alphabet = alphabet.decode('utf-8')
    
    def cipher(self, mode, str):
        return ''.join(self.alphabet[(self.alphabet.index(m) +
                  mode * self.alphabet.index(k)) % len(self.alphabet)]
                  if m in self.alphabet else m for m, k in zip(str.decode('utf-8'),
                  cycle(self.key))).encode('utf-8')
    
    def encode(self, str): return self.cipher(1, str)
    def decode(self, str): return self.cipher(-1, str)