0

我在试图弄清楚如何让我的解密功能工作时遇到了很多麻烦。它适用于味精可被 3 整除的理想情况，但在那之后我完全迷失了。我对我必须做什么有一个模糊的想法，因为我能够让它与两个导轨一起工作，但是三个导轨有更多的可能性。我很迷茫:(所有这些打印语句也只是为了帮助我弄清楚发生了什么。

import sys

def main():
    plaintext="abcdefgh"
    print(threeRailEncrypt(plaintext))
    print(threeRailDecrypt(threeRailEncrypt(plaintext)))


def threeRailEncrypt(plaintext):
    ciphertext=""
    rail1=""
    rail2=""
    rail3=""

    for i in range(len(plaintext)):
        if i%3 == 0:
            rail1=rail1+plaintext[i]
        elif i%3 == 1:
            rail2=rail2+plaintext[i]
        else:
            rail3=rail3+plaintext[i]

    ciphertext=rail1+rail2+rail3

    return(ciphertext)

def threeRailDecrypt(msg):
    if len(msg)%3==0:
        third=len(msg)//3
        print(third)
        rail1=msg[:third]
        rail2=msg[third:third*2]
        rail3=msg[third*2:]
        print(rail1,rail2,rail3)
        dm=""
        for i in range(third):
            dm=dm+rail1[i]
            dm=dm+rail2[i]
            dm=dm+rail3[i]
    else:
        third=(len(msg)//3)+1
        print(third)
        rail1=msg[:third]
        rail2=msg[third:third*2]
        rail3=msg[third*2:]
        print(rail1,rail2,rail3)
        dm=""
        for i in range(third):
            dm=dm+rail1[i]
            print(dm)
            dm=dm+rail2[i]
            print(dm)
            dm=dm+rail3[i]
            print(dm)
            if  len(rail2)>len(rail3):
                dm=dm+rail2[-1]
        return(dm)
main()
4

1 回答 1

2

您可以使用 stride 对字符串（或任何其他序列）进行切片；你可以选择每第三个字符：

def threeRailEncrypt(plaintext):
    return plaintext[::3] + plaintext[1::3] + plaintext[2::3]

为了扭转这种趋势，将密文分成三块，itertools.zip_longest()然后重新组合它们：

from itertools import zip_longest, chain

def threeRailDecrypt(ciphertext):
    stride, remainder = divmod(len(ciphertext), 3)
    # how large was each of the three sections?
    sizes = [stride] * 3
    for i in range(remainder):
        sizes[i] += 1
    # slice ciphertext up into 3 sections again
    pos = 0
    pieces = []
    for s in sizes:
        pieces.append(ciphertext[pos:pos + s])
        pos += s
    # recombine the triplets
    return ''.join(chain.from_iterable(zip_longest(*pieces, fillvalue='')))

演示：

>>> threeRailEncrypt('Foo bar baz and all')
'F raa lob znaoab dl'
>>> threeRailDecrypt(threeRailEncrypt('Foo bar baz and all'))
'Foo bar baz and all'

没有的最后一部分zip_longest是：

# recombine the triplets
plaintext = []
for i in range(len(parts[0])):
    for part in parts:
        try:
            plaintext.append(part[i])
        except IndexError:  # part too short
            pass
return ''.join(plaintext)
于 2013-10-03T00:01:29.207 回答