题目链接:
力扣https://leetcode-cn.com/problems/zigzag-conversion/
【分析】一开始用二维数组模拟这个过程,然后遍历二维数组找符合条件的字符,但是超时了。所以想到了用StringBuilder来表示每一行。对字符串中的字符进行遍历,然后加入到各自行对应的StringBuilder中即可。
class Solution {
public String convert(String s, int numRows) {
int r = numRows, n = s.length();
List<StringBuilder> list = new ArrayList<>();
int i, j, k = 0, flag = 0;
for(i = 0; i < r; i++) list.add(new StringBuilder());
while(true){
for(i = 0; i < r; i++) {
list.get(i).append(s.charAt(k++));
if(k == n){
flag = 1;
break;
}
}
if(flag == 1) break;
for(i = r - 2; i >= 1; i --) {
list.get(i).append(s.charAt(k++));
if(k == n){
flag = 1;
break;
}
}
if(flag == 1) break;
}
StringBuilder ans = new StringBuilder();
for(i = 0; i < r; i++) {
System.out.println(list.get(i).toString());
ans.append(list.get(i));
}
return ans.toString();
}
}
【改进】可以用一个t来表示方向,另外由StringBuilder 组成的链表可以替换成StringBuilder的数组,但是要注意初始化。
class Solution {
public String convert(String s, int numRows) {
int r = numRows, n = s.length(), i, j = 0, t = 1;
if(r == 1) return s;
StringBuilder[] sbs = new StringBuilder[r];
for(i = 0; i < r; i++) sbs[i] = new StringBuilder();
for(i = 0; i < n; i++){
sbs[j].append(s.charAt(i));
j += t;
if(j == r){
t = -1;
j += t + t;
}
if(j == -1){
t = 1;
j += t + t;
}
}
for(i = 1; i < r; i++) sbs[0].append(sbs[i]);
return sbs[0].toString();
}
}