====== Load Balancing (Silver) ====== ===== 풀이 ===== * [[ps:problems:boj:11990]]의 쉬운 버전. * N의 범위가 1000이하이므로, 모든 가능한 x좌표와 y좌표에 대해서 그점을 중심으로 가로펜스와 세로펜스를 그었을때의 M값을 계산해보고, 그중에서 최소 M값을 찾는 O(N^2) 알고리즘으로도 통과된다. * 물론 여기에서 M값을 계산하는 것은 O(1)에 처리할수 있어야 한다. O(N)만큼의 메모리만을 사용해서 계산하게 하는 것도 가능은 하지만, 같은 X좌표에 있는 소들과 같은 Y좌표에 있는 소들을 정확히 처리하면서 구현하려면 코드가 꽤 복잡해진다. * 차라리 메모리를 좀 많이 쓰고 전처리 시간이 걸리더라도, X좌표와 Y좌표들을 모두 좌표압축하고, 각 좌표의 소의 갯수를 1과 0으로 표현하는 2d 테이블을 만든 다음에, 2d 누적합을 구해서 계산하는게 심플하다. * 총 시간복잡도는 좌표압축에 O(NlogN), 2D 누적합 계산에 O(N^2), 모든 좌표에 대해서 M값을 계산하는것에 O(N^2)이 걸리므로 총 O(N^2). ===== 코드 ===== """Solution code for "BOJ 11997. Load Balancing (Silver)". - Problem link: https://www.acmicpc.net/problem/11997 - Solution link: http://www.teferi.net/ps/problems/boj/11997 """ import sys INF = float('inf') def create_2d_prefix_sum(nums): prefix_sum = [[0] * (len(nums[0]) + 1)] for row in nums: prefix_sum.append([ps := 0] + [(ps := ps + num) + p for num, p in zip(row, prefix_sum[-1][1:])]) return prefix_sum def main(): N = int(sys.stdin.readline()) x_coords, y_coords = [], [] for _ in range(N): x, y = [int(x) for x in sys.stdin.readline().split()] x_coords.append(x) y_coords.append(y) x_comp_map = {x: i for i, x in enumerate(sorted(set(x_coords)))} y_comp_map = {y: i for i, y in enumerate(sorted(set(y_coords)))} nums = [[0] * len(x_comp_map) for _ in y_comp_map] for x_i, y_i in zip(x_coords, y_coords): nums[y_comp_map[y_i]][x_comp_map[x_i]] = 1 prefix_sum = create_2d_prefix_sum(nums) bottom_row = prefix_sum[-1] total = bottom_row[-1] answer = INF for row in prefix_sum: ps_r = row[-1] for ps_rc, ps_c in zip(row, bottom_row): up_left = ps_rc up_right = ps_r - up_left down_left = ps_c - up_left down_right = total - up_right - down_left - up_left answer = min(answer, max(up_left, up_right, down_left, down_right)) print(answer) if __name__ == '__main__': main() {{tag>BOJ ps:problems:boj:골드_4}}