본문 바로가기
프로그래밍/알고리즘

백준 10815

by laoching 2023. 10. 22.
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
반응형

댓글