[LeetCode]006. Zigzag Conversion JAVA

Zigzag Conversion - LeetCode

---

📌 문제

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)

P   A   H   N
A P L S I I G
Y   I   R

And then read line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string s, int numRows);

---

📌 예시

Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"

 

Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"
Explanation:
P     I    N
A   L S  I G
Y A   H R
P     I

---

📌 내가 푼 방법 설명

이차원 배열이 아닌, 이차원 배열 리스트를 사용한 이유

처음에는 이차원 배열 형식으로 풀려고 했으나, 열의 크기를 알 수 없어 가변 배열인 ArrayList를 사용하였다.

이차원 ArrayList를 만들어주고, 행의 크기는 numRows 설정해주었다.

 

ArrayList도 초기화 해주고 시작하였다. 

만일 초기화 해주지 않으면, 해당 행 별로 ArrayList가 존재하지 않기 때문에 NPE가 발생한다.

public static String convert(String s, int numRows) {
	if (numRows == 1) {
		return s;
	}

	ArrayList<String>[] arr = new ArrayList[numRows];

	for (int i = 0; i < numRows; i++) {
		arr[i] = new ArrayList<>();
	}
}

풀이

시작하는 행의 값을 0으로, 시작하는 문자열의 인덱스 값을 0으로 초기화 해주었다.

각 행의 첫 열에는 각 문자가 다 들어가므로, 하나씩 다 넣어주었다.

아래 사진을 참고하면, 지그재그로 문자가 어떻게 들어갔는지 확인 할 수 있다.때문에 행의 값을 -2 시켜주었다.왜냐하면, 이미 반복문 안의 반복문에서 행의 값을 ++시켜주었기 때문이다.

그리고 해당 행 부터 -1씩 감소시키며 문자들을 넣어주었다.

나중에 문자열 형태로 출력할 예정이며, 각 행별로 한개씩 들어가기 때문에 이렇게 넣어줘도 상관 없다고 판단하였다.

public static String convert(String s, int numRows) {
   
   ...

   int curRow = 0;
   int curStringIdx = 0;

   while(curStringIdx < s.length()) {
      while (curRow < numRows && curStringIdx < s.length()) {
         arr[curRow].add(String.valueOf(s.charAt(curStringIdx)));
         curRow++;
         curStringIdx++;
      }

      curRow -= 2;

      while (curRow > 0 && curStringIdx < s.length()) {
         arr[curRow].add(String.valueOf(s.charAt(curStringIdx)));
         curRow--;
         curStringIdx++;
      }
   }

   ...
}

 

출력

StringBuilder를 사용하여 하나씩 문자를 출력해주었다.

배열 리스트를 쓴 만큼 스트림을 시도해 보았으나, 배열 리스트 안의 배열이라 출력이 어려웠다.그래서 반복문을 돌려 하나씩 출력해주었다.

public static String convert(String s, int numRows) {
   ...

   StringBuilder sb = new StringBuilder();
   for (int i = 0; i < numRows; i++) {
      for (int j = 0 ; j < arr[i].size(); j++) {
         sb.append(arr[i].get(j));
      }
   }

   return sb.toString();
}

---

📌 최종 풀이

public class ZigzagConversion {
   public static void main(String[] args) {
      System.out.println(convert("PAYPALISHIRING", 3));
      System.out.println(convert("PAYPALISHIRING", 4));
   }
   public static String convert(String s, int numRows) {
      if (numRows == 1) {
         return s;
      }

      ArrayList<String>[] arr = new ArrayList[numRows];

      int curRow = 0;
      int curStringIdx = 0;

      for (int i = 0; i < numRows; i++) {
         arr[i] = new ArrayList<>();
      }

      while(curStringIdx < s.length()) {
         while (curRow < numRows && curStringIdx < s.length()) {
            arr[curRow].add(String.valueOf(s.charAt(curStringIdx)));
            curRow++;
            curStringIdx++;
         }

         curRow -= 2;

         while (curRow > 0 && curStringIdx < s.length()) {
            arr[curRow].add(String.valueOf(s.charAt(curStringIdx)));
            curRow--;
            curStringIdx++;
         }
      }

      StringBuilder sb = new StringBuilder();
      for (int i = 0; i < numRows; i++) {
         for (int j = 0 ; j < arr[i].size(); j++) {
            sb.append(arr[i].get(j));
         }
      }

      return sb.toString();
   }
}

---