Coding Test/Combination
[1947] 선물 전달하기 (완전 순열, 교란 순열)
1947번: 선물 전달 경우의 수를 1,000,000,000으로 나눈 나머지를 첫째 줄에 출력한다. www.acmicpc.net 분석 완전 순열 : n개의 원소가 모두 자기 자신이 아닌 값으로 배정되는 순열 (모든 원소의 위치를 바꾸는 순열) A가 B에게 선물을 줬다고 가정 1. B도 A에게 선물을 줬을 때 (양방향 교환) : N명 중 2명이 교환을 완료했으므로 남은 경우의 수는 D[N-2] 2. B는 A가 아닌 다른 사람에게 선물을 선달할 때 (단방향 교환) : N명 중 B만 받은 선물이 정해진 상태이므로 남은 학생은 N-1이며 경우의 수는 D[N-1] 풀이 n=int(input()) mod=1000000000 d=[0]*10000001 d[1]=0 # 혼자서는 선물을 교환할 수 없음 d[2]=1 # ..
[13251] 조약돌 꺼내기
13251번: 조약돌 꺼내기 첫째 줄에 뽑은 조약돌이 모두 같은 색일 확률을 출력한다. 정답과의 절대/상대 오차는 10-9까지 허용한다. www.acmicpc.net 풀이 1. 색깔별 조약돌의 개수에서 k를 뽑을 수 잇는 경우의수 / 전체 돌에 관해 k개를 뽑는 경우의 수 import sys input=sys.stdin.readline import math m = int(input()) # 색 종류개수 arr = list(map(int,input().split())) # 색깔별 조약돌 개수 k = int(sys.stdin.readline()) # 뽑을 개수 t = sum(arr) # 전체 조약돌 개수 total = math.comb(t, k) # 전체 돌에 관해 k개를 뽑는 경우의 수 same_color..
[1010] 다리 놓기
1010번: 다리 놓기 입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트케이스에 대해 강의 서쪽과 동쪽에 있는 사이트의 개수 정수 N, M (0 < N ≤ M < 30)이 주어진다. www.acmicpc.net 분석 서쪽N ≤ 동쪽 M 다리끼리 서로 겹쳐질 수 없다 ➔ 동쪽에서 N개를 선택한 후 가장 위쪽 사이트에서부터 차례대로 연결 M개의 사이트에서 N개를 선택하는 문제로 변경 풀이 """ 강의 서쪽에는 N개의 사이트가 있고 동쪽에는 M개의 사이트가 있다는 것을 알았다. (N ≤ M) 원이는 다리를 최대한 많이 지으려고 하기 때문에 서쪽의 사이트 개수만큼 (N개) 다리를 지으려고 한다. 다리끼리는 서로 겹쳐질 수 없다고 할 때 다리를 지을 수 있는 경우의 수 입력의 ..
수들의 조합(DFS)
import sys input=sys.stdin.readline n,k=map(int,input().split()) arr=list(map(int, input().split())) m=int(input()) def dfs(x,s,sum): global cnt if x==k: if sum%m == 0: cnt += 1 else: for i in range(s,n): dfs(x+1,i+1,sum+arr[i]) cnt=0 dfs(0,0,0) print(cnt)
중복순열 구하기(itertools 라이브러리)
중복순열 구하기(DFS / itertools 라이브러리) 분석 풀이 import sys input=sys.stdin.readline n,m=map(int, input().split()) res=[0]*n total=0 def dfs(x): global total if x==m: for j in range(m): print(res[j], end=' ') total+=1 print() else: for i in range(1,n+1): res[x]=i dfs(x+1) dfs(0) print(total karla.tistory.com """ 1부터 N까지 번호가 적힌 구슬 이 중 중복을 허락하여 M번을 뽑아 일렬로 나열 하는 방법을 모두 출력 N(3
수들의 조합(itertools 라이브러리)
수들의 조합(DFS) import sys input=sys.stdin.readline n,k=map(int,input().split()) arr=list(map(int, input().split())) m=int(input()) def dfs(x,s,sum): global cnt if x==k: if sum%m == 0: cnt += 1 else: for i in range(s,n): dfs(x+1,i+1,sum+arr[i]) cnt=0 dfs(0,0,0) print(cnt) karla.tistory.com """ N개의 정수 K개를 뽑는 조합의 합이 임의의 정수 M의 배수인 개수 는 몇 개가 있는지 출력 5개의 숫자 2 4 5 8 12 3개를 뽑은 조합의 합이 6의 배수인 조합을 찾으면 4+8+12 2..
순열 추측하기(itertools 라이브러리)
순열 추측하기(DFS, 파스칼 삼각형, 이항계수)분석 이항계수 리스트 선언 풀이 """ 가장 윗줄에 1부터 N까지의 숫자, 둘째 줄부터 위의 두개를 더한 값이 저장 N: 4, 가장윗줄: 3 1 2 4 3 1 2 4 4 3 6 7 9 16 N(1~10)과 가장 밑에 있는 숫자(F)가 주어져 있karla.tistory.com import itertools as itimport sysinput=sys.stdin.readlinen,f=map(int,input().split())visited=[False]*(n+1)p=[0]*n# 이항계수b=[1]*nfor i in range(1,n): b[i]=b[i-1]*(n-i)//i# 순열 구하기a=list(range(1,n+1))for tmp in it.permu..
조합 구하기(DFS)
분석 매개변수 s : 가지를 받는 첫번째 수(for문 시작 숫자) 풀이 """ 1~N 번호 구슬, M개 뽑는 방법의 수 출력 N(3