[BOJ][🟡5][백준#03107] IPv6

문제 출처

BAEKJOON Online Judge #3107

문제

IPv6은 길이가 128비트인 차세대 인터넷 프로토콜이다. IPv6의 주소는 32자리의 16진수를 4자리씩 끊어 나타낸다. 이때, 각 그룹은 콜론 (:)으로 구분해서 나타낸다. 예를 들면, 다음과 같다.

2001:0db8:85a3:0000:0000:8a2e:0370:7334 32자리의 16진수는 사람이 읽고 쓰기에 불편하고, 대부분의 자리가 0이기 때문에 아래와 같이 축약할 수 있다.

각 그룹의 앞자리의 0의 전체 또는 일부를 생략 할 수 있다. 위의 IPv6을 축약하면, 다음과 같다

2001:db8:85a3:0:00:8a2e:370:7334

만약 0으로만 이루어져 있는 그룹이 있을 경우 그 중 한 개 이상 연속된 그룹을 하나 골라 콜론 2개(::)로 바꿀 수 있다.

2001:db8:85a3::8a2e:370:7334 2번째 규칙은 모호함을 방지하기 위해서 오직 한 번만 사용할 수 있다. 올바른 축약형 IPv6주소가 주어졌을 때, 이를 원래 IPv6 (32자리의 16진수)로 복원하는 프로그램을 작성하시오.

입력

첫째 줄에 올바른 IPv6 주소가 주어진다. 이 주소는 최대 39글자이다. 또한, 주소는 숫자 0-9, 알파벳 소문자 a-f, 콜론 :으로만 이루어져 있다.

출력

첫째 줄에, 입력으로 주어진 IPv6의 축약되지 않은 형태를 출력한다.

예제

예제 1

입력

25:09:1985:aa:091:4846:374:bb

출력

0025:0009:1985:00aa:0091:4846:0374:00bb

예제 2

입력

::1

출력

0000:0000:0000:0000:0000:0000:0000:0001

My Sol

txt = input()
arr0 = txt.split('::')
l0 = len(arr0)
if l0==1:
    pre_ret = arr0[0].split(':')
else:
    arr1 = arr0[0].split(':')
    arr2 = arr0[1].split(':')
    l1, l2 = len(arr1), len(arr2)
    pre_ret = [*arr1, *['' for _ in range(8-(l1+l2))], *arr2]

ret = []
for p in pre_ret:
    pl = len(p)
    ret.append('0'*(4-pl)+p)
print(':'.join(ret))

생각보다 굉장히 쉽게 풀어냈던 문제였다. 우선 :: 의 유무에 따라 다르게 문제에 접근한다고 생각하는 것이 우선이다. 또한 8개 원소로 구분될 수 있다는 사실을 인지하고, 마지막으로 처리되기 이전의 8개 원소의 리스트 각각의 원소 길이가 4가 될 때까지 원소 앞에 ‘0’을 붙여준다고 생각한다.

즉, 1차적인 목표는 8개 원소를 가진 각각의 ‘0’을 붙이기 이전의 원소를 담은 리스트이다. 이를 pre_ret으로 두겠다.

1. 입력을 받아 처리한다.
2. arr0은 ::으로 입력 txt를 분리한 것이다. 만약 txt에 ::이 포함되어 있다면 그 앞과 뒤로 자른 2개의 원소를 가지는 list가, 아니라면 그 자체로 1개의 원소를 가지는 list가 되었을 것이다. 이때의 arr0의 길이를 l0으로 둔다.
3. 만약 l0이 1이면 ‘::’이 없으므로 ‘:’로 자른 list를 pre_ret으로 둔다.
4. 만약 l0이 2라면, ‘::’으로 한 번 잘렸다는 뜻이므로, 앞과 뒤를 ‘:’로 자른 것을 각각 arr1과 arr2로 둔다. 그리고 각 리스트의 길이를 l1, l2로 두겠다.
5. 두 리스트 사이에서 :: 속에 몇 개의 연속된 ‘0’들이 함축되어 있는지 알 수 없다. 다만 원소는 8개이고, l1과 l2를 알기 때문에 8에서 l1과 l2를 뺀 것이 ‘0’의 연속 개수임을 알 수 있겠다.
6. pre_ret은 arr1, 0들의 연속, arr2 이므로 이를 list spread를 활용해 pre_ret = [*arr1, *['' for _ in range(8-(l1+l2))], *arr2]로 하나의 리스트로 다시 묶어주었다.
7. 이 후 이렇게 준비된 pre_ret으로부터 각 원소가 4자리가 될 때까지 ‘0’을 붙여주어 ret에 하나씩 넣는다.
8. 이를 ‘:’을 seperation으로 하여 하나의 문자열로 묶은 뒤, 출력한다.

결과

맞았습니다!!