leetcode 6. Z 字形变换

题目描述:

思路:

$$模拟就行了...$$

代码:

class Solution {
    public String convert(String s, int numRows) {
        if(numRows == 1) return s;
        List<StringBuilder> rows = new ArrayList<StringBuilder>();
        for (int i = 0; i < numRows; i++) {
            rows.add(new StringBuilder());
        }

        int length = s.length();
        int flag = 1, count = 0;

        for (int i = 0; i < length; i++) {
            if(flag == 1) {
                rows.get(count++).append(s.charAt(i));
            }else {
                rows.get(count--).append(s.charAt(i));
            }

            if(count == 0 || count == numRows - 1) {
                flag = -flag;
            }
        }

        StringBuilder sb = new StringBuilder();
        for (StringBuilder row : rows) {
            sb.append(row);
        }

        return sb.toString();
    }
}