leetcode6

# -*- coding:utf-8 -*-


# 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
# A P L S I I G
# 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
#
def loop_set(y,numRows,index_str,strs,dot,x):
    # print(index_str)
    if len(strs)-1==index_str:
        return dot
    if y==0:
        while y<=numRows-1:
            if len(strs) - 1 == index_str:
                return dot
            dot[y][x]=strs[index_str]
            index_str+=1
            y+=1
        y=y-1
    while y>1:
        y-=1
        x+=1
        if len(strs) - 1 == index_str:
            return dot
        dot[y][x]=strs[index_str]
        index_str+=1
    y=y-1
    x=x+1
    return loop_set(y,numRows,index_str,strs,dot,x)
def zigZag(strs,numRows):
    x=0
    index_str=0
    y=0
    dot=[['0' for j in range(len(strs))] for i in range(numRows)]
    return loop_set(y,numRows,index_str,strs,dot,x)
def print_zigzag(dot):
    for i in dot:
        for charac in i :
            if charac!='0':
                print(charac)
print_zigzag(zigZag("PAYPALISHIRING",4))
# print(zigZag("PAYPALISHIRING",4))