ps:problems:boj:4373
수집합
ps | |
---|---|
링크 | acmicpc.net/… |
출처 | BOJ |
문제 번호 | 4373 |
문제명 | 수집합 |
레벨 | 골드 1 |
시간복잡도 | O(n^2) |
인풋사이즈 | n<=1000 |
사용한 언어 | Python |
제출기록 | 153824KB / 1176ms |
최고기록 | 1176ms |
해결날짜 | 2022/05/13 |
풀이
- 네 수의 합이 0이 되는 조합을 찾는 문제의 변형이다.
- 이런 문제들은, 모든 네 수의 조합에 대해서 합을 구해보는 O(n^4)방식 대신에, 모든 페어들에 대해서 합을 구해서 저장하고, 그것들을 이용해서 네 수의 합이 특정 값이 되는지 찾는 방식으로 시간 복잡도를 O(n^2)으로 줄일수 있다.
- 이 문제에서는 a+b+c=d의 형태이고, 이것을 a+b = d-c로 바꿔쓰면, 모든 페어의 합과 모든 페어의 차를 구해서 중복되는 것이 있는지 찾는 문제로 바뀐다.
- 물론 단순하게 중복되는 값을 찾기만 하면 되는것은 아니고, 합을 만들때 사용된 두 수와, 차를 만들때 사용된 두 수간에 중복이 없는지를 체크해야 한다. 그것때문에 구현이 좀더 복잡해지긴 한다. 여기서는 합을 키로 갖고, 합을 만들때 사용된 페어들의 리스트를 밸류로 갖는 딕셔너리를 이용해서 구현했다.
- 총 시간복잡도는 O(n^2)
코드
"""Solution code for "BOJ 4373. 수집합".
- Problem link: https://www.acmicpc.net/problem/4373
- Solution link: http://www.teferi.net/ps/problems/boj/4373
"""
import collections
import itertools
import sys
INF = float('inf')
def main():
while True:
n = int(sys.stdin.readline())
if n == 0:
break
S = [int(sys.stdin.readline()) for _ in range(n)]
answer = -INF
pairs_by_sum = collections.defaultdict(list)
for a, b in itertools.combinations(S, 2):
pairs_by_sum[a + b].append({a, b})
for d in S:
if d <= answer:
continue
for c in S:
if c == d:
continue
cd = {c, d}
ab_list = pairs_by_sum[d - c]
if any(not (ab & cd) for ab in ab_list):
answer = d
break
print('no solution' if answer == -INF else answer)
if __name__ == '__main__':
main()
ps/problems/boj/4373.txt · 마지막으로 수정됨: 2022/05/13 12:30 저자 teferi
토론