力扣题(6) —— z字形变换

z字形变换

题目描述

将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 “PAYPALISHIRING” 行数为 3 时,排列如下:

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

示例 3:

输入:s = "A", numRows = 1
输出:"A"

提示:

1 <= s.length <= 1000
s 由英文字母(小写和大写)、',''.' 组成
1 <= numRows <= 1000

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zigzag-conversion

做法一

一开始真的不知道怎么做,那就只好把问题写下来找规律了,为了规律可见,我们将字符串的索引进行排序。如图:力扣题(6) —— z字形变换_第1张图片
我的建议是找到规律后自己实现一下代码,说实话我自己的代码也写得很乱(毕竟是小白)

s = input()

numRows = int(input())

def convert(s, numRows):
    T = (numRows-1) * 2 # 周期

    if T == 0: # 此时 numRows == 1,其实就不用变化
        return s
    else:
        res = ''
        count = 0 # 表示是第几行
        while count < numRows:
            index = count # index 表示索引
            flag = True
            while flag:
                if count == 0 or count == numRows-1: # 第 0 行和最后一行不用考虑 V 字的右边
                    if index < len(s): # 判断是否越界
                        res += s[index]
                        index += T
                    else:
                        break
                 # 剩下的条件语句,都是考虑可能出现的情况
                elif (index - count*2) > 0 and index < len(s) :
                    res += s[index-count*2] + s[index]
                    index += T
                elif  0 < (index - count*2) < len(s) and index >= len(s):
                    res += s[index-count*2]
                    index += T
                elif (index - count) == 0 and count < len(s): # 避免越界,比如 s = 'A', numRows = 3
                    res += s[count]
                    index += T
                else:
                    flag = False
            count += 1

    return res

print(convert(s, numRows))

做法二

字符串是怎么变化的我们就怎么解决
还是以索引为例(其实直接用字符串也可以,因为不用刻意找规律)
力扣题(6) —— z字形变换_第2张图片
不过要把这样的思路转换成代码还是比前一个解法要困难一点的。

s = input()

numRows = int(input())

def convert(s, numRows):
    if numRows == 1:
        return s
        
    # 建立一个二维数组
    res_list = []
    for i in range(numRows):
        res_list.append([])
	
	# 先把第一个字符加入到list1
    res_list[0].append(s[0])
    res = ""
    index = 1
    flag = 1 # 因为有list1 到 list2 到 list3,到 list2 这样的规律,我们就立一个 flag 来判断是正序还是逆序。我用 flag = 1 表示正序。
    count = 1 # 表示 list几
    while index < len(s):
        if flag == 1:
            res_list[count].append(s[index])
            if index % (numRows - 1) == 0:
                index += 1
                flag = -flag 
                count -= 1
     '''
     何时从正序到逆序(或是从逆序到正序)我们又是找规律,发现当 index 是 (numrows-1) 的倍数时转换。这也是把index = 0 特别讨论的原因,0 % 任何数 == 0,但实际上index == 0 顺序并不改变
     '''
            else:
                index += 1
                count += 1
        else:
            res_list[count].append(s[index])
            if index % (numRows - 1) == 0:
                index += 1
                flag = -flag
                count += 1
            else:
                index += 1
                count -= 1

    for i in res_list:
        for j in i:
            res += j

    return res

print(convert(s, numRows))

你可能感兴趣的:(python,的一些编程题,python,leetcode)