RUBY

[BOJ] N과 M (9) 15663 R 본문

PS/BOJ

[BOJ] N과 M (9) 15663 R

RUBY_루비 2020. 6. 30. 17:20

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

 

1. 문제 이해 및 해결과정

 - 중복되는 경우의 수를 제외한 수열

 

2. 풀이방법

 1. permutaion 라이브러리 이용 , set()이용

#N과 M (9)
#https://www.acmicpc.net/problem/15663
import sys
import itertools as it
sys.stdin = open("input.txt","r")

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

for k in sorted(set(it.permutations(a,m))):
    #print(k)
    print(' '.join(map(str,k)))

  

  2. DFS, counter이용

#N과 M (9) 2
#https://www.acmicpc.net/problem/15663
import sys
from collections import Counter
def DFS(L):
    if L==m:
        for x in res:
            print(x,end=' ')
        print()
        return
    else:
        for i in range(len(cnt)):
            if cnt[i]:
                res[L]=num[i]
                cnt[i]-=1
                DFS(L+1)
                cnt[i]+=1

if __name__=="__main__":
    n, m = map(int, input().split())
    a = sorted(list(map(int, input().split())))
    num, cnt = map(list, zip(*list(Counter(a).items())))
    res=[0]*m
    DFS(0)

 

  

3. 오답원인

 

4. 알게된 점

 

join 
print(' '.join(map(str,k)))

: k를 문자열로 연결해서 출력하라

- 만약 print()였다면

 

zip(*iterable)

: 동일한 개수로 이루어진 자료형을 묶어 주는 역할을 하는 함수

 

counter()
from collections import Counter

Counter('hello') # Counter({'l': 3, 'o': 2, 'h': 1, 'e': 1})

 

num, cnt = map(list, zip(*list(Counter(a).items())))

: 이 코드가 숫자 별 개수 를 각각 세서 만들어주는 코드이다.

num, cnt 이용하기

결과는 이와 같다. 

c.items()                       # convert to a list of (elem, cnt) pairs
Counter(dict(list_of_pairs))    # convert from a list of (elem, cnt) pairs
print(Counter(a).items()) ##dict_items([(1, 2), (2, 2)])
print(list(Counter(a).items())) ##[(1, 2), (2, 2)]
print(zip(*list(Counter(a).items()))) ##<zip object at 0x020DB6C8>

 

 

 

 

 

 

 

'PS > BOJ' 카테고리의 다른 글

[BOJ] N과 M (11) 15665  (0) 2020.07.01
[BOJ] N과 M (10) 15664 R  (0) 2020.06.30
[BOJ] N과 M (8) 15657  (0) 2020.06.30
[BOJ] N과 M (7) 15656  (0) 2020.06.30
[BOJ] N과 M (6) 15655  (0) 2020.06.30
Comments