Pink Rose Flower

Python/Beakjoon

6603번 : 로또

hyunjoo 2021. 8. 7. 14:45

6603번: 로또 (acmicpc.net)

 

6603번: 로또

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로

www.acmicpc.net

 

>>

itertools 라이브러리를 이용하여 조합 구하기

k와 s의 원소를 하나의 리스트에 모두 입력받고 첫번째 원소는 k 변수에 저장하고 나머지는 s집합 리스트에 저장한다.

s의 집합 원소들을 조합함수를 이용하여 6개 뽑을 수 있는 경우의 수를 모두 출력한다.

 

 

from itertools import combinations


while True:
    a=list(map(int,input().split()))
    k=a[0]
    if k==0:
        break
    s=a[1:]

    for i in combinations(s,6):
        for j in i:
            print(j,end=' ')
        print('')
    print('')
반응형

'Python > Beakjoon' 카테고리의 다른 글

2953번: 나는 요리사다  (0) 2021.08.07
1026: 보물  (0) 2021.08.07
1010번: 다리놓기  (0) 2021.08.06
3052번: 나머지  (0) 2021.08.04
4153번: 직각삼각형  (0) 2021.08.04