三角形计数

三角形计数

题目：

描述
给定一个整数数组，在该数组中，寻找三个数，分别代表三角形三条边的长度，问，可以寻找到多少组这样的三个数来组成三角形？

样例
样例 1:
输入: [3, 4, 6, 7]
输出: 3
解释:
可以组成的是 (3, 4, 6), 
           (3, 6, 7),
           (4, 6, 7)

样例 2:
输入: [4, 4, 4, 4]
输出: 4
解释:
任何三个数都可以构成三角形
所以答案为 C(3, 4) = 4

解题思路：先将数组排序，然后遍历数组，假设当前遍历的是三角形中的最长边，再用双指针来寻找和大于该边的组合即可

public class Solution {
    /**
     * @param S: A list of integers
     * @return: An integer
     */
    public int triangleCount(int[] nums) {
        // write your code here
        int len = nums.length;
        if(len < 3) {
            return 0;
        }

        Arrays.sort(nums);
        int ans = 0;

        for(int i = 2; i < len; i++) {
            int l = 0, r = i - 1;
            while(l < r) {
                int sum = nums[l] + nums[r];
                if(sum > nums[i]) {
                	// nums[;] + nums[r] > nums[i]
                	// 说明nums[i + 1.. r - 1] + nums[r] > nums[i]
                	// 所以 ans += r - l 减少重复计算
                    ans += r - l;
                    r--;
                } else {
                    l++;
                }
            }
        }

        return ans;
    }
}