https://leetcode.com/problems/zigzag-conversion/

水题纯考细心

题目：依照Z字形来把一个字符串写成矩阵，然后逐行输出矩阵。

O(n)能够处理掉

记i为行数

第0行和第numRow-1行。 ans += str[i+k*(numRows*2-2)], k=0,1,2,...

其它, 每一个Z字形(事实上仅仅是一竖一斜两条线)须要加上两个，下标见代码。

特殊处理：numRows=1的时候numRows*2-2==0 ,会死循环的。另外numRows=1的时候直接输出即可

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

class Solution {public:    string convert(string s, int numRows) {        string ret;        if(numRows == 1){            return s;        }        for(int i=0; i<numRows; i++){            if(i == 0 || i == numRows-1){                int j=i;                while(j<s.size()){                    ret = ret + s[j];                    j += numRows*2-2;                }            }else{                int j=i;                while(j<s.size()){                    ret = ret + s[j];                    if(j+numRows*2-2-(i*2) < s.size())ret = ret + s[j+numRows*2-2-(i*2)];                    j += numRows*2-2;                }            }        }        return ret;    }};

int main(){
    string s;
    int numRows;
    Solution sol;
    while(cin >> s >> numRows){
        cout << sol.convert(s, numRows) << endl;
    }
    return 0;
}