【算法】leetcode-6 Z字形变换

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

比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：

P A H N A P L S I I G Y I R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“PAHNAPLSIIGYIR”。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1：

输入：s = “PAYPALISHIRING”, numRows = 3 输出：“PAHNAPLSIIGYIR”

class Solution {
    public String convert(String s, int numRows) {
        StringBuilder sb = new StringBuilder();
        if(s.length()<=numRows || numRows==1) return s;
        int gap = numRows*2 - 2 ;
        for(int i=0;i<numRows;i++){
            int step = gap - 2*i;
            for (int j = i; j < s.length(); j=j+gap) {
                //每层的循环
                sb.append(s.charAt(j));
                if(step<gap && step>0 && (j+step)<s.length()){
                    sb.append(s.charAt(j+step));
                }
            }
        }
        return sb.toString();
    }
}

class Solution {
    public String convert(String s, int numRows) {
        /*
          行数为n时，对字符串进行(n+n-2)分段划分。
        */
        if(numRows==1) return s;
        ArrayList<StringBuilder> strArr = new ArrayList<>();
        for(int i=0;i<numRows;i++){
            strArr.add(new StringBuilder());
        }
        for(int i=0;i<s.length();i+=2*numRows-2){
            for(int index=0;index<2*numRows-2 && index+i<s.length();index++){
                if(index<numRows){
                    strArr.get(index).append(s.charAt(i+index));
                }else{
                    strArr.get(2*numRows-index-2).append(s.charAt(i+index));
                }
            }
        }
        StringBuilder sb = new StringBuilder();
        for(StringBuilder subStr:strArr){
            sb.append(subStr);
        }
        return sb.toString();
    }
}