[Softeer] 슈퍼컴퓨터 클러스터

문제

요약

컴퓨터 성능을 향상 시킬 때, 비용은 (향상하려는 목표 - 컴퓨터 성능)의 제곱 만큼이 사용된다.

b 이내에서 최소 값을 최대로 끌어 올릴 수 있는 최소값은?

 

원본

https://softeer.ai/practice/6252

Softeer - 현대자동차그룹 SW인재확보플랫폼

풀이

문제를 자세히 제대로 읽자!

처음에 문제를 잘못 이해해서 한참을 엉뚱한 풀이로 풀고 있었다.

예를 들어

5 3 9 8 4 3 1 8 6

이라고 입력값이 들어올때, 사용되는 비용을 1을 5까지 올린다고 하면,

비용 계산을 1->3, 3->5 이렇게 계산해서 (3-1)**2 + 3*(5-3)**2 해서 4+12 = 12 인줄 알았다.

이렇게 계산하는게 아니라 (5-1)**2+3*(5-3)**2 해서 16+12=28이다,,, 에휴휴

 

잘못 이해해서 내멋대로 써본 코드 (정답아님)

import sys
input = sys.stdin.readline
answer = 0

n, b = map(float, input().split())
n = int(n)
lst = list(map(int, input().split()))
print('ori', lst)
d = {}
for i in range(n):
    if d.get(lst[i]):
        d[lst[i]].append(i)
    else:
        d[lst[i]] = [i]

tmp = sorted(list(d.keys()))

for i in range(len(tmp)-1):
    lw, up = tmp[i], tmp[i+1]
    x = 0

    while lw <= up:
        mid = (lw + up)//2
        if (mid - tmp[i]) ** 2 * len(d[tmp[i]]) <= b:
            x = mid
            lw += 1
        else:
            up -= 1
            
    b -= (mid - tmp[i]) ** 2 * len(d[tmp[i]])
    if x < tmp[i+1]:
        for j in d[tmp[i]]:
            lst[j] = x
        break
    
    for j in d[tmp[i]]:
        d[tmp[i+1]].append(j)
        lst[j] = tmp[i+1]

last = tmp[-1]
if len(d[last]) == n:
    lw = last + 1
    up = int(b-1)

    while lw <= up:
        mid = (lw+up) // 2
        if n * (mid - last)**2 <= b:
            x = mid
            lw += 1
        else:
            up -= 1
    print(x)
else:
    print(min(lst))

 

 

제대로 이해하고 푼 코드

최근에 이진탐색을 공부해서 같은 유형으로 풀어보고 싶었는데 딱 그문제 였다.

와중에 upper의 값을 b+1로 놓고 풀어서 또 한참 헤맸다. 휴우우

그래도 이진 탐색을 어떻게 써야 할지 알게 된것 같 아 보람차다.

import sys
input = sys.stdin.readline
answer = 0

n, b = map(int, input().split())
n = int(n)
lst = list(map(int, input().split()))

lw, up = min(lst)+1, 1000000000 //여기서 b+1로 두고 풀었다가 한참 또 헤맸다.
while lw <= up:
    num = b
    mid = (lw+up)//2
    for i in range(n):
        if lst[i] < mid:
            num -= ((mid - lst[i]) ** 2)
        if num < 0:
            up = mid - 1
            break
    else:
        answer = int(mid)
        lw = mid + 1

print(answer)