LeetCode862。少なくともKを持つ最短のサブ配列(接頭辞と+ deque単調スタック)

1.タイトル

戻り最短非空連続サブアレイ、サブアレイ及び少なくともK

少なくともKの空でないサブ配列がない場合は、-1を返します。

示例 1
输入:A = [1], K = 1
输出:1

示例 2
输入:A = [1,2], K = 4
输出:-1

示例 3
输入:A = [2,-1,2], K = 3
输出:3
 
提示:
1 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
1 <= K <= 10 ^ 9

出典:LeetCode
リンク:https ://leetcode-cn.com/problems/shortest-subarray-with-sum-at-least-k
著作権は控除ネットワークに属しています。商用転載の正式な許可書に連絡し、非商用転載の出典を明記してください。

2.問題解決

同様のトピック:LeetCode560。SumはKのサブ配列です(プレフィックスと差分)。

  • 公式のアイデアを参照してください
  • dequeはプレフィックスとサブスクリプトを格納します。チーム内のプレフィックスは厳密に単調に増加する必要があります
  • リーダーとの差が> = kの場合、最小の長さを記録し、リーダーを削除します
    ここに画像の説明を挿入
class Solution {
public:
    int shortestSubarray(vector<int>& A, int K) {
    	vector<int> presum(A.size()+1,0);
    	int i, minLen = INT_MAX, sum = 0;
    	deque<int> q;//存储下标,按照队内presum升序
    	q.push_back(0);//边界条件,前缀和为0,下标为0
    	for(i = 0; i < A.size(); ++i)
    	{
    		presum[i+1] = presum[i] + A[i];//前缀和
    		while(!q.empty() && presum[q.back()] >= presum[i+1])
    			q.pop_back();
    		while(!q.empty() && (presum[i+1]-presum[q.front()] >= K))
    		{
    			minLen = min(minLen, i+1-q.front());
    			q.pop_front();
    		}
    		q.push_back(i+1);
    	}
    	if(minLen == INT_MAX)
    		return -1;
    	return minLen;
    }
};

ここに画像の説明を挿入

公開された897元の記事 ウォンの賞賛2749 ビュー48万+

おすすめ

転載: blog.csdn.net/qq_21201267/article/details/105662545