카카오 신입 공채 1차 코딩 테스트 문제 7. 추석 트래픽(난이도: 상) (Python)

2017/11/03 - [공부] - 카카오 신입 공채 1차 코딩 테스트 문제 7. 추석 트래픽(난이도: 상)

Java에 이어 Python으로도 작성해 봤다.

def question7(line):
    total = [[0 for col in range(2)] for row in range(len(line))]
 
    for i in range(len(line)):
        s = line[i]
 
        if len(s) < 1 or len(s) > 2000:
            return None
 
        hh = 0
        mm = 0
        ss = 0
 
        temp_log = s.split(' ')
        temp_time = temp_log[1].split(':')
 
        hh = int(temp_time[0]) * 1000
        mm = int(temp_time[1]) * 1000
        ss = int(float(temp_time[2]) * 1000)
 
        total[i][0] = (hh * 60 * 60) + (mm * 60) + ss
        total[i][1] = int(float(temp_log[2].replace('s', '')) * 1000)
 
    max_traffic = 0
 
    for i in range(len(total)):
        temp_traffic = 0
 
        for j in range(len(total)):
            if total[i][0] - total[i][1] + 1 - 999 \
                    <= total[j][0] - total[j][1] + 1 \
                    <= total[i][0] - total[i][1] + 1:
                temp_traffic += 1
            elif total[i][0] - total[i][1] + 1 - 999 \
                    <= total[j][0] \
                    <= total[i][0] - total[i][1] + 1:
                temp_traffic += 1
            elif total[i][0] - total[i][1] + 1 - 999 > total[j][0] - total[j][1] \
                    and total[i][0] - total[i][1] + 1 < total[j][0]:
                temp_traffic += 1
 
            if max_traffic < temp_traffic:
                max_traffic = temp_traffic
 
    return max_traffic