ps:problems:boj:19535
ㄷㄷㄷㅈ
| ps | |
|---|---|
| 링크 | acmicpc.net/… |
| 출처 | BOJ |
| 문제 번호 | 19535 |
| 문제명 | ㄷㄷㄷㅈ |
| 레벨 | 골드 3 |
| 분류 |
조합론 |
| 시간복잡도 | O(n) |
| 인풋사이즈 | n<=300,000 |
| 사용한 언어 | Python 3.13 |
| 제출기록 | 87552KB / 400ms |
| 최고기록 | 320ms |
| 해결날짜 | 2025/12/02 |
풀이
- 얼핏 보면 모든 노드 4개짜리 서브그래프에 대해서 ㄷ 인지 ㅈ인지를 구분해야 할 것처럼 보이지만, 이것은 매우 비효율적이다.
- 그냥 ㄷ과 ㅈ을 각각 세어주는 편이 빠르다. 조합론적으로 O(n)에 처리할 수 있다.
- ㄷ의 경우는, 가운데 엣지 (u,v) 를 고정시키면, (deg(u)-1)*(deg(v)-1) 로 개수를 셀 수 있고, ㅈ의 경우는 가운데 노드 u를 고정시키면 comb(deg(u),3)으로 셀 수 있다.
- 트리 DP를 이용해도 O(n)에 구할수 있고 기여란에는 트리DP로 푼 사람도 많지만, 조합론적 방법이 구현이 훨씬 간단하다. 에디토리얼에서 소개하는 정해도 조합론적 풀이.
코드
"""Solution code for "BOJ 19535. ㄷㄷㄷㅈ".
- Problem link: https://www.acmicpc.net/problem/19535
- Solution link: http://www.teferi.net/ps/problems/boj/19535
Tags: [combinatorics]
"""
import sys
import math
def main():
N = int(sys.stdin.readline())
edges = [
[int(x) - 1 for x in sys.stdin.readline().split()] for _ in range(N - 1)
]
degs = [0] * N
for u, v in edges:
degs[u] += 1
degs[v] += 1
d_count = sum((degs[u] - 1) * (degs[v] - 1) for u, v in edges)
g_count = sum(math.comb(deg_u, 3) for deg_u in degs)
if d_count > g_count * 3:
print('D')
elif d_count < g_count * 3:
print('G')
else:
print('DUDUDUNGA')
if __name__ == '__main__':
main()
ps/problems/boj/19535.txt · 마지막으로 수정됨: 2025/12/02 13:43 저자 teferi

토론