LeetCode：06Z字形变换

Z字形变换

题目

将一个给定字符串 $s$ 根据给定的行数 $numRows$ ，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“PAHNAPLSIIGYIR”。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例

---

方法一：模拟变换过程

分析题目，将Z字形看作一个二维矩阵，变换的过程简单描述为：将字符串中的字符依次填入二维矩阵中，填入顺序为从[0][0]开始往下，填入 numRows 行，然后按照 [行数-1][列数+1] 每次填入一个字符，直到 行数=0，然后再依次往下填入，填入 numRows 行后再次按照 [行数-1][列数+1] 每次填入一个字符，循环往复直到字符串中的字符全部填入，最终按照行读取的方式整理出最终的字符串。

实现方式

使用一个Vector容器来存储每一行的元素，最终的Z字形一共 numRows 行，所以先初始化一个含有 numRows 个元素的Vector，按照Z字形变换的顺序，依次将第 i 行的字符放入到 Vector 中的第 i 个元素中，最终将Vector中每个元素中的字符串拼接成最终结果。

设置一个索引 vIndex 在[0,numRows-1]之间来回移动，表示当前字符对应在二维矩阵中的行数，字符串中的字符依次加入到 Vector中的第 vIndex 个元素中。

代码实现

class Solution {
public:
    string convert(string s, int numRows) {
      //用来遍历字符串 s 的索引
      int index = 0;
      int size = s.length();
    
      if (numRows == 1 || numRows >= size) {
        cout << s << endl;
        return 0;
      }
      
      vector<string> nRes(numRows, "");
      //vIndex用来指向当前元素对应在二维矩阵的行数，同样是Vector的元素索引
      int vIndex = 0;
      //判断当前行数索引应该递增还是递减
      bool add = true;
      
      while (index < size)
      {
        //从第0行往下依次填入 numRows 个元素
        while (index < size && add) {
          nRes[vIndex] += s[index];
          vIndex++;
          index++;
          if (vIndex == numRows) {
            add = false;
          }
        }
        vIndex -= 2;
        //按照 [行数-1][列数+1] 每次填入一个字符，直到 行数=0
        while (!add && index < size) {
          nRes[vIndex] += s[index];
          vIndex--;
          index++;
          if (vIndex == -1) {
            add = true;
          }
        }
        vIndex += 2;
      }
      string res = "";
      for (string var : nRes)
      {
        res += var;
      }
      return res;
};

---

看了力扣上的题解，也是找到了变换的规律，并使用数学公式表示出来，思路类似，都可以作为参考。