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())))
: 이 코드가 숫자 별 개수 를 각각 세서 만들어주는 코드이다.
결과는 이와 같다.
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>