[Python/파이썬] 백준 3613번 Java vs C++

3613번: Java vs C++

 

코드

input_str = input()

if input_str[0] == '_' or input_str[0].isupper() or input_str[-1] == '_':
    print("Error!")
    exit()

if input_str.count('_') > 0:    # C++ => Java
    new_str = ''
    underbar = False
    for i in range(len(input_str)):
        if input_str[i].isupper():
            new_str = "Error!"
            break
        if input_str[i].isalpha():
            if underbar:
                new_str += input_str[i].upper()
                underbar = False
            else:
                new_str += input_str[i]
        elif not underbar and input_str[i] == '_':
            underbar = True
        else:
            new_str = "Error!"
            break
else:   # Java => C++
    new_str = ''
    for i in range(len(input_str)):
        if input_str[i].isupper():
            new_str += ('_'+input_str[i].lower())
        elif input_str[i].islower():
            new_str += input_str[i]
        else:
            new_str = "Error!"
            break
print(new_str)

 

Error! 결과가 나오는 경우를 잘 처리해야 하는 문제이다.

 

1. _asd와 같이 맨 앞에 '_'가 들어가는 경우

2. asd_와 같이 맨 뒤에 '_'가 들어가는 경우

3. asd__asd와 같이 '_'가 연속하여 2개인 경우

4. asd_Asd와 같이 C++과 Java가 섞인 형태

5. AbcDef와 같이 맨 앞이 대문자인 경우

 

위와 같은 에러들을 처리해주기만 하면 파이썬으로는 쉽게 풀 수 있는 문제였다.