题目描述:The string “PAYPALISHIRING” is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility) (之字形转换)
input:s = "PAYPALISHIRING", numRows = 3 output:"PAHNAPLSIIGYIR" P A H N A P L S I I G Y I R input:s = "PAYPALISHIRING", numRows = 4 output:"PINALSIGYAHRPI" P I N A L S I G Y A H R P I第一种方法:就样例标准来看,输出的图形是有规律的,string的每个组成所在的行是周期分布,该周期为T=numRows+numRows-2;就numRows=4来说,周期为6,分别是0,1,2,3,2,1;可以按照这个规律分别保存string的每个组成的行和列,然后再按行添加的一个新的string中 代码:
string convert(string s, int numRows) { if (s.size() <= numRows||numRows==1) { return s; } else { //result保存结果,result.first保存string的元素下标,result.second保存该下标的行和列 vector<pair<int, pair<int, int>>> result; vector<int> get_row; string ret; for (int i = 0; i != numRows; ++i) { get_row.push_back(i); } for (int i = numRows - 2; i != 0; --i) { get_row.push_back(i); } int t = get_row.size(); int row, col = 0, pos = 0; for (int i = 0; i != s.size(); ++i) { pos = i % t; row = get_row[pos]; if (pos > numRows - 1) { col = (col + 1); } if ((i / t) != 0 && pos == 0) col++; result.push_back({ i,{row,col} }); } //按照行的顺序排列,即0行的在前面,依次排列 std::sort(result.begin(), result.end(), [](pair<int, pair<int, int>>& a, pair<int, pair<int, int>>& b)->bool {return a.second.first < b.second.first; }); for (auto i : result) { ret.push_back(s[i.first]); } return ret; } }分析:该算法的时间复杂度为O(s.size()),空间复杂度很高,所以memory limit exceeded,不推荐使用
第二种方法:观察每行的string的下标和T发现,第一行都为i%T==0的,第二行与T的余数都和周期相差1(以numRows=4为例,第二行余数为1或5,在周期上和6均差1),依次; 代码:
string convert(string s, int numRows) { if(numRows==1) { return s; } else { string ret; int T = numRows + numRows - 2; for (int i = 0; i != numRows; i++) { for (int j = 0; j != s.size(); j++) { if (j%T == i || T - j % T == i) { ret.push_back(s[j]); } } } return ret; } }分析:时间复杂度(numRows*n),空间复杂度O(n);
第三种方法(官方)
string convert(string s, int numRows) { if (numRows == 1) return s; vector<string> rows(min(numRows, int(s.size()))); int curRow = 0; bool goingDown = false; for (char c : s) { rows[curRow] += c; if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown; curRow += goingDown ? 1 : -1; } string ret; for (string row : rows) ret += row; return ret; }curRow保存当前要处理的行,goingDown控制向下进行还是向上进行,向下进行的条件为curRow==0(在第一行),向上进行的条件为curRow=numRows-1(在最后一行),所以当进行到这两行时,goingDown翻转一次,时间复杂度O(n),空间复杂度O(n);