题目链接:https://leetcode-cn.com/problems/zigzag-conversion/t
题目描述:
将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:
P A H N
A P L S I I G
Y I R
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"。
请你实现这个将字符串进行指定行数变换的函数:
思路
找规律,第一行与最后一行之间的元素间隔距离都是固定的,元素间隔距离为 numRows*2-2,第一排从0开始遍历,最后一行从numRows-1开始遍历。从第二行开始除了固定间隔的元素之外,中间还会有一个斜排的元素,仔细观察可以发现这个元素的位置与同一行前一个元素的位置相差了(元素固定间隔距离 - 2*行数) ,于是可得中间行数在转换时,先添加固定间隔距离的元素,然后在通过前一个元素位置计算添加斜排的元素即可。
代码
public static String convert(String s, int numRows) {
if(numRows==1) return s;
StringBuffer result = new StringBuffer();
char ss[] =s.toCharArray();
int flag = numRows+numRows-2;
int len = s.length();
//第一排
for(int i=0;i<len;i+=flag){
result.append(ss[i]);
}
//中间的
for(int i=1;i<numRows-1;i++){
int zj = flag-i*2;
for(int j=i;j<len;j+=flag){
result.append(ss[j]);
//添加斜排元素
if(j+zj<len){
result.append(ss[j+zj]);
}
}
}
//最后一排
for(int i=numRows-1;i<len;i+=flag){
result.append(ss[i]);
}
return result.toString();
}