LeetCode ZigZag Conversion解題レポート


入力文字列に対して、蛇の形を変化させ、行で出力します.
https://oj.leetcode.com/problems/zigzag-conversion/
例えば、The string「PAYPALISHIRING」 蛇の形は次のように変化します.
P        A           H        N
A   P   L    S     I     I   G
Y         I            R
最後に出力を要求するのは「PAHNAPLSIIGYIR」
Write the code that will take a string and make this conversion given a number of rows:
convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".
解題構想:数式を押す方法で結果を出すことができるようですが、ここではプログラムシミュレーションの方法で結果を出すことができます.
プログラムを使用すると、column行の文字配列を初期化し、元の文字列の各文字を文字列の最後までルールに従って対応する行に追加します.
文字列「PAYPALISHIRING」を例にとると、 ”PAY  P   ALI  S  HIR I NG“  ,新規文字配列 matrix[3]
そして、matrix[0]= matrix[0] + "P";  matrix[1] = matrix[1] + "A";  matrix[2] = matrix[2] + "Y";
                                                        matrix[1] = matrix[1] + "P";
            matrix[0] = matrix[0] + "A";  matrix[1] = matrix[1] + "L";  matrix[2] = matrix[2] + "I"; 
                                                        matrix[1] = matrix[1] + "S";
            matrix[0] = matrix[0] + "H";  matrix[1] = matrix[1] + "I";  matrix[2] = matrix[2] + "R";
                                                        matrix[1] = matrix[1] + "I";             matrix[0] = matrix[0] + "N";  matrix[1] = matrix[1] + "G";  
最終出力columnの各文字列:PAHN
APLSIIG
YIR
    ,最終的な結果が得られます.
public class Solution {
    public String convert(String s, int nRows) {
       if(s.length()==1||nRows==1) {
	<span style="white-space:pre">	</span>return s;
	}
		
	StringBuffer[] matrix = new StringBuffer[nRows];
	int i = 0, end = nRows - 2, index = 0;
	for(i=0;i<nRows;i++) {
	<span style="white-space:pre">	</span>matrix[i] = new StringBuffer("");
	}

	while(index<s.length()) {
		//    
<span style="white-space:pre">		</span>for(i=0;index<s.length()&&i<=(nRows-1);i++) {
			matrix[i].append(s.charAt(index++));
		}	
		//    ,     	
		for(end=nRows-2;index<s.length()&&end>0;end--) {
			matrix[end].append(s.charAt(index++));
		}
	}
		
        StringBuffer result = new StringBuffer("");
	for(int a=0;a<nRows;a++) {
		result.append(matrix[a]);
	}
	//System.out.println("result="+result.toString());
	return result.toString();
    }
}