【Lintcode】576. Split Array

题目地址：

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

给定一个长$N$的数组$A$，题目保证$N\ge 5$。要求两个数$p$和$q$满足$0<p<q<N-1,q-p>1$，使得$A[p]+A[q]$最小，返回这个最小和。

求最小的三个数，如果最小的两个数相距超过$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(1)$。