[프로그래머스 lv 0] 연속된 수의 합 (코딩테스트 입문) 2️⃣

아이디어 : SUM 값을 NUM 값으로 나눈 평균 값이 가운데에 있을 것이라는 걸 생각하면 될 것 같다.

 

def solution(num, total):
    answer = []
    avg = total/num
    if total%num==0:
        answer.append(avg)
        for i in range(1,num//2+1):
            answer.append(avg-i)
            answer.append(avg+i)
    else:
        for i in range(num//2):
            answer.append(total//num-i)
            answer.append(total//num+1+i)
    answer.sort()
    return answer

 

+250818

def solution(num, total):
    answer = []
    flag = 0
    if num%2==0:
        flag = total // num
        for i in range(num):
            answer.append(flag-num//2+i+1)
    else:
        flag = total // num
        for i in range(num):
            answer.append(flag-num//2+i)
    return answer