#include <bits/stdc++.h>
using namespace std;
int distinctSubarrayCount(vector<int> &arr, int k){
int count = 0;
for (int i = 0; i < arr.size(); i++){
unordered_map<int, int> freq;
for (int j = i; j < arr.size(); j++){
freq[arr[j]]++;
int n = freq.size();
if (n <= k) count++;
}
}
return count;
}
int main() {
vector<int> arr = {1, 2, 3};
cout << distinctSubarrayCount(arr, 2);
return 0;
}
I2luY2x1ZGUgPGJpdHMvc3RkYysrLmg+CnVzaW5nIG5hbWVzcGFjZSBzdGQ7CgppbnQgZGlzdGluY3RTdWJhcnJheUNvdW50KHZlY3RvcjxpbnQ+ICZhcnIsIGludCBrKXsKCWludCBjb3VudCA9IDA7Cglmb3IgKGludCBpID0gMDsgaSA8IGFyci5zaXplKCk7IGkrKyl7CgkJCgkJdW5vcmRlcmVkX21hcDxpbnQsIGludD4gZnJlcTsKCQlmb3IgKGludCBqID0gaTsgaiA8IGFyci5zaXplKCk7IGorKyl7CgkJCWZyZXFbYXJyW2pdXSsrOwoJCQkKCQkJaW50IG4gPSBmcmVxLnNpemUoKTsKCQkJCgkJCWlmIChuIDw9IGspIGNvdW50Kys7IAoJCX0KCX0KCQoJcmV0dXJuIGNvdW50Owp9CgppbnQgbWFpbigpIHsKCXZlY3RvcjxpbnQ+IGFyciA9IHsxLCAyLCAzfTsKCWNvdXQgPDwgZGlzdGluY3RTdWJhcnJheUNvdW50KGFyciwgMik7CglyZXR1cm4gMDsKfQ==