問題描述

將一個給定字串根據給定的行數，以從上往下、從左到右進行Z 字形排列。

比如輸入字串為 "LEETCODEISHIRING"行數為 3 時，排列如下：

L C I R
E T O E S I I G
E D H N
之後，你的輸出需要從左往右逐行讀取，產生出一個新的字串，比如："LCIRETOESIIGEDHN"。

請你實現這個將字串進行指定行數變換的函式：

string convert(string s, int numRows);
示例1:

輸入: s = "LEETCODEISHIRING", numRows = 3
輸出: "LCIRETOESIIGEDHN"
示例2:

輸入: s = "LEETCODEISHIRING", numRows =4

L D R
E O E I I
E C I H N
T S G

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/zigzag-conversion

解答

class Solution {
    public String convert(String s, int numRows) {
        if(numRows == 1)return s;
        StringBuilder res = new StringBuilder();
        
 
int step = 2*(numRows-1), little_step, len = s.length(), flag;
        for(int i=1;i<=numRows;i++){
            flag = 1;
            if(i==1 || i == numRows){
                little_step = i-1;
                while(little_step < len){
                    res.append(s.charAt(little_step));
                    little_step 
 
+= step;
                }
            }else{
                little_step = i-1;
                while(little_step < len){
                    res.append(s.charAt(little_step));
                    if(flag == 0)flag = 1; //非中間節點，則改變步伐。
                    else flag = 0;
                    if((numRows-i)*2 != numRows-1){
                        if(flag == 1){
                            little_step += step - (numRows-i)*2;
                        }else little_step += (numRows-i)*2;
                    }else little_step += (numRows-i)*2;
                }
            }
        }
        return res.toString();
    }
}