【java经典面试题-】zigzag存储数组后按行遍历

目录

 

题目：

方案一：根据zigzag的生成规律，求z子型排布中每一行的元素

方案二：找zigzag的排布规律，按照zigzag的按行输出规律访问原字符串字符。

---

题目：

Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"

直观看来数组结构如下：

P   A   H   N
A P L S I I G
Y   I   R

根据原数组顺序按照z子型重新排布数组元素，再按行输出，形成新的字符串。

方案一：根据zigzag的生成规律，求z子型排布中每一行的元素

class Solution {
    public String convert(String s, int numRows) {
        if(numRows == 1) return s;// 特殊情况处理
        List<StringBuilder> result = new ArrayList<StringBuilder>();//泛型
        int curRows = 0;
        int index = 0;
        boolean godown = false;
        numRows = Math.min(numRows,s.length());
        for(int i = 0;i<numRows;i++){
            result.add(new StringBuilder());
        }
            while(index<s.length()){
                result.get(curRows).append(s.charAt(index++));
                if(curRows == numRows-1 || curRows == 0)
                      godown = !godown;//调换行号的变化方向
                curRows += godown?1:-1;
                
            }
   
        StringBuilder res = new StringBuilder();
        for(int i = 0;i<numRows;i++){
            res.append(result.get(i));
        }
        return res.toString();
    }
}

方案二：找zigzag的排布规律，按照zigzag的按行输出规律访问原字符串字符。

在"

PAYPALISHIRING

numRows = 4, 转化为

P     I    N
A   L S  I G
Y A   H R
P     I

的例子中，我们发现，zigzag数组的第一行对应原字符串s[0],s[6],s[12],最后一行字符串对应原字符串s[3],s[9]

寻找规律发现，zigzag数组的第一行对应原字符串的下标为k*(2*numRows-2),最后一行字符串对应原字符串下标为

k*(2*numRows-2)+(numRows-1);

又看到第二行

A的index为1，第一行P的index为0

L的index为5，第一行I的index为6

S的index为7，第一行I的index为6

I的index为11，第一行N的index为12

G的index为13，第一行N的index为12

可找到规律：中间第i行对应原字符串下标为k*(2*numRows-2)+i，k*(2*numRows-2)-i（在第一行索引基础上进行计算得出）

代码实现：

class Solution {
    public String convert(String s, int numRows) {
    StringBuilder res = new StringBuilder();
    
    numRows = Math.min(s.length(),numRows);
    if(numRows == 1) return s;
    int cyclenum = 2*numRows-2;
    for(int i = 0;i<numRows;i++)
        for(int j = 0;j*cyclenum+i<s.length();j++){
              if(i==0)
                 res.append(s.charAt(j*cyclenum));
              else if(i!=numRows-1){
                  res.append(s.charAt(j*cyclenum+i));
                 if((j+1)*cyclenum-i<s.length())
                  res.append(s.charAt((j+1)*cyclenum-i));
              }
            else{
                res.append(s.charAt(j*cyclenum+numRows-1));
            }
                 
        }
    return res.toString();
  
    }
}