力扣6.N字形变换

题目描述

思路

模拟轨迹，每当行数i到最顶（0），或者最底（numRows）的时候，就会反方向走。

用flag来标记方向，在题解里看到，真的很巧妙5555！

代码

class Solution {
   
    public String convert(String s, int numRows) {
   
        if(numRows < 2) return s;
        //用ArrayList，方便后续输出，链表里存的是可修改的StringBuilder类型
        List<StringBuilder> rows = new ArrayList<StringBuilder>();
        //遍历每一行，每一行的字符串为rows[i]
        for(int i = 0; i < numRows; i++) rows.add(new StringBuilder());
        //用flag标记方向
        int i = 0, flag = -1;
        for(char c : s.toCharArray()) {
   
        	//把s中的字符添加进rows[i]中
            rows.get(i).append(c);
            //每当到最顶或者最底时，由于flag的存在，i就会反向走
            if(i == 0 || i == numRows -1) flag = - flag;
            i += flag;
        }
        StringBuilder res = new StringBuilder();
        for(StringBuilder row : rows) res.append(row);
        return res.toString();
    }
}