[leetcode]ZigZag Conversion

水题。主要就是画画图，找找规律。稍微要注意的是：1. nRow为1的情况；2. 采用变量index方便边界判断

public class Solution {
    public String convert(String s, int nRows) {
        StringBuilder sb = new StringBuilder();
        if (nRows == 1) return s;
        // nRows >= 2
        for (int i = 0; i * (2 * nRows - 2) < s.length(); i++)
        {
            sb.append(s.charAt(i * (2 * nRows - 2)));
        }
        
        for (int i = 1; i < nRows - 1; i++)
        {
        	for (int j = 0;; j++)
            {
        		int index = j * (2 * nRows - 2) - i;
        		if ( index > 0 && index < s.length())
        		{
        			sb.append(s.charAt(index));
        		}
        		index = j * (2 * nRows - 2) + i;
        		if ( index > 0 && index < s.length())
        		{
        			sb.append(s.charAt(index));
        		}
        		else
        		{
        			break;
        		}
            }
        }
        
        // last
        if (nRows != 0 && nRows != 1)
        {
	        for (int i = 0; i * (2 * nRows - 2) + nRows - 1 < s.length(); i++)
	        {
	        	sb.append(s.charAt(i * (2 * nRows - 2) + nRows - 1));
	        }
        }
        
        return sb.toString();
    }
}

第二刷，先找规律，写公式，然后写代码，要注意行数为1的情况：

class Solution {
public:
    string convert(string s, int nRows) {
        if (nRows == 1) return s;
        string res;
        int inc = (nRows - 1) * 2;
        for (int i = 0; i < nRows; i++)
        {
            for (int j = i; j < s.length(); j += inc)
            {
                res.push_back(s[j]);
                if (i == 0 || i == nRows - 1)
                    continue;
                if (j + inc - i * 2 < s.length())
                    res.push_back(s[j + inc - i * 2]);
                    
            }
        }
        return res;
    }
};