6. N 字形变换

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

javascript

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"
解释：

javascript

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）
链接：https://leetcode.cn/problems/zigzag-conversion

javascript

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
// var convert = function(s, numRows) {
//   const template = getTemplate(numRows)
//   const res = Array.from({length: numRows}, () => Array.from({length: 1}, () => ''))
//   let index = 0, turns = 0
//   while(index < s.length){
//     let j = 0
//     while(j < numRows){
//       for (let i = 0; i < template.length; i++) {
//         if(template[i][j]){
//           res[i][j + turns * numRows] = s[index]
//           index++
//         }
//       }
//       j++
//     }
//     turns++
//   }
//   return res.map(item => item.join('')).join('')
// }

// function getTemplate(numRows){
//   return Array.from({length: numRows}, (_, i) => Array.from({length: ( numRows - 1 || 1)}, (_, j) => j === 0 || i + j === numRows - 1 ? 1 : 0))
// }

// var convert = function(s, numRows) {
//   const { count, template } = getNTemplate(numRows), turns = Math.ceil( s.length / count)
//   const res = Array.from({length: numRows}, () => Array.from({length: 1}, () => ''))
//   let index = 0, turn = 0
//   while(turn < turns){
//     for (let j = 0; j < template[0].length; j++) {
//       for (let i = 0; i < template.length; i++) {
//         if(template[i][j]){
//           res[i][j + turn * numRows] = s[index]
//           index++
//         }
//       }
//     }
//     turn++
//     if(!s[index]) break
//   }
//   return res.map(item => item.join('')).join('')
// }

// function getNTemplate(numRows){
//   let count = 0
//   const template = Array.from({length: numRows}, (_, i) => Array.from({length: ( numRows - 1 || 1)}, (_, j) => {
//     if(j === 0 || i + j === numRows - 1){
//       count++
//       return 1
//     }else {
//       return 0
//     }
//   }))
//   return {
//     count,
//     template
//   }
// }

var convert = function(s, numRows) {
  if (numRows < 2) return s
  const size = 2 * numRows - 2, turns = Math.ceil(s.length / size)
  let res = ''
  for (let i = 0; i < numRows; i++) {
    for (let j = 0; j < turns; j++) {
      const l = j * size + i, r = (j + 1) * size - i
      res += s[l] || ''
      if (i > 0 && l !== r) {
        res += s[r] || ''
      }
    }
  }
  return res
}

console.log(convert("PAYPALISHIRING", 3) === "PAHNAPLSIIGYIR") // "PAHNAPLSIIGYIR"
// console.log(convert("PAYPALISHIRING", 4) === "PINALSIGYAHRPI")
// console.log(convert("A", 1))

6. N 字形变换 #

6. N 字形变换