和3Sum那题非常类似。3Sum我们是固定i,令j=i+1, k=n-1。
本题由于三角形需要两边之和大于第三边,nums[i]+nums[j]>nums[k]。因此我们可以固定k,令i=0, j=k-1。
如果nums[i]+nums[j]>nums[k],说明 [i,j-1] 范围内的所有数作为i,加上j和k都是可以的。
反之如果nums[i]+nums[j]≤nums[k],那我们必须增加nums[i],和才会增大。
class Solution {
public:
int triangleNumber(vector<int>& nums) {
int n=nums.size();
int res=0;
sort(nums.begin(),nums.end());
for (int k=n-1;k>=0;--k){
int i=0, j=k-1;
while (i<j){
if (nums[i]+nums[j]>nums[k]){
res += j-i;
--j;
}else{
++i;
}
}
}
return res;
}
};
时间复杂度 O(n^2)
Reference
https://leetcode.com/problems/valid-triangle-number/discuss/128135/A-similar-O(n2)-solution-to-3-Sum