【Leetcode】6. N 字形变换

最新推荐文章于 2024-09-30 14:51:11 发布

枫落淮南

最新推荐文章于 2024-09-30 14:51:11 发布

阅读量60

点赞数

分类专栏： leetcode 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/Bobbie_/article/details/129730273

版权

leetcode 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

地址：6.N字形变换

题目

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

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串

示例

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

思路

方法一：

根据字母的下标找规律

得出结论：

1.每一行的第一个起始下标都为当前行的行号

2.第一行和最后一行的间距都为step = 2*numrows - 2

3.除了首尾行，中间行数的下标以(step-2*row) - > （2*row) 的规律交替取值

4. 下标的范围在0-(len(s)-1)之间，要判断越界

代码

class Solution {
public:
    string convert(string s, int n) {
        int len = s.size(),step = 2 * n - 2; 
        string res = "";
        if(n == 1 )
            return s;
        for(int i=0;i<n;++i){   //依次遍历每一行
            if(i == 0 || i == n-1){     //首尾行单独处理
                for(int j=i;j<len;){
                    if(j < len)
                        res += s[j];
                    j += step;
                }
            }
            else{   //中间行
                int time = 1;
                for(int j=i;j<len;){
                    if(time%2==1){   //奇数次——step - 2*n
                        if(j < len)
                            res += s[j];
                        else
                            break;
                        j += step - 2 * i;
                        
                    }
                    else{       //偶数次——2*n
                        if(j < len)
                            res += s[j];
                        else
                            break;
                        j += 2 * i;
                    }
                    ++time;
                }
            }
        }
        return res;
    }
};