【Lintcode】576. Split Array
阿新 • • 發佈:2020-12-27
技術標籤:# 陣列、連結串列與模擬演算法javaleetcode
題目地址:
https://www.lintcode.com/problem/split-array/description
給定一個長 N N N的陣列 A A A,題目保證 N ≥ 5 N\ge 5 N≥5。要求兩個數 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<N−1,q−p>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)。