🔗 문제 링크

컬러볼

✔️ 소요된 시간

2H

✨ 수도 코드

골2 호기롭게 건드렸다가 맞왜틀 지옥에 갇혔습니다..

일단 문제를 보시면,

1. 완탐으로는 접근이 어렵다.
2. 이거 정렬 때리면 누적합으로 풀리겠는데?
   까지는 떠올리실 것 같습니다.

다만,,,

1. 이전 공과 크기는 같지만 색깔이 다른 경우
2. 이전 공과 크기는 다르지만 색깔이 같은 경우
3. 이전 공과 크기와 색깔이 모두 같은 경우
4. 이전 공과 크기와 색깔이 모두 다른 경우

의 로직을 다르게 처리하는 것이 이 문제의 핵심입니다.

제 생각에는,
정렬을 했냐 안했냐, 어떤 기준으로 정렬을 하였는가 에 따라 4가지 로직을 처리하는 방식이 다소 다를 것 같습니다.

일단 저는 공의 크기와 공의 색깔을 기준으로 정렬을 하였습니다.

이 경우에는
같은 색깔의 공의 합과 같은 사이즈의 공의 합을 따로 관리해주어야 합니다.

대부분의 케이스는
(사로 잡을 수 있는 공의 크기) = (전체 공의 크기) - (같은 컬러의 크기) - (같은 사이즈의 크기)
로 정리됩니다.

다만, 3번의 경우에는
이전 dp를 그대로 복사해주기만 하면 된다는 것을 알 수 있습니다.

즉,

0. 공의 크기, 색깔, 인덱스를 따로 저장한다. (info)
1. 공의 크기와, 색깔에 따라 오름차순으로 정렬한다.
2. 이전 공과 비교하여 동일한 공이면, dp를 복사한다.
3. 동일한 공이 아니면, (전체 공의 크기) - (같은 컬러의 크기) - (같은 사이즈의 크기)를 계산하여 dp에 저장한다.
4. 동일한 공인지 확인하기 위해 공의 정보를 pre_size, pre_color에 저장한다.
5. dp를 출력한다.

이 흐름대로 코드를 작성하였습니다.

추가로!!!
출력시에 주의할 점은

들어온 공의 인덱스에 맞게, 출력해주어야 한다는 것입니다.

저의 경우에는 sort를 사용하였기에, 공의 번호 순서가 바뀌게 됩니다.

따라서, 기존 공의 index를 사용할 수 없기에
입력 처리할 때, +[i]를 통해서 index를 따로 저장하였습니다.

# 3
# 1 4
# 2 4
# 1 4

# 6  
# 1 5
# 2 6
# 7 6
# 5 6
# 2 6
# 1 6

# 5
# 3 15
# 3 15
# 3 5
# 1 15
# 2 10

📚 새롭게 알게된 내용