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 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;
    }
};

你可能感兴趣的:(Leetcode刷题,leetcode,算法,职场和发展)