第6题 ZigZag转换

题目描述如下：

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

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

And then read line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".

即给出一个字符串以及行数，根据该行数进行Z字形排列（自上而下，自左而右）：

然后从左到右，自上而下进行读取。

 

方法1：

该方法思路比较清晰，实现简单，但效率不高。

step1：

Z字形排列后，可以看作是由多个V构成，根据nRows和字符串的总长度可以确定V的个数和每个V的大小。

代码如下：

int vLen = 2*numRows-2;//每个V的元素个数
        int numV = s.length()/vLen;//V的个数
        if(s.length()%vLen!=0){
            numV=numV+1;
        }

step2：

依次将字符串的每个字符放入到每个V中，两个for循环，外层循环遍历每个字符，内层循环遍历每个V。代码如下：

for(int i=1;i<=numV;i++){
            String V="";
            for(int j=1;j<=vLen&&(i-1)*vLen+j<=s.length();j++){
                V=V+s.charAt((i-1)*vLen+j -1);
            }
            //System.out.println(V);
            list.add(V);
}

for循环内的判断条件(i-1)*vLen+j<=s.length()是为了保证不溢出，因为最后一个V的长度可能小于vLen。

step3：

依次取出所有V的第一行，第二行……第numLine行的字符。代码如下：

String result = "";
        for(int i=1;i<=numRows;i++){
            for(int j=0;j<list.size();j++){
                result=result+getVString(list.get(j),i,vLen);
        }
}

getVString(list.get(j),i,vLen)函数用于取出V中第i行的所有字符。代码如下：

public static String getVString(String V,int linNum,int vLen){
        if(V.length()<linNum) return (String) "";//如果V的长度小于行数linNum
        if(linNum==1) return V.charAt(0)+"";//如果linNum==1,返回V的第一个元素
        if(V.length()<=(vLen+2)/2) return V.charAt(linNum-1)+"";//如果V只有左侧有值
        if(linNum==(vLen+2)/2) return V.charAt((vLen+2)/2-1)+"";//如果行数等于V的总行数，则去V的底部顶点值
        if(vLen+2-linNum<=V.length()) return V.charAt(linNum-1)+""+V.charAt(vLen+2-linNum-1)+"";
        if(vLen+2-linNum>V.length()) return V.charAt(linNum-1)+"";
        return "";
}

完整代码如下：

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

public class ZigZagConversion {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
//        String s = "PAYPALISHIRING";
        String s="ABCDE";
        System.out.println(convert(s,4));
    }
    public static String convert(String s, int numRows) {
        if(numRows==1||s.length()<=1){
            return s;
        }
        List<String> list = new ArrayList<String>();//存放Z的V部分，每个V用一个字符串表示
        int vLen = 2*numRows-2;//每个V的元素个数
        int numV = s.length()/vLen;//V的个数
        if(s.length()%vLen!=0){
            numV=numV+1;
        }
        for(int i=1;i<=numV;i++){
            String V="";
            for(int j=1;j<=vLen&&(i-1)*vLen+j<=s.length();j++){
                V=V+s.charAt((i-1)*vLen+j -1);
            }
            //System.out.println(V);
            list.add(V);
        }
        String result = "";
        for(int i=1;i<=numRows;i++){
            for(int j=0;j<list.size();j++){
                result=result+getVString(list.get(j),i,vLen);
                System.out.println(j+":"+i);
                System.out.println(list.get(j));
                System.out.println(getVString(list.get(j),i,vLen));
            }
        }
        return result;
    }
    public static String getVString(String V,int linNum,int vLen){
        if(V.length()<linNum) return (String) "";//如果V的长度小于行数linNum
        if(linNum==1) return V.charAt(0)+"";//如果linNum==1,返回V的第一个元素
        if(V.length()<=(vLen+2)/2) return V.charAt(linNum-1)+"";//如果V只有左侧有值
        if(linNum==(vLen+2)/2) return V.charAt((vLen+2)/2-1)+"";//如果行数等于V的总行数，则去V的底部顶点值
        if(vLen+2-linNum<=V.length()) return V.charAt(linNum-1)+""+V.charAt(vLen+2-linNum-1)+"";
        if(vLen+2-linNum>V.length()) return V.charAt(linNum-1)+"";
        return "";
    }
}