题解 | #小黑的区间# D 双指针

小黑的区间

https://ac.nowcoder.com/acm/contest/84244/D

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

相关推荐

不愿透露姓名的神秘牛友
09-23 18:40
点赞 评论 收藏
分享
10-09 17:17
已编辑
门头沟学院 Java
活泼的代码渣渣在泡池...:同学你好,我也是学院本,后天要面这个亚信科技,是实习,请问问题都啥样呀,我项目就做了网上的,这是第一次面试
投递多益网络等公司10个岗位
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务