[LeetCode] Partition Array Into Three Parts With Equal Sum

아르당·2026년 3월 24일

LeetCode

목록 보기
219/265
post-thumbnail

문제를 이해하고 있다면 바로 풀이를 보면 됨
전체 코드로 바로 넘어가도 됨
마음대로 번역해서 오역이 있을 수 있음

Problem

정수 배열 arr이 주어졌을 때, 배열의 합이 비어 있지 않은 세 부분으로 나눌 수 있다면 true를 반환해라.
형식적으로, (arr[0] + arr[1] + ... + arr[i] == arr[i + 1] + arr[i + 2] + ... + arr[j - 1] == arr[j] + arr[j + 1] + ... + arr[arr.length - 1])를 만족하는 인덱스 i + 1 < j를 찾을 수 있다면 배열을 분할할 수 있다.

Example

#1
Input: arr = [0, 2, 1, -6, 6, -7, 9, 1, 2, 0, 1]
Output: true
Explanation: 0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1

#2
Input: arr = [0, 2, 1, -6, 6, 7, 9, -1, 2, 0, 1]
Output: false

#3
Input: arr = [3, 3, 6, 5, -2, 2, 5, 1, -9, 4]
Output: true
Explanation: 3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4

Constraints

  • 3 <= arr.length <= 5 * 10^4
  • -10^4 <= arr[i] <= 10^4

Solved

class Solution {
    public boolean canThreePartsEqualSum(int[] arr) {
        int sum = 0;

        for(int a : arr){
            sum += a;
        }

        if(sum % 3 != 0){
            return false;
        }

        int target = sum / 3;
        sum = 0;
        int count = 0;

        for(int i = 0; i < arr.length - 1; i++){
            sum += arr[i];

            if(sum == target){
                sum = 0;
                count++;

                if(count == 2){
                    return true;
                }
            }
        }

        return false;
    }
}
profile
내 마음대로 코드 작성하는 세상

0개의 댓글