-
Notifications
You must be signed in to change notification settings - Fork 0
/
new_encryption_algo.py
59 lines (50 loc) · 1.26 KB
/
new_encryption_algo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/bin/python3
import math
import os
import random
import re
import sys
#
# Complete the 'encryption' function below.
#
# The function is expected to return a STRING.
# The function accepts STRING s as parameter.
#
def get_r_c(s):
s = s.replace(' ', '')
if int(math.sqrt(len(s))) * int(math.sqrt(len(s))) == int(len(s)):
return int(math.sqrt(len(s))) ,int(math.sqrt(len(s)))
if int(math.sqrt(len(s))) * int(math.sqrt(len(s))) + 1 < len(s):
return int(math.sqrt(len(s))) + 1, int(math.sqrt(len(s))) + 1
else:
return int(math.sqrt(len(s))), int(math.sqrt(len(s))) + 1
def encryption(s):
r, c = get_r_c(s)
z = 0
s = s.replace(' ', '')
data = []
print('this is s : %s', s)
for i in range(int(len(s) / c)):
data.append(s[z:z+c])
z = z + c
if z < len(s):
data.append(s[z:])
i = 0
res = ''
while i <= c - 1:
for x in data :
if i < len(x):
res += x[i]
else:
res += ''
res += ' '
i += 1
print(res)
return res
if __name__ == '__main__':
fptr = open(os.environ['OUTPUT_PATH'], 'w')
s = input()
result = encryption(s)
print(result)
fptr.write(result + '\n')
fptr.close()