사용자 도구

사이트 도구


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()

토론

댓글을 입력하세요:
D G L K R
 
ps/problems/boj/4373.txt · 마지막으로 수정됨: 2022/05/13 12:30 저자 teferi