1. 程式人生 > 其它 >【Lintcode】576. Split Array

【Lintcode】576. Split Array

技術標籤:# 陣列、連結串列與模擬演算法javaleetcode

題目地址:

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)