python3 解法, 执行用时: 420 ms, 内存消耗: 26.8 MB, 提交时间: 2023-09-17 12:13:21

class Solution:
    def decodeCiphertext(self, encodedText: str, rows: int) -> str:
        cols = len(encodedText) // rows   # 辅助矩阵的列数
        res = []   # 遍历到的字符
        for i in range(cols):
            # 从左至右枚举每一条路径
            r = 0
            c = i
            while r < rows and c < cols:
                res.append(encodedText[r*cols+c])
                r += 1
                c += 1
        # 删去末尾空格
        while res and res[-1] == ' ':
            res.pop()
        return "".join(res)

cpp 解法, 执行用时: 120 ms, 内存消耗: 38.4 MB, 提交时间: 2023-09-17 12:13:07

class Solution {
public:
    string decodeCiphertext(string encodedText, int rows) {
        int cols = encodedText.size() / rows;   // 辅助矩阵的列数
        string res;   // 遍历到的字符
        for (int i = 0; i < cols; ++i){
            // 从左至右枚举每一条路径
            int r = 0;
            int c = i;
            while (r < rows && c < cols){
                res.push_back(encodedText[r*cols+c]);
                ++r;
                ++c;
            }
        }
        // 删去末尾空格
        while (res.size() and res.back() == ' '){
            res.pop_back();
        }
        return res;
    }
};

java 解法, 执行用时: 27 ms, 内存消耗: 53 MB, 提交时间: 2023-09-17 12:12:50

class Solution {
    public String decodeCiphertext(String encodedText, int rows) {
        /*
        模拟:
        先建立一个矩阵把字母填进去，然后再通过解码的方式读取
        注意:
        1.可能后前导空格，但是没有尾随空格
        2.可能开始解码是从(0,0)开始的
        3.矩阵的列数col可以由len/encodedText计算
        0 <= encodedText.length <= 1e6
        时间复杂度:O(N)  空间复杂度:O(N)
         */
        // 原位解密法
        int n = encodedText.length();
        int cols = n / rows;
        // 解密字符串并填入sb中
        StringBuilder sb = new StringBuilder();
        for (int j = 0; j < cols; j++) {
            for (int i = 0; i < rows && i + j < cols; i++) {
                int curIdx = i * cols + (i + j);    // 逆运算出在原来字符串的位置
                sb.append(encodedText.charAt(curIdx));
            }
        }
        // 去掉尾随空格
        int idx = sb.length() - 1;
        while (idx > 0 && sb.charAt(idx) == ' ') idx--;
        return sb.substring(0, idx + 1);
    }
}

golang 解法, 执行用时: 24 ms, 内存消耗: 8.8 MB, 提交时间: 2023-09-17 12:12:34

func decodeCiphertext(encodedText string, rows int) string {
	ans := []byte{}
	for i, j, k, cols := 0, 0, 0, len(encodedText)/rows; k < cols; {
		ans = append(ans, encodedText[i*cols+j]) // 转换成在 encodedText 上的下标
		i++
		j++
		if i == rows || j == cols { // 触及边界
			k++
			i, j = 0, k // 移至下一条斜向
		}
	}
	return string(bytes.TrimRight(ans, " ")) // 移除末尾多余空格
}