LeetCode 724. Find Pivot Index

🚀 문제: 724. Find Pivot Index

• 플랫폼: LeetCode

• 번호: 724

• 제목: Find Pivot Index

• 링크: https://leetcode.com/problems/find-pivot-index/description/

• 난이도: Easy

📝 문제 설명

주어진 정수 배열 nums에서 피벗 인덱스(pivot index)를 찾는 문제입니다.

피벗 인덱스란, 해당 인덱스를 기준으로 왼쪽에 있는 모든 요소들의 합과 오른쪽에 있는 모든 요소들의 합이 정확히 같은 지점의 인덱스를 말합니다.

만약 피벗 인덱스가 여러 개 있다면, 가장 왼쪽에 있는 인덱스를 반환해야 합니다. 피벗 인덱스를 찾을 수 없다면 -1을 반환합니다.

배열의 양쪽 끝에서는 다음과 같이 간주합니다:

• 가장 왼쪽 인덱스(0)의 왼쪽 합은 0입니다.

• 가장 오른쪽 인덱스(n-1)의 오른쪽 합은 0입니다.

예시 1:

• 입력: nums = [1, 7, 3, 6, 5, 6]

• 출력: 3

• 설명: 인덱스 3 (값 6)을 기준으로 왼쪽 합(1 + 7 + 3 = 11)과 오른쪽 합(5 + 6 = 11)이 같습니다.

예시 2:

• 입력: nums = [1, 2, 3]

• 출력: -1

• 설명: 피벗 인덱스가 존재하지 않습니다.

예시 3:

• 입력: nums = [2, 1, -1]

• 출력: 0

• 설명: 인덱스 0을 기준으로 왼쪽 합은 0, 오른쪽 합은 1 + (-1) = 0입니다.

🤔 풀이 아이디어: 왼쪽 합과 오른쪽 합 추적하며 비교

이 문제는 배열을 순회하면서 각 인덱스가 피벗 인덱스의 조건을 만족하는지 확인하는 방식으로 접근할 수 있습니다. 제가 사용한 핵심 아이디어는 현재 인덱스를 기준으로 왼쪽 요소들의 합(left_sum)과 오른쪽 요소들의 합(right_sum)을 계속해서 업데이트하고 비교하는 것입니다.

1. 초기값 설정:

   • left_sum: 인덱스 0의 왼쪽에는 요소가 없으므로 0으로 시작합니다.

   • right_sum: 인덱스 0의 오른쪽에 있는 모든 요소들의 합으로 시작합니다. (즉, nums[1]부터 끝까지의 합)

2. 첫 번째 인덱스(0) 확인: 계산된 초기 left_sum과 right_sum이 같은지 확인합니다. 만약 같다면 인덱스 0이 피벗이므로 바로 0을 반환합니다.

3. 인덱스 이동 및 합 업데이트:

   • while 루프를 사용하여 인덱스 i를 0부터 n-2까지 증가시킵니다. (여기서 n은 배열의 길이입니다.)

   • 루프의 각 단계에서는 다음 인덱스(i+1)가 피벗이 될 가능성을 ...