ps:problems:boj:1033
칵테일
ps | |
---|---|
링크 | acmicpc.net/… |
출처 | BOJ |
문제 번호 | 1033 |
문제명 | 칵테일 |
레벨 | 골드 2 |
분류 |
수학 |
시간복잡도 | O(nlogm) |
인풋사이즈 | n<=10, m<=9 |
사용한 언어 | Python 3.11 |
제출기록 | 39512KB / 120ms |
최고기록 | 33ms |
해결날짜 | 2024/10/25 |
풀이
- 사실 n이 워낙 작아서 어떤 방식으로 풀든 시간안에 풀리기는 한다
- 칵테일의 재료를 노드로, 비율이 주어진 재료쌍을 엣지로 하는 그래프를 생각하자.
- 문제가 약간 애매하게 쓰여있는데, “이 비율을 이용해서 칵테일에 들어가는 전체 재료의 비율을 알아낼 수 있다” 라는 말이 결국 모든 노드들이 연결되어 있다는 것을 의미한다. 이제 엣지의 갯수가 N-1개이므로, 그래프가 트리라는 것도 알 수 있다.
- 트리라는 것을 알면 이후는 간단하다. 노드 한개를 루트로 놓고, 질량을 X라고 고정하면, 다른 노드들의 질량은 부모노드의 질량에 비율을 곱해서 모두 구할 수 있다.
- 모든 노드의 질량을 X*p[u]/q[u] 형태로 표현하고 나면 (p[u]/q[u]는 기약분수), X의 최솟값은 모든 q[u]의 최소공배수가 된다. 이 값을 구해주면 끝.
- 트리를 탐색하며 각 노드의 비율을 계산하는데에 O(n), lcm을 구하는데에 O(n*logm)이 걸리므로 총 시간복잡도는 O(nlogm)이다
- 분자와 분모를 따로 계산해서 저장하는 방법과, 그냥 fractions 라이브러리를 쓰는 방법 두가지로 구현했는데, fractions를 쓸 때가 코드는 간단했지만 시간은 훨씬 오래 걸렸다.
코드
"""Solution code for "BOJ 1033. 칵테일".
- Problem link: https://www.acmicpc.net/problem/1033
- Solution link: http://www.teferi.net/ps/problems/boj/1033
Tags: [math]
"""
import math
import fractions
def main():
N = int(input())
wtree = [{} for _ in range(N)]
for _ in range(N - 1):
a, b, p, q = [int(x) for x in input().split()]
wtree[a][b] = fractions.Fraction(q, p)
wtree[b][a] = fractions.Fraction(p, q)
ratio = [fractions.Fraction(1) for _ in range(N)]
root = 0
stack = [root]
visited = [False] * N
visited[root] = True
while stack:
u = stack.pop()
for v, w in wtree[u].items():
if visited[v]:
continue
visited[v] = True
ratio[v] = ratio[u] * w
stack.append(v)
lcm = math.lcm(*(ratio_u.denominator for ratio_u in ratio))
answer = [ratio_u * lcm for ratio_u in ratio]
print(*answer)
if __name__ == '__main__':
main()
ps/problems/boj/1033.txt · 마지막으로 수정됨: 2024/10/25 06:32 저자 teferi
토론