题目描述:(中等难度)
Java代码
class Solution {
public String convert(String s, int numRows) {
if(numRows < 2) return s;
List<StringBuilder> rows = new ArrayList<StringBuilder>();
for(int i = 0; i < numRows; i++) rows.add(new StringBuilder());
int i = 0, flag = -1;
for(char c : s.toCharArray()) {
rows.get(i).append(c);
if(i == 0 || i == numRows -1) flag = - flag;
i += flag;
}
StringBuilder res = new StringBuilder();
for(StringBuilder row : rows) res.append(row);
return res.toString();
}
}
Python 代码
class Solution:
def convert(self, s: str, numRows: int) -> str:
if numRows <= 1 :
return s
res = ["" for _ in range(numRows)]
i,flag = 0,-1
for c in s:
res[i] += c
if i == 0 or i == numRows - 1:
flag = -flag
i += flag
return "".join(res)
s = Solution()
res = s.convert('LEETCODEISHIRING',2)
print(res)