hashmap과 for문의 접목

 

def solution(rsp):
    d = {'0':'5','2':'0','5':'2'}
    return ''.join(d[i] for i in rsp)

 

 

파이썬의 장점은 for문을 한줄로 줄일 수 있다는 것...

조금씩 익숙하게 만들어서 더 효율적으로 for문을 쓸 수 있도록 노력해야겠다.

 

def solution(rsp):
    temp = ""
    ans = {"2":"0", "0":"5", "5":"2"}
    for i in range(len(rsp)):
        for a,b in ans.items() :
            if rsp[i] == a :
                temp += b
    return temp