Lintcode三角形计数

最新推荐文章于 2018-11-23 21:21:35 发布

年轻的老干爹

最新推荐文章于 2018-11-23 21:21:35 发布

阅读量252

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/u013174482/article/details/76588596

本文介绍了一种针对三角形计数问题的算法优化方法。通过对比原始暴力解法与改进后的高效解法，展示了如何利用排序和双指针技巧减少时间复杂度，从而有效提高算法效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

自己写的代码，简单粗暴，但是超时。

class Solution {
public:
    /*
     * @param : A list of integers
     * @return: An integer
     */
    int triangleCount(vector<int> S) {
        // write your code here
        int len=S.size();
        int num=0;
        for(int i=0;i<len-2;i++){
            for(int j=i+1;j<len-1;j++){
                for(int k=j+1;k<len;k++){
                    if(S[i]+S[j]>S[k])
                    num++;
                }
            }
        }
        return num;
    }
};

改进的代码

class Solution {  
public:  
    /** 
     * @param S: A list of integers 
     * @return: An integer 
     */  
    int triangleCount(vector<int> &S) {  
        // write your code here  
        int ret = 0;  
        int n = S.size();  
        sort(S.begin(), S.end());  
        for (int i = 0;i<n;i++)  
        {  

            for (int j = i + 1;j<n;j++)  
            {  
                int l = i+1;  
                int r = j;  
                int target = S[j]-S[i];  
                while(l<r)  
                {  
                    int mid = (l+r)/2;  
                    if(S[mid]>target)  
                        r = mid;  
                    else  
                        l = mid+1;  
                }  
                ret += (j-l);  
            }  
        }  
        return ret;  
    }  
};