【leetcode】Z字形变换(桶)

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

链接：https://leetcode-cn.com/problems/zigzag-conversion

思路分析：

一开始是想在对应下标上找规律的。尝试了一下就放弃了…
于是，换别的方法。我们发现Z字形的特点是将字符串先从上到下走n格，然后从下到上走n-2格，直到字符串遍历完。也就是说，我们可以确定每个字符所在行。因此我们可以想到每一行当作一个桶，将字符加到该行对应的桶，最后按行顺序将每个桶的字符串加起来。下面上代码：

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows == 0) return "";
        if(s.length() == 0) return "";
        string st[numRows];
        string ans;
        int t = 0;
        while(t < s.length())
        {
            for(int i = 0;i < numRows && t < s.length();i++)	//从上往下
            {
                st[i].push_back(s[t++]);
            }
            for(int j = numRows-2;j >= 1 && t < s.length();j--)	//从下往上
            {
                st[j].push_back(s[t++]);
            }
        }
        for(int i = 0;i < numRows;i++)
        ans += st[i];
        return ans;
    }
};