枚举可以选取的区间的最后一个数字,可以选取的的区间一定是以枚举的数字为末尾的一段,而且这一段内相同数字的下标之差<= k

因此当枚举的位置向右移动一位后,区间的的左端可能不变,也可能因为区间内划入了新加入的这个数导致左端必须向右移动,而无论哪种情况不会让左端左移,所以随着枚举位置右移,左端也是单调右移的,因此可以用双指针维护区间。 维护每个数上一次出现的位置即可,细节见代码。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int n, k;
int main() {
    cin >> n >> k;
    vector<int> a(n);
    for (auto& x : a) {
        cin >> x;
    }
    
    vector<int> last(100010, -1);
    ll res = 0;
    for (int i = 0, j = 0; i < n; i ++) {
        if (last[a[i]] >= 0 and i - last[a[i]] > k) {
            j = max(j, last[a[i]] + 1);
        }
        res += max(i - j + 1, 0);
        last[a[i]] = i;
    }
    
    cout << res << '\n';
    return 0;
}