题目描述:
将一个给定字符串 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);
思路:
把排序的矩阵倒过来,就是一个数组。使用指针来完成z字变换,把字符一一放上去,最后按照顺序读出来即可。
代码:
public static String convert(String s, int numRows) {
if (numRows == 1) {
return s;
}
String[] strList = new String[numRows];
for (int i = 0; i < numRows; i++) {
strList[i] = "";
}
AtomicInteger count = new AtomicInteger(0);
boolean roundRight = true;
for (char c:s.toCharArray()) {
if (roundRight) {
strList[count.get()] += c;
if (count.get() +1 == numRows) {
roundRight = false;
count.addAndGet(-1);
}
else {
count.getAndAdd(1);
}
}
else {
strList[count.get()] += c;
if (count.get() == 0) {
roundRight = true;
count.getAndAdd(1);
}
else {
count.addAndGet(-1);
}
}
}
String re = "";
for (int i = 0; i < numRows; i++) {
re += strList[i];
}
return re;
}