[Python] 랜덤 리스트 비교

백준 알고리즘을 풀다가, 그냥 만들어보고 싶어져서 만들어 본 코드

 

조건 

- 첫째 줄엔 정수 N을 받아옴

- 1부터 N까지의 숫자가 랜덤으로 담긴 리스트를 출력

- 정수 M을 받아옴

- 1부터 M까지의 숫자가 랜덤으로 담긴 리스트를 출력

- 두 리스트의 값을 순서대로 비교했을 때 같으면 1 출력 다르면 0 출력

 

import random

N = int(input())
a = list(range(1,N+1))
sampleList1 = random.sample(a, N)
print(sampleList1)

M = int(input())
b = list(range(1,M+1))
sampleList2 = random.sample(b, M)
print(sampleList2)

if N > M :
  for k in range(M):
    if sampleList1[k] == sampleList2[k]:
      print(1)
    else:
      print(0)