[TIL] 백준 - 덩치 / 빗물

[주절주절]

SQLD 공부하고~ 알고리즘 풀고~ 야구 보고~ 인터뷰 준비하고~ (인터뷰 확정 메일은 아직 못받은게 함정 뭐야 얼른 알려줘요)

 

[Today I Learned]

# 백준 - 덩치

문제
우리는 사람의 덩치를 키와 몸무게, 이 두 개의 값으로 표현하여 그 등수를 매겨보려고 한다. 어떤 사람의 몸무게가 x kg이고 키가 y cm라면 이 사람의 덩치는 (x, y)로 표시된다. 두 사람 A 와 B의 덩치가 각각 (x, y), (p, q)라고 할 때 x > p 그리고 y > q 이라면 우리는 A의 덩치가 B의 덩치보다 "더 크다"고 말한다. 예를 들어 어떤 A, B 두 사람의 덩치가 각각 (56, 177), (45, 165) 라고 한다면 A의 덩치가 B보다 큰 셈이 된다. 그런데 서로 다른 덩치끼리 크기를 정할 수 없는 경우도 있다. 예를 들어 두 사람 C와 D의 덩치가 각각 (45, 181), (55, 173)이라면 몸무게는 D가 C보다 더 무겁고, 키는 C가 더 크므로, "덩치"로만 볼 때 C와 D는 누구도 상대방보다 더 크다고 말할 수 없다.
N명의 집단에서 각 사람의 덩치 등수는 자신보다 더 "큰 덩치"의 사람의 수로 정해진다. 만일 자신보다 더 큰 덩치의 사람이 k명이라면 그 사람의 덩치 등수는 k+1이 된다. 이렇게 등수를 결정하면 같은 덩치 등수를 가진 사람은 여러 명도 가능하다. 아래는 5명으로 이루어진 집단에서 각 사람의 덩치와 그 등수가 표시된 표이다.
위 표에서 C보다 더 큰 덩치의 사람이 없으므로 C는 1등이 된다. 그리고 A, B, D 각각의 덩치보다 큰 사람은 C뿐이므로 이들은 모두 2등이 된다. 그리고 E보다 큰 덩치는 A, B, C, D 이렇게 4명이므로 E의 덩치는 5등이 된다. 위 경우에 3등과 4등은 존재하지 않는다. 여러분은 학생 N명의 몸무게와 키가 담긴 입력을 읽어서 각 사람의 덩치 등수를 계산하여 출력해야 한다.

 

@solution.py

차분히 생각해보면 어렵지 않은 문제인데, 첨부터 나이스하게 풀어보겠다고 (나이스한 방법이 아니었는데! 😡) itertools combination 불러오고 Counter 불러오고... 그랬다. 근데 Counter로 공동 순위가 있는 등수 처리하는 코드 짜는게 생각보다 어려워서 쩔쩔매다가 ㅠ_ㅠ 처음으로 돌아가서 다시 생각해봤는데, 그냥 쌩 반복문 쓰면서 한 큐에 처리하는게 제일 깔끔할 것 같아서, 바로 그렇게 풀었다.

bodies = [list(map(int,input().split())) for _ in range(int(input()))]

for idx, b in enumerate(bodies):

    rank = 1

    for idx2, c in enumerate(bodies):
        if idx == idx2:
            pass

        if (b[0] < c[0]) and (b[1] < c[1]):
            rank += 1

    print(rank, end=' ')

기본을 잊지말자! 다짐하게 되었던 문제.

 

# 백준 - 빗물

문제
첫 번째 줄에는 2차원 세계의 세로 길이 H과 2차원 세계의 가로 길이 W가 주어진다. (1 ≤ H, W ≤ 500)
두 번째 줄에는 블록이 쌓인 높이를 의미하는 0이상 H이하의 정수가 2차원 세계의 맨 왼쪽 위치부터 차례대로 W개 주어진다.
따라서 블록 내부의 빈 공간이 생길 수 없다. 또 2차원 세계의 바닥은 항상 막혀있다고 가정하여도 좋다.

 

@solution.py

예전에 풀었던 빗물 트래핑과 매우 흡사한 문제. 스택으로 끙끙 대면서 풀었었는데... 구글링하다 엄청 간단한 풀이를 발견해서 기록해본다.

h, w = map(int, input().split())
blocks = list(map(int, input().split()))

water = 0

for i in range(1, w-1):

    left_max = max(blocks[:i])
    right_max = max(blocks[i+1:])

    height = min(left_max, right_max)

    if blocks[i] < height:
        water += height -  blocks[i]

print(water)

아이디어는 단순하다. i번째 포인트를 기준으로 왼쪽에서 가장 높은 벽, 오른쪽에서 가장 높은 벽 중 더 작은 값을 구하고, 그 값보다 i번째 벽의 높이가 더 낮으면 그 차이만큼 water를 더하는 식이다. 

어떤 식으로 동작하는지 이해는 되었는데, 직관적으로 아직 팍! 이해는 못했다. 알쏭달쏭 알고리즘... 이렇게 공부하는게 맞는지, 참 어렵다. : /