【Lintcode】576. Split Array

記錄演算法發表於2020-12-26

題目地址:

https://www.lintcode.com/problem/split-array/description

給定一個長 N N N的陣列 A A A,題目保證 N ≥ 5 N\ge 5 N5。要求兩個數 p p p q q q滿足 0 < p < q < N − 1 , q − p > 1 0<p<q<N-1, q-p>1 0<p<q<N1,qp>1,使得 A [ p ] + A [ q ] A[p]+A[q] A[p]+A[q]最小,返回這個最小和。

求最小的三個數,如果最小的兩個數相距超過 1 1 1則返回它們的和;否則如果最小和第三小的兩個數相距超過 1 1 1則返回它們的和;否則返回第二小和第三小的數之和。程式碼如下:

public class Solution {
    /**
     * @param arr: an inter array
     * @return: return the min sum
     */
    public int splitArray(int[] arr) {
        // write your code here
        int min1, min2, min3, pos1, pos2, pos3;
        min1 = min2 = min3 = Integer.MAX_VALUE;
        pos1 = pos2 = pos3 = -1;
        for (int i = 1; i < arr.length - 1; i++) {
            if (arr[i] < min1) {
                min3 = min2;
                min2 = min1;
                min1 = arr[i];
                
                pos3 = pos2;
                pos2 = pos1;
                pos1 = i;
            } else if (arr[i] < min2) {
                min3 = min2;
                min2 = arr[i];
                
                pos3 = pos2;
                pos2 = i;
            } else if (arr[i] < min3) {
                min3 = arr[i];
                
                pos3 = i;
            }
        }
        
        if (Math.abs(pos1 - pos2) > 1) {
            return min1 + min2;
        } else if (Math.abs(pos1 - pos3) > 1) {
            return min1 + min3;
        } else {
            return min2 + min3;
        }
    }
}

時間複雜度 O ( N ) O(N) O(N),空間 O ( 1 ) O(1) O(1)

相關文章