算法day7-Z-字形变换-O2算法思路整理
目录
【算法day7】 Z 字形变换 （O2算法思路整理）
Z 字形变换，算法思路整理
将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：
P A H N
A P L S I I G
Y I R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“PAHNAPLSIIGYIR”。
请你实现这个将字符串进行指定行数变换的函数：
- 不擅长打印图形，我将把所有矩阵和一维坐标的映射关系都学习一遍
class Solution {
public:
string convert(string s, int numRows) {
int s_len = s.size();
if (s_len <= numRows || s_len <= 1||numRows==1) {
return s;
}
int T = numRows + numRows - 2;
int T_length = 1 + numRows - 2;
int T_num = s.size() / T + 1;
int numCols = T_num * T_length;
vector<vector<char>> ch(numRows, vector<char>(numCols));
int index_s = 0;
for (int i = 0; i < numCols; i++) {
for (int j = 0; j < numRows; j++) {
if (index_s == s_len)
break;
if (index_s % T < numRows) {
ch[j][i] = s[index_s];
index_s++;
} else {
// 向着右上移动,右上移动时的（x,y）坐标Y正好随着周期变化
int row_id = numRows - i % T_length - 1;
ch[row_id][i] = s[index_s];
index_s++;
break;
}
}
}
string ans;
for (auto c_line : ch) {
for (auto c : c_line) {
if (c)
ans += c;
}
}
return ans;
}
};