LeetCode ZigZag Conversion

ZigZag Conversion

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) {
    	int len = s.size();
		int small = max(0, nRows-2);
		stringstream oss;
		for(int i = 1; i <= nRows; ++i)
			for(int j = 1; j <= (len/(small+nRows) + 1)*2; ++j)
				if(isChar(i, j, nRows, small))
				{
					int idx = xy2Int(i, j, nRows, small);
					if(idx <= len)	oss << s[idx-1];
				}
		return oss.str();
    }
	bool isChar(int x, int y, int big, int small)
	{
		if(1 == y%2)
			return x >= 1 && x <= big;
		else
			return x >= 2 && x <= small + 1;
	}
	int xy2Int(int x, int y, int big, int small)
	{
		int idx = 0;
		idx += (big + small)*((y-1)/2);
		idx += ((y+1)%2)*big;
		if(0 == y%2 && x >= 1 && x <= small+1)
			idx += small - x + 2;
		else
			idx += x;
		return idx;
	}
};



你可能感兴趣的:(String)