코드포스 131A 풀이

a=input()
a=[x for x in a]
flag= True
if a[0].islower() == True:
    
    for i in range(len(a)-1):
        if a[i+1].islower()==True:
            flag = False
    for j in range(len(a)-1):
        if flag:
            a[j+1]=a[j+1].lower()
    if flag:  
        a[0]=a[0].upper()  
        
else:
    for i in range(len(a)-1):
        if a[i+1].islower()==True:
            flag=False
    for i in range(len(a)):
        if flag:
            a[i]=a[i].lower()
print("".join(a))

문제가 첫번째만 소문자고 나머지는 다 대문자일 경우와 전부 대문자일 경우만 수정을 하라고 애매하게 되있어서 좀 애를 먹었다. 경우를 잘 나눠서 오류가 나면 answer를 확인하고 고쳤다.