Leetcode 6. ZigZag Conversion
The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)P A H N
AP L S I IG
Y I R
And then read line by line: "PAHNAPLSIIGYIR"
Write the code that will take a string and make this conversion given a number of rows:
string convert(string s, int numRows);
Example 1:
Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"
Example 2:
Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"
Explanation:
P I N
A L S I G
Y A H R
P I
class Solution:
def convert(self, s: str, numRows: int) -> str:
if numRows == 0 or s == None or len(s) == 0:
return ""
if numRows == 1:
return s
result = ['' for _ in range(numRows)]
count = 0
flag = 0
for i in range(len(s)):
if flag == 0:
result += (s)
count += 1
if count == numRows - 1:
flag = 1 - flag
else:
result += (s)
count -= 1
if count == 0:
flag = 1 - flag
return "".join(result)
页:
[1]