Leetcode-Day05

题目：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"
示例 2：
输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P            I            N
A       L   S       I   G
Y   A       H   R
P             I
示例 3：

输入：s = "A", numRows = 1
输出："A"

解法一（暴力）：

根据输出解释能够发现第一列是正常输出的，而第二列依次递减输出一个字符，根据这个规律创建一个二维字符数组来还原过程，代码如下：

class Solution {
    public String convert(String s, int numRows) {
        String[][] rs = new String[numRows][s.length()];
        int index = 0;  //负责字符串的索引
        int column = 0; //负责列
        int mark = 0;   //每一列的输出的位置
        String re = "";
        while(index < s.length()){
            for(int i = 0;i < numRows;i++){
                if(mark == 0 && index < s.length()){
                    String temp = String.valueOf(s.charAt(index));
                    rs[i][column] = temp;
                    index++;
                }else if(mark != 0 && (numRows-mark-1) == i && index < s.length()){
                    String temp = String.valueOf(s.charAt(index));
                    rs[i][column] = temp;
                    index++;
                }else{
                    rs[i][column] = " ";
                }
            }
            if((numRows-1) != 0){
               mark = (mark+1) % (numRows-1);
            }
            
            column++;
        }
        for(int i = 0;i < numRows;i++){
            for(int j = 0;j < column;j++){
                if(rs[i][j] != " "){
                    re += rs[i][j];
                }
            }
        }
        return re;
    }
}

但这种方式的时间复杂度很大，达到O(n^2)，因此参考了评论大牛的解法，时间复杂度直接下降到O(n)。

解法二代码如下：

class Solution {
public:
    string convert(string s, int numRows) {
        vector<string> temp(numRows);
        string res;
        if(s.empty() || numRows < 1) return res;
        if(numRows == 1) return s;
        for(int i = 0; i < s.size(); i++){
            int ans = i / (numRows-1);
            int cur = i % (numRows-1);
            if(ans % 2 == 0){  //结果为偶数或0
                temp[cur].push_back(s[i]); //按余数正序保存
            }
            if(ans % 2 != 0){  //结果为奇数
                temp[numRows-cur-1].push_back(s[i]); //按余数倒序保存
            }
        }
       for(int i = 0; i < temp.size(); i++){
               res += temp[i];
        }
        return res;
    }
};