../ Competitive Programming
-rw-r--r--aguang648 B1 min read

Leet Code Daily 2024-06-22 - Count Number of Nice Subarrays

Solution

Preffix sum.

Code

c++
class Solution {
public:
    int numberOfSubarrays(vector<int>& nums, int k) {

        vector<int> sums = {0};
        unordered_map<int, int> cnt;
        cnt[0]++;

        int ans = 0;
        for (const auto& num : nums) {
            const int sum = sums.back() + (num % 2);
            sums.push_back(sum);
            if (sum - k >= 0) {
                ans += cnt[sum - k];
            }
            cnt[sum]++;
        }
        return ans;
    }
};

COMMAND INTERFACE

↑ ↓ navigate · Enter execute · Esc exit · try “whoami”

Loading filesystem…

:help aguang.xyz

A personal website disguised as an operating system.

Ctrl / Cmd + K
Command palette + full-text search
Tab / Shift + Tab
Move between links and controls
Enter
Open a focused link or search result
↑ / ↓ in search
Navigate search results
In a directory filter
Enter/Done or ↓ focuses a result; Esc clears the filter
Esc
Clear a search or close a window