LeetCode (Z)

ZigZag Conversion

 

AC Rate: 200/827

My Submissions

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P   A   H   N
A P L S I I G
Y   I   R

And then read line by line:  "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3) should return  "PAHNAPLSIIGYIR".

class Solution {
public:
	string convert(string s, int nRows) {
	    if (nRows == 1) return s;
		vector<string> vec(nRows);
		for (int i = 0, j = 0, k = 1; s[i]; ++i, j += k) {
			vec[j] += s[i];
			if (j == nRows - 1)
				k = -1;
			else if (j == 0)
				k = 1;
		}
		string ans;
		for (int i = 0; i < nRows; ++i)
			ans += vec[i];
		return ans;
	}
};