6.Z字形变换

152 阅读1分钟

将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时,排列如下:

L C I R E T O E S I I G E D H N 之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"LCIRETOESIIGEDHN"。

public String convert(String s, int numRows) {
        if(numRows <= 1) return s;
        List<StringBuffer> list = new ArrayList<>();
        for(int i = 0; i < numRows; i++){  //注意要初始化
            list.add(new StringBuffer());
        }
        int flag = -1;  //设置flag每次在当前行加1
        int index = 0;  //当前行
        for(int i = 0; i < s.length(); i++){
            list.get(index).append(s.charAt(i));  //当前行的字符串后添加字符
            if(index == 0 || index == numRows-1){
                flag = -flag;
            }
            index += flag;
        }

        StringBuffer sb = new StringBuffer();
        for(int i = 0; i < list.size(); i++){
            sb.append(list.get(i));
        }

        return sb.toString();
    }