枚举可以选取的区间的最后一个数字,可以选取的的区间一定是以枚举的数字为末尾的一段,而且这一段内相同数字的下标之差<= 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;
}