6. 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"
提示:
1 <= s.length <= 1000
s
由英文字母(小写和大写)、','
和'.'
组成1 <= numRows <= 1000
分析:
- 面对这道算法题目,二当家的陷入了沉思。
- 直接找到规律模拟即可。
题解
rust:
impl Solution {
pub fn convert(s: String, num_rows: i32) -> String {
// 输入字符串的长度
let n = s.len();
let num_rows = num_rows as usize;
if num_rows == 1 || num_rows >= n { return s; }
// 结果
let mut ret = String::with_capacity(s.len());
// 一次循环串的长度
let cycle_len = 2 * num_rows - 2;
// 将字符串转为字符Vec
let s = s.chars().into_iter().collect::<Vec<char>>();
// 注意字符串长度小于行数的问题
for i in 0..num_rows.min(n) {
(0..n - i).step_by(cycle_len).for_each(|j| {
ret.push(s[j + i]);
if i != 0 && i != num_rows - 1 && j + cycle_len - i < n {
ret.push(s[j + cycle_len - i]);
}
});
}
ret
}
}
go:
func convert(s string, numRows int) string {
n := len(s)
if numRows == 1 || numRows >= n {
return s
}
// 一次循环串的长度
cycleLen := 2*numRows - 2
ans := make([]byte, 0, n)
for i := 0; i < numRows; i++ {
for j := 0; j+i < n; j += cycleLen {
ans = append(ans, s[j+i])
if 0 < i && i < numRows-1 && j+cycleLen-i < n {
ans = append(ans, s[j+cycleLen-i])
}
}
}
return string(ans)
}
c++:
class Solution {
public:
string convert(string s, int numRows) {
int n = s.length();
if (numRows == 1 || numRows >= n) {
return s;
}
string ans;
const int cycleLen = 2 * numRows - 2;
for (int i = 0; i < numRows; ++i) {
for (int j = 0; j + i < n; j += cycleLen) {
ans += s[j + i];
if (0 < i && i < numRows - 1 && j + cycleLen - i < n) {
ans += s[j + cycleLen - i];
}
}
}
return ans;
}
};
python:
class Solution:
def convert(self, s: str, numRows: int) -> str:
n = len(s)
if numRows == 1 or numRows >= n:
return s
# 一次循环串的长度
cycle_len = 2 * numRows - 2
ans = []
for i in range(numRows):
for j in range(0, n - i, cycle_len):
ans.append(s[j + i])
if 0 < i < numRows - 1 and j + cycle_len - i < n:
ans.append(s[j + cycle_len - i])
return ''.join(ans)
java:
class Solution {
public String convert(String s, int numRows) {
final int n = s.length();
if (numRows == 1 || numRows >= n) return s;
StringBuilder ret = new StringBuilder();
int cycleLen = 2 * numRows - 2;
for (int i = 0; i < numRows; i++) {
for (int j = 0; j + i < n; j += cycleLen) {
ret.append(s.charAt(j + i));
if (i != 0 && i != numRows - 1 && j + cycleLen - i < n)
ret.append(s.charAt(j + cycleLen - i));
}
}
return ret.toString();
}
}
非常感谢你阅读本文~
放弃不难,但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子:https://developer.aliyun.com/profile/sqd6avc7qgj7y 博客原创~