一.題目內容: 對已排序數組進行 three-sum-faster
二.實現方式:
首先將數組進行排序,然後設置三個指針,位置爲數組頭部(lo),數組尾部(hi)和數組尾部-1(k),然後計算三個數相加的結果:
1.如果相加的結果大於0,則說明右邊的兩個數之和明顯大於左邊的,則將k–.
2.如果相加的結果小於0,則說明左邊的數負的太厲害,所以lo++.
3.如果相加結果爲0,則要找到有多少三個相同的pair然後計算有多少種排列組合.
4.完成一次k的循環以後,還要將lo=0,hi–,k=hi-1.
public static int threeSumFaster(int[] a){
int count = 0;
Arrays.sort(a);
int lo = 0;
int hi = a.length-1;
int k = hi-1;
while(a[hi]>0){
while(k>lo){
int result = a[lo]+a[hi]+a[k];
if(result>0){
k--;
}else if(result<0){
lo++;
}else{
count++;
int sl = lo;
int sk = k;
while(a[++lo] == a[sl] && lo<k){
count++;
}
while(a[--k] == a[sk] && k>sl){
count++;
}
}
}
hi--;
k=hi-1;
lo = 0;
}
return count;
}