题目链接:https://leetcode-cn.com/problems/zigzag-conversion/description/
题目描述
将字符串 “PAYPALISHIRING” 以Z字形排列成给定的行数:
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
解决方法
规律:第一行和最后一行的字母下标数是一个等差数列,差值为d=2*numRows-2;其他行的把满列的和单列的数字分成两部分来看,满列的依然是差值为d的等差数列,单列的值为pos+d-2*i(pos是前面一个整列数字的位置,i是行数)
class Solution {
public:
string convert(string s, int numRows) {
if (numRows<=1) return s; //特殊情况
string result;
int d=2*numRows-2;
for (int i=0;i<numRows;i++){ //从第一行遍历到最后一行
for (int pos=i;pos<s.size();pos+=d){
//第一行、最后一行数字和其他行的整列数字
result+=s[pos];
//其他行的单列数字
int tmp=pos+d-2*i;
if (i!=0 && i!=numRows-1 && tmp<s.size())
result+=s[tmp];
}
}
return result;
}
};