[SWEA] 13428. 숫자 조작(python)

📌문제

💪아이디어

1. 수 나열하기
   인덱스별로 검사하면서 입력된 string[i]와 작은 수(최솟값의 경우)/큰 수(최댓값의 경우) 비교하다가 해당인덱스의 갑소가 수가 다르면 자리를 바꿔준다. string에 작은 수와 큰 수를 비교하기위하여 min heap과 max heap을 사용하였다
2. 0 처리하기
   입력값이 0이 들어온 경우를 제외하고는 인덱스 0에 ‘0’이 올 수 없다. 따라서 최솟값을 처리할 때 min heap에 0이 있을 수 있으니 인덱스 0을 검사 후에 min heap에 0을 넣도록 하였다. max heap은 heap을 생성할 떄 0을 넣어도 상관이 없으나 함수의 재사용을 위하여 max heap생성시에 0을 넣어주지 않았다.
   
3. 자리 바꾸기
   중복된 수가 입력값에 없으면 상관이 없는데 중복된 수가 있는 경우에는 가장 멀리있는 인덱스의 값과 바꿔야한다. 따라서 해당 값이 몇 번 인덱스에 있는지 알기 위하여 defaultdict을 사용하였다.
   506060 이 들어왔다고 가정하면, 최솟값으로 500066이 나와야 하기 때문에 ‘0’이 존재하는 인덱스 중에 가장 멀리 있는 인덱스를 갖고오기위하여 index_dic[‘0’][-1]을 사용한 것이다.
   

🥂코드

from collections import defaultdict
from heapq import heappush,heappop

def Find_val(string,heap,sign):
    answer=list(string)
    for i,num in enumerate(string):
        if heap:
            val = str(heappop(heap)*sign)
            if num!=val:
                inx = index_dic[val][-1]
                answer[inx]=answer[i]
                answer[i]=val
                break
        if i==0 and index_dic['0']:
            for _ in index_dic['0']:
                heappush(heap,0)
    return ''.join(answer)

T=int(input())
for tc in range(1,T+1):
    string = input()
    min_heap, max_heap=[],[]
    index_dic=defaultdict(list)

    for inx,num in enumerate(string):
        if num !='0':
            heappush(min_heap,int(num))
            heappush(max_heap,int(num)*(-1))
        index_dic[num].append(inx)

    min_answer=Find_val(string,min_heap,1)
    max_answer=Find_val(string,max_heap,-1)
    
    print('#{} {} {}'.format(tc,min_answer,max_answer))