【每天一道编程系列-2018.2.20】（Ans）

【题目描述】

　　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 
　　APLSIIG 
　　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字型输出。 

【解题思路】

　　计算出字符的最大列数，根据列数和行数创建一个一维数组，再计算每个字符中一维数组中的位置，再对一维数组中的字符进行紧凑操作，返回结果。 

【本题答案】

/**
 * @author yesr
 * @create 2018-02-20 下午10:40
 * @desc
 **/
public class Test0220 {
    public String convert(String s, int nRows) {

        if (s == null || s.length() <= nRows || nRows == 1) {
            return s;
        }

        int index = s.length();
        int rowLength = 0; // 计算行的长度，包括最后换行字符

        int slash = nRows - 2; // 一个斜线除去首尾所占用的行数

        while (index > 0) {
            // 竖形的一列
            index -= nRows;
            rowLength++;

            // 斜着的列数
            for (int i = 0; i < slash && index > 0; i++) {
                rowLength++;
                index--;
            }
        }

        char[] result = new char[nRows * rowLength]; // 保存结果的数组，最后一列用于保存换行符
        for (int i = 0; i < result.length; i++) { // 初始化为空格
            result[i] = ' ';
        }

        int curColumn = 0; // 当前处理的行数
        index = 0;
        while (index < s.length()) {
            // 处理竖线
            for (int i = 0; i < nRows && index < s.length(); i++) {
                result[rowLength * i + curColumn] = s.charAt(index);
                index++;
            }
            curColumn++;
            // 处理斜线
            for (int i = nRows - 2; i > 0 && index < s.length(); i--) {
                result[rowLength * i + curColumn] = s.charAt(index);
                curColumn++;
                index++;
            }
        }

        // 对字符数组进行紧凑操作
        index = 0;
        while (index < s.length() && result[index] != ' ') { // 找第一个是空格的字符位置
            index++;
        }
        int next = index + 1;
        while (index < s.length()) {
            while (next < result.length && result[next] == ' ') { // 找不是空格的元素
                next++;
            }
            result[index] = result[next];
            index++;
            next++;
        }
        return new String(result, 0, index);
    }
}