leetcode 862. Shortest Subarray with Sum at Least K

一只走向程序员之路的菜鸟

于 2018-07-10 13:10:22 发布

阅读量366

点赞数

分类专栏： leetcode 文章标签： c++ leetcode

本文链接：https://blog.youkuaiyun.com/jfnfjivkdnfjf/article/details/80984330

版权

leetcode 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一个使用C++ STL中set和map解决特定问题的实例：寻找最短子数组，其元素之和至少为给定值K。通过实际代码展示了如何利用set和map来高效地解决这个问题，并分享了作者对于STL进一步学习的认识。

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

尴尬，第一次知道原来set是有序的，，，丢人丢人。。。

看来最近在恶补STL是很正确的选择。。。。。

class Solution{
public:
    int shortestSubarray(vector<int> &A , int K){
        map<long long , int> ma;
        set<long long> se;
        se.insert(0);
        ma[0]=0;
        int ans=INT_MAX;
        long long tot=0;
        for(int i=0;i<A.size();i++){
            tot+=A[i];
            ma[tot]=i+1;
            auto au = se.lower_bound(tot - K);
            if(au != se.end() && *au == tot-K){
                ans = min(ans , i-ma[*au]+1);
            }
            else if(au != se.begin()){
                au--;
                ans=min(ans , i - ma[*au] + 1);
            }
            se.insert(tot);
            au = se.lower_bound(tot);
            se.erase(au , se.end());
            se.insert(tot);
        }
        return ans == INT_MAX? -1:ans;
    }
};