当前位置:   article > 正文

leetcode6 -- z字形变换

leetcode6 -- z字形变换

题目描述:

将一个给定字符串 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);

示例 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

示例 3:

输入:s = "A", numRows = 1
输出:"A"

提示:

  • 1 <= s.length <= 1000
  • s 由英文字母(小写和大写)、',' 和 '.' 组成
  • 1 <= numRows <= 1000

解法:

Z字形变换先向下,后斜向上,再向下循环,可以用一个boolean值goDown来判断遍历过程中的方向是向上还是向下,当当前行到了第一行或者最后一行时,goDown的状态改变,当前行+1或-1。

1.创建一个列表,列表长度为numRows,里面的每个元素都是一个StringBuilder,来存储对应一行的所有元素,最后再把每一行的元素拼接起来

2.创建两个变量,nowRow表示当前行,goDown表示当前前进的方向,当goDown为true时,向下走,goDown为false时,向上走。

3.将每行对应字符添加到对应的StringBuilder中。

代码:

  1. class Solution {
  2. public String convert(String s, int numRows) {
  3. if (numRows == 1 || numRows >= s.length()) {
  4. return s;
  5. }
  6. List<StringBuilder> rows = new ArrayList<>();
  7. for (int i = 0; i < numRows; i++) {
  8. rows.add(new StringBuilder());
  9. }
  10. boolean goDown = false;
  11. int nowRow = 0;
  12. for (int i = 0; i < s.length(); i++) {
  13. // 第i行对应加当前字符
  14. rows.get(nowRow).append(s.charAt(i));
  15. if (nowRow == 0 || nowRow == numRows - 1) {
  16. goDown = !goDown; // 第一次会变成true,向下走,其余情况取反
  17. }
  18. nowRow += goDown ? 1 : -1; // goDown为true也就是向下走,+1,否则-1
  19. }
  20. StringBuilder res = new StringBuilder();
  21. for (StringBuilder row : rows) {
  22. res.append(row);
  23. }
  24. return res.toString();
  25. }
  26. }

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/酷酷是懒虫/article/detail/907982
推荐阅读
相关标签
  

闽ICP备14008679号