코딩테스트 - 연결된 수의 합/ 파이썬 Python

문제>

문제의 규칙>

num=3, total=12, result=[3, 4, 5]를 보면 평균(total//num=4)이 가운데 에 위치해 있고, 

양 옆의 숫자는 -1, +1 한 숫자입니다.

 

num=5, total=15, result=[1, 2, 3, 4, 5]를 보면 평균(total//num=3)이 가운데 에 위치해 있고, 

양 옆의 숫자는 -2, -1, 0, +1, +2 한 숫자입니다.

 

num=4, total=14, result=[2, 3, 4, 5]를 보면 평균(total//num=3 )이 가운데 왼쪽에 위치해 있고,

양 옆의 숫자는 -1, +1, +2 한 숫자입니다.

 

num=5, total=5, result=[-1, 0, 1, 2, 3]를 보면 평균(total//num=1)이 가운데 위치해 있고,

양 옆의 숫자는 -2, -1, 0, +1, +2 한 숫자입니다.

 

-> 위의 규칙에서 발견할 수 있는 것은 num 이 홀수이면, 가운데에 위치하고 

num 이 짝수이면 가운데 왼쪽에 위치한다는 것을 발견할 수 있습니다.

 

 

문제해결 1>

def solution(num, total):
    middle_idx = num // 2
    
    average = total // num
    
    if num%2==1:
        start_num = middle_idx * - 1
    else:
        start_num = middle_idx * -1 +1
    end_num = middle_idx + 1
    
    return [num + average for num in range(start_num, end_num)]

위의 규칙을 이용하여 코딩한 방법입니다.

num 이 홀수이면, 가운데에 위치하고 

num 이 짝수이면 가운데 왼쪽에 위치한다는 것을 이용해서 if-else문을 작성해주었습니다.

 

 

문제해결 2>

def solution(num, total):
    average = total // num
    return [i for i in range(average - (num-1)//2, average + (num + 2)//2)]

문제해결1과 해결 방법은 비슷하지만 인덱스를 없애고 num - 1로 하고 우측은 num + 2로 짝수, 홀수를

구분해 주었습니다.

 

좌, 우측 숫자의 개수는 num이 짝수이냐 홀수이냐에 따라 다르기 때문에 왼쪽 끝 값을 구하기 

위해서는 num - 1로 하고 우측은 num + 2를 하면 짝수, 홀수를 구분하지 않고 연속적인 리스트를 구할 수 있습니다.

 

 

 

❗ 다른 문제들과 비교적 풀기어려웠습니다. 다른 lv.0에 있는 문제와 비교했을 때 정답률도 낮은 편입니다.

다른 예시를 넣었을 때에도 이 식이 성립하려나? 를 자꾸 고민하다보니 

생각이 많아진 것 같습니다. 문제에 나온 예시들로만 규칙을 찾고 저 예시에 답이 부합하도록

코드를 짜면 되는 것 같습니다.