1. 程式人生 > 其它 >6-- Z 字形變換(利用連結串列替代二維陣列)

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"

思路

可以利用一個List,List rows=new ArrayList<>();其中向連結串列中新增一個StringBuilder就是一行。
可以利用這個建立一個類似於二維陣列,然後通過特性就是每一次都是第一行或是最後一行的時候改變標識。

解答

點選檢視程式碼
package Com.Xu.String;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class Six {
    public static String convert(String s, int numRows) {
        if(numRows==1){
            return s;
        }
        List<StringBuilder> rows=new ArrayList<>();
        for(int i=0;i< Math.min(numRows,s.length());i++){
            rows.add(new StringBuilder());
        }
        int curRow=0;
        boolean goingDown=false;
        for(Character c:s.toCharArray()){
            rows.get(curRow).append(c);
            if(curRow==0||curRow==numRows-1){
                goingDown=!goingDown;
            }
            curRow+=goingDown?1:-1;
        }
        StringBuilder ret=new StringBuilder();
        for(StringBuilder row:rows){
            ret.append(row);
        }
        return ret.toString();
    }

    public static void main(String[] args) {
        String s="ABCD";
        System.out.println(convert(s,3));
    }
}