题目
将一个给定字符串 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
复制代码
解题思路
numRows是多少代表是多少数组
但是换行这个规律有点难找 = =
对于n行的, s中的第i个字符:
对余数进行判断
i%(2n-2) == 0 ----> row0
i%(2n-2) == 1 & 2n-2-1 ----> row1
i%(2n-2) == 2 & 2n-2-2 ----> row2
...
i%(2n-2) == n-1 ----> row(n-1)
复制代码
Java代码
class Solution {
public String convert(String s, int numRows) {
if (numRows == 1) {
return s;
}
List<StringBuilder> arr = new ArrayList<StringBuilder>();
for(int i = 0; i < numRows; i++){
arr.add(new StringBuilder());
}
char[] strArr = s.toCharArray();
for (int i = 0; i < strArr.length; i++) {
int num = i%(numRows*2-2);
if (num < numRows) {
arr.get(num).append(strArr[i]);
}else {
arr.get(numRows*2-2-num).append(strArr[i]);
}
}
String str = "";
for (StringBuilder item : arr) {
str = str + item.toString();
}
return str;
}
}
复制代码
Swift代码
class Solution {
func convert(_ s: String, _ numRows: Int) -> String {
if numRows == 1 {
return s
}
var arr = [String]()
for _ in 0..<numRows {
let arr1:String = ""
arr.append(arr1)
}
let strArr = Array(s)
for (i,value) in strArr.enumerated() {
let num = i%(numRows*2-2)
if num<numRows {
arr[num].append(value.description)
}else{
arr[numRows*2-2-num].append(value.description)
}
}
print(arr)
return arr.joined()
}
}
复制代码
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END