[LeetCode] Zigzag Conversion

문제링크

Zigzag Conversion - LeetCode

 

일반적인 형태의 문자열 ("ABCDEFG")을 지그재그 형태로 변환하고 이를 row 기준으로 재정렬하면 어떻게 변하는지 구하는 문제이다.

접근 방법

지그재그로 변한 형태를 index번호로 바꿔 적어보면 패턴이 보인다. (numRows = 4일때)
0         6             12
1     5  7        11
2   4    8   10
3         9

한번의 순환이 이뤄질때마다 (numRows - 1) * 2 개의 요소가 반복된다.
0,1,2,3,4,5 -> 6,7,8,9,10,11 - > ...

이를 통해 numRows 길이의 빈 문자열 배열을 만들고 index에 맞춰 삽입하면 해결된다.

0 -> 0번 index  -> i % (numRows - 1) * 2
1 -> 1번 index  -> i % (numRows - 1) * 2
2 -> 2번 index  -> i % (numRows - 1) * 2
3 -> 3번 index  -> i % (numRows - 1) * 2
4 -> 2번 index -> ((numRows - 1) * 2) -  ( i % (numRows - 1) * 2 )
5 -> 1번 index -> ((numRows - 1) * 2) -  ( i % (numRows - 1) * 2 )

6 -> 똑같이 반복

// 일반적인 문자열 s를 지그재그 형태로 바꿨을때 변환되는 문자열을 구하는 문제이다.

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
var convert = function (s, numRows) {
  let answer = new Array(numRows).fill('');
  let patternLen = (numRows - 1) * 2;

  if (numRows < 2) return s;

  for (let i = 0; i < s.length; i++) {
    let patternidx = i % patternLen;
    if (patternidx < numRows) answer[patternidx] += s[i];
    else answer[patternLen - patternidx] += s[i];
  }

  return answer.join('');
};