[백준/Python] 15663번 : N과 M (9)

2024. 9. 29. 21:20· Koala - 16기/코딩테스트 심화 스터디
목차
  1. 문제
  2. 풀이
  3. 코드

https://www.acmicpc.net/problem/15663

문제

N개의 자연수가 주어지고, M개를 고른 수열을 출력한다.

조건1. 중복되는 수열을 출력하면 안된다.
조건2. 사전 순으로 증가하는 순서로 출력한다.

풀이

사전 순으로 출력해야 했기 때문에 입력받은 N들을 sort() 로 정렬했다.

2가지를 고려해야 했다.

1. 똑같은 조합이 나오면 출력하면 안된다.

3 1
4 4 2


위의 입력의 경우, 결과가

2
4
4 # 중복이여서 안됨

 

위와 같이 똑같은 수열 4가 중복해서 출력되면 안된다.

2. 같은 수가 여러번 나올 수는 있다.
아래와 같은 입력의 경우

4 2
9 7 9 1

아래와 같이 숫자가 중복 될수는 있다. (9 9)

1 7
1 9
7 1
7 9
9 1
9 7
9 9 # 이건 가능

 

기본적으로 백트래킹을 이용해 풀었다.

1의 경우를 해결하기 위해, 같은 함수호출 수준에서  바로 앞에 어떤 것이 정답 배열(ans)에 있었는지 기억했다(prev).

2의 경우 때문에 set을 이용해 입력을 처리할 수 없었다.
또한 같은 숫자여도 다르게 취급해야 했기 때문에 defaultdict()를 활용했다. N들을 담은 리스트(li)에서의 인덱스를 key로 하여, ans에 들어간 경우 해당 value를 1로 바꿔주었다.

코드

from collections import defaultdict

n, m = map(int, input().split())

li = sorted(list(map(int, input().split())))

ans = []

def go(count, hist):
    if count >= m :
        print(' '.join(map(str,ans)))
        return
    else:
        prev = None
        for i, n in enumerate(li):
            if prev == n:
                continue
            else:
                if hist[i] == 1 : continue
                hist[i] = 1
                ans.append(n)
                go(count+1, hist.copy())

            prev = ans.pop(-1)
            hist[i] = 0

hist = defaultdict(int)
go(0, hist.copy())
저작자표시 (새창열림)

'Koala - 16기 > 코딩테스트 심화 스터디' 카테고리의 다른 글

[BOJ/Python3] 1065번: 한수  (0) 2024.09.30
백준15655 / 파이썬 / N과 M(6)  (0) 2024.09.29
[백준/C++] 1895번: 필터  (0) 2024.09.29
[백준/Python] 14888번: 연산자 끼워넣기  (0) 2024.09.29
[백준/Python] 14888번: 연산자 끼워넣기  (0) 2024.09.29
  1. 문제
  2. 풀이
  3. 코드
'Koala - 16기/코딩테스트 심화 스터디' 카테고리의 다른 글
  • [BOJ/Python3] 1065번: 한수
  • 백준15655 / 파이썬 / N과 M(6)
  • [백준/C++] 1895번: 필터
  • [백준/Python] 14888번: 연산자 끼워넣기
KauKoala
KauKoala
항공대 알고리즘 동아리 Koala 🥰
KauKoala
Koala
KauKoala
전체
오늘
어제
  • 분류 전체보기 (1888)
    • 공지 게시판 (10)
    • 정보 게시판 (8)
    • Codeforce (15)
    • acm-icpc (6)
    • Koala - 1기 (16)
    • Koala - 2기 (111)
      • Programming Contest (1)
      • A반 (20)
      • B반 (39)
      • C반 (22)
      • 기초 강의 (18)
    • Koala - 3기 (10)
      • 기초 스터디 (7)
    • Koala - 4기 (67)
    • Koala - 5기 (144)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (68)
    • Koala - 6기 (102)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (25)
      • 모의 테스트 스터디 (1)
    • Koala - 7기 (167)
      • 기초 알고리즘 스터디 (97)
      • 코딩테스트 준비 스터디 (68)
      • 모의 테스트 스터디 (1)
    • Koala - 8기 (44)
      • 기초 알고리즘 스터디 (32)
      • 코딩테스트 준비 스터디 (10)
      • 코드포스 버츄얼 스터디 (0)
      • 프로그래머스 LV2 스터디 (0)
    • Koala - 9기 (205)
      • 기초 알고리즘 스터디 (138)
      • 코딩테스트 준비 스터디 (64)
      • 모의테스트 준비 스터디 (1)
    • Koala - 10기 (117)
      • 기초 알고리즘 스터디 (30)
      • 코딩테스트 준비 스터디 (86)
      • 모의테스트 준비 스터디 (1)
    • Koala - 11기 (151)
      • 기초 알고리즘 스터디 (46)
      • 코딩테스트 준비 스터디 (104)
      • 모의테스트 준비 스터디 (1)
    • Koala - 12기 (86)
      • 기초 알고리즘 스터디 (31)
      • 코딩테스트 준비 스터디 (55)
    • Koala - 13기 (119)
      • 기초 알고리즘 스터디 (52)
      • 코딩테스트 준비 스터디 (67)
    • Koala - 14기 (116)
      • 기초 알고리즘 스터디 (39)
      • 코딩테스트 준비 스터디 (77)
    • Koala - 15기 (138)
      • 기초 알고리즘 스터디 (73)
      • 코딩테스트 준비 스터디 (65)
    • Koala - 16기 (47)
      • 코딩테스트 기초 스터디 (16)
      • 코딩테스트 심화 스터디 (31)
    • Koala - 17기 (62)
      • 코딩테스트 기초 스터디 (15)
      • 코딩테스트 심화 스터디 (47)
    • Koala - 18기 (31)
      • 코딩테스트 기초 스터디 (11)
      • 코딩테스트 심화 스터디 (20)
    • Koala - 19기 (42)
      • 코딩테스트 기초 스터디 (7)
      • 코딩테스트 심화 스터디 (35)
    • Koala - 20기 (0)
      • 코딩테스트 기초 스터디 (0)
      • 코딩테스트 심화 스터디 (0)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

  • 🐨항공대 알고리즘 학회 Koala 3기 모집
  • 🐨항공대 알고리즘 학회 Koala 2기 모집
  • 소모임 소개

인기 글

태그

  • 파이썬
  • C++
  • dp
  • dfs
  • BOJ
  • 백준
  • 백트래킹
  • BFS

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.3.0
KauKoala
[백준/Python] 15663번 : N과 M (9)
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.