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; }