2. MinAvgTwoSlice (feat.PYTHON)

A non-empty array A consisting of N integers is given. A pair of integers (P, Q), such that 0 ≤ P < Q < N, is called a slice of array A (notice that the slice contains at least two elements). The average of a slice (P, Q) is the sum of A[P] + A[P + 1] + ... + A[Q] divided by the length of the slice. To be precise, the average equals (A[P] + A[P + 1] + ... + A[Q]) / (Q − P + 1).

For example, array A such that:

A[0] = 4 A[1] = 2 A[2] = 2 A[3] = 5 A[4] = 1 A[5] = 5 A[6] = 8

contains the following example slices:

• slice (1, 2), whose average is (2 + 2) / 2 = 2;
• slice (3, 4), whose average is (5 + 1) / 2 = 3;
• slice (1, 4), whose average is (2 + 2 + 5 + 1) / 4 = 2.5.

The goal is to find the starting position of a slice whose average is minimal.

Write a function:

def solution(A)

that, given a non-empty array A consisting of N integers, returns the starting position of the slice with the minimal average. If there is more than one slice with a minimal average, you should return the smallest starting position of such a slice.

For example, given array A such that:

A[0] = 4 A[1] = 2 A[2] = 2 A[3] = 5 A[4] = 1 A[5] = 5 A[6] = 8

the function should return 1, as explained above.

Write an efficient algorithm for the following assumptions:

• N is an integer within the range [2..100,000];
• each element of array A is an integer within the range [−10,000..10,000].

 

풀이

비어 있지 않은 N개의 정수를 가진 배열 A를 0 <= P < Q < N의 범위에서 P : Q까지 슬라이싱해

평균값을 구합니다.

 

그러므로 0, 1번 인덱스로 초기평균값을 구하고 2번 인덱스부터 2개씩 쌍으로 평균값을 비교해 나갑니다.

단, 슬라이싱한 원소의 개수가 3개인 배열이 있을 수 있는데 이때 2번 인덱스 부터 루프를 돌개 되면 마지막

원소를 검사하지 않게 되므로 원소가 3개일 때를 고려해 예외처리를 합니다.

 

마지막으로 해당하는 최소평균값을 가지는 구간의 시작 index값을 반환합니다.

 

def solution(A):
    # write your code in Python 3.6
    minAvg = (A[0] + A[1]) / 2           # 초기 첫번째 평균값 3
    minIdx = 0
        
    for i in range(2, len(A)):
        avg = (A[i-2] + A[i-1] + A[i])/3 # 원소가 3개일 때    8/3
        if minAvg > avg:
            minAvg = avg
            minIdx = i - 2

        avg = (A[i-1] + A[i])/2          # 1-2 구간 부터 원소 2개씩 비교
        if minAvg > avg:
            minAvg = avg
            minIdx = i - 1

    return minIdx