본문 바로가기

코딩테스트 with PYTHON

[python] 코딩테스트 대비 - 그리디(역수열)

문제)

 

답안코드)

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import sys
#sys.stdin=open("input.txt""r")
n=int(input())
a=list(map(int, input().split()))
seq=[0]*n#정답 배열
for i in range(n):
    for j in range(n):
        if(a[i]==0 and seq[j]==0):#배열 값이 0이고, 역수열 값도 0이라면
            seq[j]=i+1#삽입
            break
        elif seq[j]==0:#배열 값이 0이 아닌데, 공간이 남았다면,
            a[i]-=1#배열 값을 줄이고 다음칸으로 이동 하며 위치 찾기
 
for x in seq:
    print(x, end=' ')
cs