C++ 合法的三角形个数
题目:给一个长度为N的非负整数数组nums,请你计算一下,有多少个三元组代表的边长可以组成三角形
数据范围 : 1≤N≤1000,0≤nums[i]≤1000
该题的解法如下:
首先,对各边长进行排序,方便获取三边中最长的边。
然后,进入循环,循环体如下:
由小到大找最长边(循环)
从最长边开始由大到小找次长边(循环)
从次长边开始由大到小找最短边。(循环计数法,或者用二分查找法找上界,然后用计算中间个数)
int validTriangleNumber(vector<int>& nums) { int length = nums.size(); if (length < 3) return 0; sort(nums.begin(), nums.end()); int res = 0; for (int right = 2; right < length; ++right) { for (int mid = right - 1; mid > 0; --mid) { int temp = nums[right] - nums[mid]; if (temp > nums[mid]) break; auto bg = nums.begin(); auto upperBound = upper_bound(bg, bg + mid, temp); res += mid - (upperBound - bg); } } return res; }
//二分查找上界法 int validTriangleNumber(vector<int>& nums) { int length = nums.size(); if (length < 3) return 0; sort(nums.begin(), nums.end());//排序 int res = 0; for (int right = 2; right < length; ++right)//right最长边序号 { for (int mid = right - 1; mid > 0; --mid)//mid次长边序号 { int temp = nums[right] - nums[mid];//找出最短边的最短长度temp if (temp > nums[mid]) break;//如果最短边长度的最短长度已经比次长长,打断循环 auto bg = nums.begin(); auto upperBound = upper_bound(bg, bg + mid, temp);//利用二分法找出最短边长度上界 res += mid - (upperBound - bg);//计算中间边的个数 } } return res; }