C++ 中小於 K 的子陣列乘積
假設我們給定了一個正整數陣列 nums。我們必須計算並打印出 (連續) 子陣列的數量,其中每個子陣列中元素的乘積都小於 k。因此,如果輸入類似於 [10, 5, 2, 6] 且 k := 100,則輸出將為 8。因此,子陣列將為 [[10], [5], [2], [6], [10, 5], [5, 2], [2, 6] 和 [5, 2, 6]]
為了解決這個問題,我們將遵循以下步驟 -
- temp := 1,j := 0 且 ans := 0
- 對於 i 的範圍從 0 到陣列的大小
- temp := temp * nums[i]
- 當 temp >= k 且 j <= i 時,執行
- temp := temp / nums[j]
- 將 j 增加 1
- ans := ans + (i – j + 1)
- 返回 ans
示例(C++)
讓我們看看以下實現,以便更好地理解 -
#include <bits/stdc++.h> using namespace std; typedef long long int lli; class Solution { public: int numSubarrayProductLessThanK(vector<int>& nums, int k) { lli temp = 1; int j = 0; int ans = 0; for(int i = 0; i < nums.size(); i++){ temp *= nums[i]; while(temp >= k && j <= i) { temp /= nums[j]; j++; } ans += (i - j + 1); } return ans; } }; main(){ Solution ob; vector<int> v = {10,5,2,6}; cout << (ob.numSubarrayProductLessThanK(v, 100)); }
輸入
[10,5,2,6] 100
輸出
8
廣告