[路飞]_LeetCode题6 Z 字形变换

186 阅读1分钟

题目描述

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

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/zi… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解思路

image.png

题解代码

/*
 * @lc app=leetcode.cn id=6 lang=javascript
 *
 * [6] Z 字形变换
 */

// @lc code=start
/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
var convert = function(s, numRows) {
  if(s.length <= numRows || numRows === 1) return s;
  let convertArr = new Array(numRows).fill("");
  let loc = 0;//赋值位置
  let turn = false;//是否转向
  for(const c of s) {
    convertArr[loc] += c;
    //当到达z型的顶点即首行或尾行位置就要变向
    if(loc == 0 || loc == numRows - 1)
      turn = !turn;
    //turn为true时为正向向下,loc位置+1,false时为逆向,loc位置-1
    loc += turn ? 1 : -1;
  }
  return convertArr.join('');
};
// @lc code=end