欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

#leetcode刷题之路6- Z 字形变换

程序员文章站 2022-06-22 12:52:57
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时,排列如下:L C I RE T O E S I I GE D H N之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"LCIRETOESII ......

将一个给定字符串根据给定的行数,以从上往下、从左到右进行 z 字形排列。
比如输入字符串为 "leetcodeishiring" 行数为 3 时,排列如下:
l     c     i   r
e t o e s i i g
e    d     h  n
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"lciretoesiigedhn"。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numrows);

示例 1:
输入: s = "leetcodeishiring", numrows = 3
输出: "lciretoesiigedhn"

示例 2:
输入: s = "leetcodeishiring", numrows = 4
输出: "ldreoeiiecihntsg"
解释:

l        d        r
e    o e     i  i
e c     i   h   n
t        s       g

 思路:找规律,怎么找呢?

我们换一种方式来看这道题:

例如:字符串123456789...16和字符串123456789...21

#leetcode刷题之路6- Z 字形变换

 

规律1:在我圈出的每一列中,同一行之间的相邻数字只差为 len=2*nrows-2,这也就意味着我们知道了第一列每一行的元素,后面的也就全部知道了。

那么问题来了,除了第一行和最后一行,其他行的相邻元素之间会多出现一个字符。。。所以在这些行中,我们要多考虑这些多余的字符。

规律2:这些多余字符的位置也是有规律的,他们和同一行的前一个元素的位置相差len-2*i,也就是说,多余字符的位置为dif=j+len-2*i

ok,按这个规律来做这个题:

#include <iostream>
using namespace std;

string convert(string s, int nrows)
{
    if (nrows <2)
        return s;
    string ans="";
    int len = 2 * nrows - 2;
    for (int i = 0; i < nrows; ++i)
    {
        for (int j = i; j < s.length(); j += len)
        {
            ans += s[j];
            int dif = j +len - 2 * i;
            if ((i != 0) && (i != (nrows - 1)) &&(dif < s.length()))
                ans += s[dif];
        }
    }
    return ans;
}


int main() {
    string s="abcdefghijklmn";
    string ans=convert(s,3);
    std::cout <<ans << std::endl;
    return 0;
}