[자바] 백준 5426 - 비밀 편지 (java)

목차

문제 : boj5426

 

 

필요 알고리즘

• 구현, 문자열
  
  • 문제에 제시된 대로 구현해주면 된다!

※ 제 코드에서 왜 main 함수에 로직을 직접 작성하지 않았는지, 왜 Scanner를 쓰지 않고 BufferedReader를 사용했는지 등에 대해서는 '자바로 백준 풀 때의 팁 및 주의점' 글을 참고해주세요. 백준을 자바로 풀어보려고 시작하시는 분이나, 백준에서 자바로 풀 때의 팁을 원하시는 분들도 보시는걸 추천드립니다.

 

 

풀이

  문제에서 제시된 것 처럼 문자열을 배열 형태로 바꿀 수 있다면, 역으로 반시계방향으로 90도 회전해준 후, 그걸 다시 문자로 바꾸면 답이 나올 것이다.

 

  그럼 그대로 해보자. 우선은 정사각형이라고 했으므로, 배열의 크기는 문자열 길이의 제곱근이 된다.

int len = (int) Math.sqrt(encoded.length());

 

  배열의 한 변 길이가 len이라 한다면, 이제 반시계 방향으로 문자를 이동시켜야 하므로 (i, j) 위치에 있는 문자가 (len-j, i) 로 이동해야 한다. 이 때 i행 j열 위치는 문자로 따졌을 때 i*len+j 위치에 있는 문자가 된다. 따라서 말로 설명한걸 코드로 써보면 아래와 같다.

private String decode(String encoded) {
    int len = (int) Math.sqrt(encoded.length());

    StringBuilder result = new StringBuilder();
    for (int j = len-1; j >= 0; j--) {
        for (int i = 0; i < len; i++) {
            result.append(encoded.charAt(i*len+j));
        }
    }
    return result.toString();
}

 

  한방에 저렇게 전개되는게 잘 이해가 되지 않을 수 있다. 그렇다면 정말 배열에 데이터를 넣은 후 빼내는 방식으로 아래처럼 풀어도 된다. 위 decode() 함수와 하는 일 자체는 동일하다.

private String decode(String encoded) {
    int len = (int) Math.sqrt(encoded.length());

    char[][] arr = new char[len][len];
    // 그림에서 입력으로 들어온 문자를 배열로 바꾸는 부분
    for (int i = 0; i < len; i++) {
        for (int j = 0; j < len; j++) {
            arr[i][j] = encoded.charAt(i*len+j);
        }
    }

	// 반시계방향으로 90도 회전
    char[][] decodeArr = new char[len][len];
    for (int i = 0; i < len; i++) {
        for (int j = 0; j < len; j++) {
            decodeArr[len-j-1][i] = arr[i][j];	// (i, j) -> (len-j, i)
        }
    }

	// 반시계방향으로 90도 회전한걸 다시 문자로 변경
    StringBuilder result = new StringBuilder();
    for (int i = 0; i < len; i++) {
        for (int j = 0; j < len; j++) {
            result.append(decodeArr[i][j]);
        }
    }
    return result.toString();
}

 

 

코드 : github

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

    public static void main(String[] args) throws Exception {
        new Main().solution();
    }

    private void solution() throws Exception {
        int n = Integer.parseInt(br.readLine());
        StringBuilder sb = new StringBuilder();
        while (n-->0) {
            String encoded = br.readLine();
            sb.append(decode(encoded)).append('\n');
        }
        System.out.print(sb);
    }

    private String decode(String encoded) {
        int len = (int) Math.sqrt(encoded.length());

        StringBuilder result = new StringBuilder();
        for (int j = len-1; j >= 0; j--) {
            for (int i = 0; i < len; i++) {
                result.append(encoded.charAt(i*len+j));
            }
        }
        return result.toString();
    }
}