프로그래머스 - 161990번: 바탕화면 정리 [Java]

문제: https://school.programmers.co.kr/learn/courses/30/lessons/161990

프로그래머스

1. 코드

import java.util.*;
class Solution {
    public int[] solution(String[] wallpaper) {
        int[] answer = new int[4];
        Arrays.fill(answer, -1);
        for(int i = 0; i < wallpaper.length; i++) {
            if(answer[0] != -1 && answer[2] != -1)
                break;
            if(wallpaper[i].contains("#") && answer[0] == -1)
                answer[0] = i;
            if(wallpaper[wallpaper.length - i - 1].contains("#") && answer[2] == -1)
                answer[2] = wallpaper.length - i;
        }
        
        for(int i = 0; i < wallpaper[0].length(); i++) {
            for(int j = 0; j < wallpaper.length; j++) {
                if(answer[1] != -1 && answer[3] != -1)
                    break;
                if(wallpaper[j].charAt(i) == '#' && answer[1] == -1)
                    answer[1] = i;
                if(wallpaper[j].charAt(wallpaper[0].length() - i - 1) == '#' && answer[3] == -1)
                    answer[3] = wallpaper[0].length() - i;
            }
        }
        return answer;
    }
}

2. 설명

int[] answer = new int[4];
Arrays.fill(answer, -1);

좌표의 수는 2개이므로 두 개의 좌표가 저장될 4개의 배열을 선언한 뒤 -1로 채워줍니다.

for(int i = 0; i < wallpaper.length; i++) {
    if(answer[0] != -1 && answer[2] != -1)
        break;
    if(wallpaper[i].contains("#") && answer[0] == -1)
        answer[0] = i;
    if(wallpaper[wallpaper.length - i - 1].contains("#") && answer[2] == -1)
        answer[2] = wallpaper.length - i;
}

해당 반복문을 통해서 행을 기준으로 두 좌표의 x 값을 찾을 것 입니다.  두 개의 좌표를 찾으면 반복문을 종료를 하며 두 번째 if문은 위에서 아래로 진행하여 첫 번째 x좌표를 저장합니다. 세 번째 if문은 아래에서 위로 진행하여 두 번째 x 좌표를 찾아서 저장합니다.

for(int i = 0; i < wallpaper[0].length(); i++) {
    for(int j = 0; j < wallpaper.length; j++) {
        if(answer[1] != -1 && answer[3] != -1)
            break;
        if(wallpaper[j].charAt(i) == '#' && answer[1] == -1)
            answer[1] = i;
        if(wallpaper[j].charAt(wallpaper[0].length() - i - 1) == '#' && answer[3] == -1)
            answer[3] = wallpaper[0].length() - i;
    }
}

해당 반복문을 통해서 열을 기준으로 두 좌표의 y 값을 찾을 것입니다. 두 번째 if문은 왼쪽에서 오른쪽으로 진행하여 첫 번째 y 좌표를 저장합니다. 세 번째 if문은 오른쪽에서 왼쪽으로 진행되며 두 번째 y 좌표를 찾아서 저장합니다.

3. 정리

1. 배열의 접근 방법에 대해 알면 쉽게 해결 할 수 있다.

출처: 프로그래머스 코딩 테스트 연습, 
https://school.programmers.co.kr/learn/challenges