LintCode: 382 三角形计数(考点:前后指针)

题目:


分析:

其实这道题目和之前做到的三数之和相似,都是采用先固定一个数之后,采用两根指针枚举所有情况然后再通过调整指针达到目的。

此类题目首先想到的是数组先排序(数组元素位置不影响结果,且题目与元素大小有关),最简单的是暴力搜索,三层循环。

首先固定i从 2 遍历到 length(S)-1,内循环则为m和n枚举满足条件的情况。

这里有个小技巧,因为我们首先对数组进行排序(从小到大),如例1,当S[i] = 7 时, m初始化为0,n初始化为k-1,那么有S[m] + S[n] > S[i],此时不需要再检查4+6>7的情况,因为3已经满足条件了,那么3以后的元素肯定也满足。这样就会减少大量的运算时间。一次遍历即可。

public class Solution {
    /**
     * @param S: A list of integers
     * @return: An integer
     */
    public int triangleCount(int[] S) {
        // write your code here
        if(S.length<3)  return 0;
        Arrays.sort(S);
        int count=0;
        for(int i=1;i<S.length;i++){ //把最大边作为固定的
            int c=S[i];
            int m=0,n=i-1;
            while(m<n){
                int a=S[m];
                int b=S[n];
                if(a+b>c) {
                    count+=n-m;   //如果S[m]+S[n]>S[i],则可以固定n和i,m往右移,那么n和m之间的所有数(含n)都可以满足
                    n--; //i不变的情况下,S[m]+S[n]>S[i],可以尝试把n往左移动
                }else{ //i不变的情况下,S[m]+S[n]<=S[i],可以尝试把m往右移动
                    m++;
                }
            }
        }
        return count;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_27139155/article/details/80302510