Z字形变换

传送门 https://leetcode-cn.com/problems/zigzag-conversion/

题目描述：

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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);

示例：

输入: s = “LEETCODEISHIRING”, numRows = 3
输出: “LCIRETOESIIGEDHN”
解释：
L C I R
E T O E S I I G
E D H N

输入: s = “LEETCODEISHIRING”, numRows = 4
输出: “LDREOEIIECIHNTSG”
解释:
L D R
E O E I I
E C I H N
T S G

个人思路：

根据Z字形排列规律，先找出排在第1行的字符，按找出顺序加到返回的字符串中，再行数累加，重复操作。所以，关键就在于如何找出排在每行的字符。
将倒立Z的左竖列和斜对角线上的字符(不包括下一个竖列的第一个字符)作为一个周期，竖行上的字符显然满足 (j+1)%t==i 条件，而对角线上的字符需要分类讨论，处理周期问题，常用的运算符为%。所以判断条件为 numRows-((j+1)%t-numRows)==i ， 整理得 (2*numRows-(j+1)%t)==i，又因为(j+1)%t==0的情况不是预期，所以要分类讨论下。此外，由于竖行和对角线的判断在最后一行时同时满足，故需要加个格外条件。最后，遗漏了当 numRows==1 时，程序会发生错误，需将该情况另外处理。

class Solution {
public:
    string convert(string s, int numRows) {
        string str = "";
        int len = s.size();
        if(numRows==1)
            return s;
        int k=0,t = 2*numRows-2,ans=0;
        for(int i=1; i<=numRows; i++)
        {
            for(int j=0; j<len; j++)
            {
                if((j+1)%t==i && i!=numRows)
                {
                    str+=s[j];
                }

                if((j+1)%t!=0 && (2*numRows-(j+1)%t)==i)
                {
                    str+=s[j];
                }else if((j+1)%t==0 && (2*numRows-t)==i)
                {
                    str+=s[j];
                }
            }
        }
        return str;
    }
};