[프로그래머스] 🏗[PCCP 기출문제] 2번 / 석유 시추

Chobby·2023년 12월 12일
1

Programmers

목록 보기
317/349

😀문제 설명

[본 문제는 정확성과 효율성 테스트 각각 점수가 있는 문제입니다.]

세로길이가 n 가로길이가 m인 격자 모양의 땅 속에서 석유가 발견되었습니다. 석유는 여러 덩어리로 나누어 묻혀있습니다. 당신이 시추관을 수직으로 단 하나만 뚫을 수 있을 때, 가장 많은 석유를 뽑을 수 있는 시추관의 위치를 찾으려고 합니다. 시추관은 열 하나를 관통하는 형태여야 하며, 열과 열 사이에 시추관을 뚫을 수 없습니다.

예를 들어 가로가 8, 세로가 5인 격자 모양의 땅 속에 위 그림처럼 석유가 발견되었다고 가정하겠습니다. 상, 하, 좌, 우로 연결된 석유는 하나의 덩어리이며, 석유 덩어리의 크기는 덩어리에 포함된 칸의 수입니다. 그림에서 석유 덩어리의 크기는 왼쪽부터 8, 7, 2입니다.

시추관은 위 그림처럼 설치한 위치 아래로 끝까지 뻗어나갑니다. 만약 시추관이 석유 덩어리의 일부를 지나면 해당 덩어리에 속한 모든 석유를 뽑을 수 있습니다. 시추관이 뽑을 수 있는 석유량은 시추관이 지나는 석유 덩어리들의 크기를 모두 합한 값입니다. 시추관을 설치한 위치에 따라 뽑을 수 있는 석유량은 다음과 같습니다.

시추관의 위치	획득한 덩어리	총 석유량
1	[8]	8
2	[8]	8
3	[8]	8
4	[7]	7
5	[7]	7
6	[7]	7
7	[7, 2]	9
8	[2]	2

오른쪽 그림처럼 7번 열에 시추관을 설치하면 크기가 7, 2인 덩어리의 석유를 얻어 뽑을 수 있는 석유량이 9로 가장 많습니다.

석유가 묻힌 땅과 석유 덩어리를 나타내는 2차원 정수 배열 land가 매개변수로 주어집니다. 이때 시추관 하나를 설치해 뽑을 수 있는 가장 많은 석유량을 return 하도록 solution 함수를 완성해 주세요.


😁제한사항

1 ≤ land의 길이 = 땅의 세로길이 = n ≤ 500
1 ≤ land[i]의 길이 = 땅의 가로길이 = m ≤ 500
land[i][j]는 i+1행 j+1열 땅의 정보를 나타냅니다.
land[i][j]는 0 또는 1입니다.
land[i][j]가 0이면 빈 땅을, 1이면 석유가 있는 땅을 의미합니다.
정확성 테스트 케이스 제한사항
1 ≤ land의 길이 = 땅의 세로길이 = n ≤ 100
1 ≤ land[i]의 길이 = 땅의 가로길이 = m ≤ 100
효율성 테스트 케이스 제한사항
주어진 조건 외 추가 제한사항 없습니다.


😂입출력 예

land	result
[[0, 0, 0, 1, 1, 1, 0, 0], [0, 0, 0, 0, 1, 1, 0, 0], [1, 1, 0, 0, 0, 1, 1, 0], [1, 1, 1, 0, 0, 0, 0, 0], [1, 1, 1, 0, 0, 0, 1, 1]]	9
[[1, 0, 1, 0, 1, 1], [1, 0, 1, 0, 0, 0], [1, 0, 1, 0, 0, 1], [1, 0, 0, 1, 0, 0], [1, 0, 0, 1, 0, 1], [1, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1]]	16

🤣입출력 예 설명

입출력 예 #1

문제의 예시와 같습니다.

입출력 예 #2

시추관을 설치한 위치에 따라 뽑을 수 있는 석유는 다음과 같습니다.

시추관의 위치	획득한 덩어리	총 석유량
1	[12]	12
2	[12]	12
3	[3, 12]	15
4	[2, 12]	14
5	[2, 12]	14
6	[2, 1, 1, 12]	16

6번 열에 시추관을 설치하면 크기가 2, 1, 1, 12인 덩어리의 석유를 얻어 뽑을 수 있는 석유량이 16으로 가장 많습니다. 따라서 16을 return 해야 합니다.


😃제한시간 안내

정확성 테스트 : 10초
효율성 테스트 : 언어별로 작성된 정답 코드의 실행 시간의 적정 배수


😎나의풀이

포인트는,
1. 석유에 인덱스를 붙여 동일 석유인지 판단할 것
2. dfs를 재귀 방식으로 호출하지 않고 스택에 담아 사용할 것 (5번 테스트 케이스)
3. Y-Wise님의 게시물확인

function solution(land) {
    const xMax = land[0].length-1
    const yMax = land.length-1
    const visited = Array.from(Array(yMax+1), () => Array(xMax+1).fill(0))
    const history = []
    // 석유에 인덱싱을 함(1부터 시작)
    let oilIdx = 1
    let oilDict = {}
    let maxOil = Number.MIN_SAFE_INTEGER
    // 해당 좌표에 접근할 수 있는가 확인
    function IsInBoundary(x, y) {
        if(x < 0 || x > xMax) return false
        if(y < 0 || y > yMax) return false
        return true
    }
    // 연결된 석유 값 붙이기 작업
    function dfs(x, y) {
        const stack = [[x, y]]
        let oil = 0
        while(stack.length) {
            const [curX, curY] = stack.pop()
            if(!IsInBoundary(curX, curY) || visited[curY][curX] || land[curY][curX] === 0) continue
            visited[curY][curX] = 1
            oil += land[curY][curX]
            land[curY][curX] = oilIdx
            stack.push([curX, curY-1], [curX, curY+1], [curX-1, curY], [curX+1, curY])
        }
        return oil
    }
    // dfs를 통해 연속된 땅은 모두 같은 오일을 갖도록 수정
    for(let i = 0 ; i <= yMax ; i ++) {
        for(let j = 0 ; j <= xMax ; j++) {
            const curOil = dfs(j, i)
            if(!curOil) continue
             while(history.length) {
                const [hisX, hisY] = history.pop()
                land[hisY][hisX] = oilIdx
            }
            oilDict[oilIdx] = curOil
            oilIdx++
        }
    }
    // 기름의 합계를 구함
    for(let j = 0; j <= xMax; j++) {
        let curLineOil = 0
        let prevOil = {}
        for(let i = 0; i <= yMax; i++) {
            if(prevOil[land[i][j]]) continue
            if(land[i][j] !== 0) {
                prevOil[land[i][j]] = true
                curLineOil += oilDict[land[i][j]]
            }
        }
        if(curLineOil > maxOil) maxOil = curLineOil
    }
    
    return maxOil
}
profile
내 지식을 공유할 수 있는 대담함

0개의 댓글