[python] 프로그래머스 - 쿠키 구입

🤔문제 해결

Lv4

 

기준점을 하나 정한다. 0부터 쿠키의 길이-1 까지

그 기준은 첫째 아들 과자, 기준 + 1은 둘째 아들 과자로 시작한다.

1. 첫째의 과자가 적으므로 첫째에게 과자를 하나 더 준다.

2. 둘째의 과자가 더 적으므로 둘째에게 과자를 하나 더 준다.

3. 첫째의 과자가 적으므로 첫째에게 과자를 하나 더 준다.

4. 둘째의 과자가 더 적으므로 둘째에게 과자를 하나 더 준다.

5. 둘의 과자가 같으므로 값을 저장하고 첫째부터 과자를 더 줘본다. (둘째 먼저 줘도 된다.)

6. 둘째의 과자가 더 적으므로 둘째에게 과자를 하나 더 준다.

7. 첫째의 과자가 더 적어서 첫째에게 과자를 더 주고 싶지만 더 이상 줄 과자가 없다.

8. 다음 기준을 잡고 위의 과정을 반복한다.

다음 기준

💨 

 

💻소스 코드

def solution(cookie):
    answer = 0
    n = len(cookie)

    for i in range(n - 1):
        left_sum, left_idx = cookie[i], i
        right_sum, right_idx = cookie[i + 1], i + 1

        while True:
            if left_sum == right_sum:
                answer = max(answer, left_sum)
                # answer = max(answer, right_sum)
                
            if left_idx > 0 and left_sum <= right_sum:
                left_idx -= 1
                left_sum += cookie[left_idx]
            elif right_idx < n - 1 and right_sum <= left_sum:
                right_idx += 1
                right_sum += cookie[right_idx]
            else:
                break

    return answer

 

📕문제 확인

출처: 프로그래머스

링크: https://programmers.co.kr/learn/courses/30/lessons/49995

코딩테스트 연습 - 쿠키 구입

---

문제 설명

과자를 바구니 단위로 파는 가게가 있습니다. 이 가게는 1번부터 N번까지 차례로 번호가 붙은 바구니 N개가 일렬로 나열해 놨습니다.

철수는 두 아들에게 줄 과자를 사려합니다. 첫째 아들에게는 l번 바구니부터 m번 바구니까지, 둘째 아들에게는 m+1번 바구니부터 r번 바구니까지를 주려합니다. 단, 두 아들이 받을 과자 수는 같아야 합니다(1 <= l <= m, m+1 <= r <= N). 즉, A[i] 를 i번 바구니에 들어있는 과자 수라고 했을 때, A[l]+..+A[m] = A[m+1]+..+A[r] 를 만족해야 합니다.

각 바구니 안에 들은 과자 수가 차례로 들은 배열 cookie가 주어질 때, 조건에 맞게 과자를 살 경우 한 명의 아들에게 줄 수 있는 가장 많은 과자 수를 return 하는 solution 함수를 완성해주세요. (단, 조건에 맞게 과자를 구매할 수 없다면 0을 return 합니다)

제한사항

• cookie의 길이는 1 이상 2,000 이하입니다.
• cookie의 각각의 원소는 1 이상 500 이하인 자연수입니다.

---

입출력 예

cookieresult

cookie	result
[1,1,2,3]	3
[1,2,4,5]	0

입출력 예 설명

입출력 예 #1

첫째 아들에게 2, 3번 바구니를, 둘째 아들에게 4번 바구니를 사주면 두 아들은 각각 과자 3개를 받습니다.

입출력 예 #2

주어진 조건에 맞게 과자를 살 방법이 없습니다.