CS/알고리즘_문제풀이(파이썬)

순열구하기(백트래킹)

Jedy_Kim 2021. 5. 17. 14:05
728x90

문제

서로 다른 n개의 원소들 중에서 r개만을 뽑아 일렬로 나열하는 것을 순열이라 한다. 예를 들어, 3개의 원소 a, b, c 중에서 2개만을 뽑아 나열하면 ab, ac, ba, bc, ca, cb 의 6가지 경우가 있다. n과 r이 주어질 때, n개의 소문자 중에서 r개만을 뽑아 나열하는 모든 경우를 출력하는 프로그램을 작성하시오. 단, a부터 시작하여 연속으로 n개의 알파벳을 갖고 있다고 하자.  

입력

첫 번째 줄에 n과 r이 주어진다. ( 1 ≤ n ≤ 10, 0 ≤ r ≤ min(n, 7) )  

출력

각 줄에 n개의 소문자 중에서 r개만을 뽑아 나열하는 경우를 사전순으로 나열한 결과를 출력한다.

 

예제 입력 

4 2

예제 출력 

ab

ac

ad

ba

bc

bd

ca

cb

cd

da

db

dc

 

#내코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
n, r = map(int, input().split()) 
result = [ 0 for _ in range(15)]
check = [ 0 for _ in range(15)]
 
def getResult(x):
  # x번째 for문을 돌려야함.
  if x >= r: 
    for i in result:
      if i != 0:
        print(i, end='')
    print()
  else:
    for i in range(n):
      if check[i] == False:
        alpha = ord('a'+ i
        result[x] = chr(alpha)
        check[i] = True
        getResult(x+1)
        check[i] = False 
        
getResult(0)
 
cs

 

#다른 사람코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
n, r = map(int, input().split()) 
visit = [ 0 for _ in range(15)] 
 
def comb(idx, word): 
  if idx >= r: 
    print(word)
    return
  
  for i in range(n):
    if visit[i] == False:
      visit[i] = True
      comb(idx+1, word+chr(ord('a')+i))
      visit[i] = False
 
comb(0'')
cs

 

반응형

'CS > 알고리즘_문제풀이(파이썬)' 카테고리의 다른 글

inequal(백트래킹)  (0) 2021.05.17
division(백트래킹)  (0) 2021.05.17
회전판과 로봇 (robot.cpp)  (0) 2021.05.16
baseball game  (0) 2021.05.16
이동 알고리즘  (0) 2021.05.15