글 읽기 - 1522번 질문있습니다.

smkm1568 1년 전 0

런타임에러(Index Error)가 나오는데

반례가 어떤 것이 있는지 궁금해서 여쭤봅니다!

import sys
input = sys.stdin.readline

n = input().strip()
queue = []

total = 0
shift_count = []
a_count = []
moving = 0
a_start = 0

if n[0] == 'a' and n[-1] == 'a' and len(n) != 1:
    n = n[:-1]

for i in n :
    if i == 'a' :
        a_start += 1
        if moving != 0 :
            shift_count.append(moving)
            moving = 0
    else :
        moving += 1
        if a_start != 0 :
            a_count.append(a_start)
            a_start = 0

if a_start != 0 :
    a_count.append(a_start)

if moving != 0 :
    shift_count.append(moving)

total_a = 0

for a in a_count :
    total_a += a


while True :
    if len(a_count) == 1 or len(a_count) == 0 or len(n) == 0: break
    elif len(a_count) == 2 :
        total += min(shift_count)
        break
    else :
        a_list = []
        for i in range(len(a_count)) :
            if i == len(a_count) - 1 :
                if a_count[i] + a_count[0] >= shift_count[i] :
                    a_list.append([a_count[0] + a_count[i], shift_count[i] , i])
            else :
                if a_count[i] + a_count[i + 1] >= shift_count[i] :
                    a_list.append([a_count[i] + a_count[i + 1], shift_count[i], i])

        a_list.sort(key = lambda a_list : (-a_list[0] , a_list[1]))

        if a_list[0][0] == total_a : break
        idx = a_list[0][2]

        a_min = min(a_count)
        if a_min > shift_count[idx] :
            a_count[idx] = a_count[idx] + a_count[idx + 1] + shift_count[i]
            total += shift_count[idx]
            del a_count[idx + 1]
            del shift_count[idx]
            a_count[a_count.index(a_min)] -= shift_count[idx]

        elif a_min == shift_count[idx] :
            a_count[idx] = a_count[idx] + a_count[idx + 1] + shift_count[i]
            total += shift_count[idx]
            del a_count[idx + 1]
            del shift_count[idx]
            a_count.remove(a_min)
        else :
            a_count[idx] += a_min
            total += a_min
            a_count.remove(a_min)
            shift_count[idx] -= a_min

print(total)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

1522번 질문있습니다.

smkm1568 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말