Java에서 배열의 중심 인덱스를 찾는 방법-Java시작하기-php.cn

Java에서 배열의 중심 인덱스를 찾는 방법

王林

풀어 주다： 2020-04-20 16:09:59

앞으로

2443명이 탐색했습니다.

Java에서 배열의 중심 인덱스를 찾는 방법

목적:

정수 유형의 배열 수가 주어지면 배열의 "중심 인덱스"를 반환하는 메서드를 작성합니다.

배열 중심 인덱스 정의:

배열 중심 인덱스의 왼쪽에 있는 모든 요소의 합은 배열 중심 인덱스의 오른쪽에 있는 모든 요소의 합과 같습니다. 배열에 중앙 인덱스가 없으면 -1을 반환해야 합니다. 배열에 여러 개의 중앙 인덱스가 있는 경우 왼쪽에 가장 가까운 인덱스를 반환해야 합니다.

관련 비디오 튜토리얼 권장사항: java video

예 1:

输入: 
nums = [1, 7, 3, 6, 5, 6]
输出: 3
解释: 
索引3 (nums[3] = 6) 的左侧数之和(1 + 7 + 3 = 11)，与右侧数之和(5 + 6 = 11)相等。
同时, 3 也是第一个符合要求的中心索引。

로그인 후 복사

예 2:

输入: 
nums = [1, 2, 3]
输出: -1
解释: 
数组中不存在满足此条件的中心索引。

로그인 후 복사

설명:

nums의 길이 범위는 [0, 10000]입니다.

모든 nums[i]는 [-1000, 1000] 범위의 정수입니다.

해결책 아이디어:

sum, leftSum 및 rightSum을 사용하여 각각 i의 왼쪽에 있는 합계, i의 오른쪽에 있는 합계를 저장합니다. 먼저 다음의 편의를 위해 총합을 계산합니다. 그런 다음 i를 왼쪽에서 오른쪽으로 이동합니다. 참고 i가 첫 번째 요소인 경우 별도로 고려해야 합니다. i의 왼쪽과 오른쪽 값을 각각 계산하여 비교합니다. leftSum=rightSum이면 i를 반환합니다.

java 코드:

class Solution {
    public int pivotIndex(int[] nums) {
        if (nums == null || nums.length == 0) {
            return -1;
        }
 
        int sum = 0;
        int leftSum = 0;
        int rightSum = 0;
 
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
        }
 
        for (int i = 0; i < nums.length; i++) {
            if (i == 0) {
                leftSum = 0;
            } else {
                leftSum += nums[i - 1];
            }
            rightSum = sum - leftSum - nums[i];
 
            if (leftSum == rightSum) {
                return i;
            }
        }
 
        return -1;
    }
}

로그인 후 복사