🐹 IPv6

1. 문제 링크

3107번: IPv6

2. 코드

Python3 31120KB 40ms

split_ip = input().split(':')

# 규칙 1
for i in range(len(split_ip)):
    # ::으로 축약된 경우가 아니고, 축약되어있는 경우
    if split_ip[i] != '' and len(split_ip[i]) != 4:
        # 0을 추가해 원상태로 복구
        split_ip[i] = '0' * (4 - len(split_ip[i])) + split_ip[i]

# 규칙 2(해당하는 경우에만)
# 0으로만 이루어진 형태가 ::으로 축약된 경우 있는지 확인
num = 0 # 축약되지 않은 부분의 개수
if '' in split_ip:
    for i in range(len(split_ip)):
        if split_ip[i] != '':
            num += 1
        else:
            idx = i # 규칙2가 사용된 위치
    split_ip[idx] = '0000' * (8 - num)
    
ip = ''.join(split_ip)
answer = ''
for i in range(0, len(ip), 4):
    answer += ip[i:i+4]
    # 마지막에만 : 안 붙임
    if i != 28:
        answer += ':'
print(answer)

3. 해설

규칙1은 모두 해당되는 경우이고, 규칙 2는 0으로만 이루어진 그룹이 1개이상 있을 경우이므로 규칙 1은 모두에게 적용해준 뒤, 규칙 2는 해당되는 경우 마지막에 처리해준다.

:을 기준으로 배열을 나눈 후 빈 칸이 생기는 경우가 규칙 2의 경우이다.

빈칸이 아닌 칸의 개수를 세고 8에서 빼주면 0으로만 이루어진 그룹의 개수가 나오므로 빈 칸 위치에 ‘0000’ 그룹을 넣어준다.

나눠놓은 값을 모두 합쳐준 후 4칸마다 :을 넣어준다. (이때 마지막에는 넣지 않는다.)