【算法题】Z字形变换 by python

题目:

将字符串 "PAYPALISHIRING" 以Z字形排列成给定的行数:

P   A   H   N
A P L S I I G
Y   I   R

之后从左往右,逐行读取字符:"PAHNAPLSIIGYIR"

实现一个将字符串进行指定行数变换的函数:

string convert(string s, int numRows);

示例 1:

输入: s = "PAYPALISHIRING", numRows = 3
输出: "PAHNAPLSIIGYIR"

示例 2:

输入: s = "PAYPALISHIRING", numRows = 4
输出: "PINALSIGYAHRPI"
解释:

P     I    N
A   L S  I G
Y A   H R
P     I

分析:

刚开始看题目的时候被两列之间的空格给迷惑了,后面看了网上其他人的分析,知道了不需要管空格。这个采用二维数组,然后慢慢填充就行了,一次遍历填充一行中的某一个位置,然后换到下一行。

代码:

ef convert(s, numRows):
    """
    :type s: str
    :type numRows: int
    :rtype: str
    """
    if(numRows == 1): return s
    a = [[]for i in range(numRows)]#产生一个有numrows单元的list,每个list可以在后面添加元素
    r = 0
    direct = 1 #行前进的方向是向上还是向下
    for c in s:
        a[r].append(c)
        if r >=numRows-1:
            direct = -1
        elif r == 0:
            direct = 1
        r +=direct
    answer = ""
    for row in a:#按行优先打印出来
        for col in row:
            answer += col
    return answer
if __name__ == "__main__":
    print(convert("PAYPALISHIRING",4))

阅读更多
文章标签: 算法题 python
个人分类: 算法题
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭