728x90
반응형
def cc(m, sang):
result = []
loc = []
sang = numpy.array(sang)
for a in m:
loc.append(numpy.where(sang == a)[0])
loc.sort()
#for b in range(1, len(loc)):
# print(loc[b])
for b in range(len(sang)):
if b in loc:
result.append(1)
else:
result.append(0)
print(*result)
if __name__ == '__main__':
warnings.filterwarnings(action='ignore', category=DeprecationWarning)
cc([6, 3, 2, 10, -10], [10, 9, -5, 2, 3, 4, 5, -10])
백준에서는 numpy 사용 불가라고 한다
728x90
반응형
'프로그래밍 > 알고리즘' 카테고리의 다른 글
[프로그래머스] K번째 수(파이썬) (0) | 2023.06.22 |
---|---|
[프로그래머스] 명예의 전당 (1) (파이썬) (0) | 2023.06.21 |
[프로그래머스] 폰켓몬(파이썬) (0) | 2023.06.21 |
[프로그래머스] 푸드 파이트 대회(파이썬) (0) | 2023.06.21 |
백준 1373 (0) | 2022.07.07 |
댓글