글 읽기 - 파이썬 반례 질문

girlera7 8일 전 0

예제는 다 맞고, 조건들을 잘 충족시켜서 짰다고 생각했는데 혹시 어디가 문제인지 알 수 있을까요?

29% 에서 틀렸다고 나옵니다..

from functools import cmp_to_key

N = int(input())
str_arr = []


for _ in range(N):
    cur_str = input()

    cur_lst = []
    cur_digit = ""

    for elem in cur_str:
        if elem.isdigit():
            cur_digit += elem
        else:
            if len(cur_digit) != 0:
                cur_lst.append(cur_digit)
                cur_digit = ""
            cur_lst.append(elem)
    if len(cur_digit) != 0:
        cur_lst.append(cur_digit)
    str_arr.append(cur_lst)

#print(str_arr)

def front_zero_num(digits):
    count = 0

    for elem in digits:
        if elem != "0":
            break
        else:
            count += 1

    return count

def compare(item1, item2):

    length = min(len(item1), len(item2))

    for i in range(length):
        if item1[i] == item2:
            continue
        if not item1[i].isdigit() and not item2[i].isdigit():
            if item1[i].lower() < item2[i].lower():
                return -1
            elif item1[i].lower() == item2[i].lower():
                if item1[i] > item2[i]:
                    return 1
                elif item1[i] < item2[i]:
                    return -1
                else:
                    continue
            else:
                return 1

        elif item1[i].isdigit() and not item2[i].isdigit():
            return -1
        elif item2[i].isdigit() and not item1[i].isdigit():
            return 1
        else:
            if int(item1[i]) < int(item2[i]):
                return -1
            elif int(item1[i]) > int(item2[i]):
                return 1
            else:
                count1 = front_zero_num(item1[i])
                count2 = front_zero_num(item2[i])
                if count1 > count2:
                    return 1
                elif count1 < count2:
                    return -1
                else:
                    continue

    return 0
# Calling

str_arr = sorted(str_arr, key=cmp_to_key(compare))


for str_lst in str_arr:
    result = ""
    for elem in str_lst:

        if isinstance(elem, str):
            result += elem
        else:
            result += elem

    print(result)

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

문제

문제

출처

ICPC

파이썬 반례 질문

girlera7 8일 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회 / 고등학교 대회

출처

대학교 대회

도움말