洛谷题单指南-常见优化技巧-P1638 逛画展
原题链接:https://www.luogu.com.cn/problem/P1638
题意解读:在n个数中,选出a、b两个端点,使得a~b之间不同的数字为m,且b-a最小。
解题思路:
要寻找最小的包括所有数字的区间,可以采用双指针算法
1、设i,j分别是左右指针
2、如果当前区间内不同数字个数不到m,j往后移
3、记录数字个数到m时的区间距离最小的左、右指针下标
4、i右移,重复上述过程
100分代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 1000005, M = 2005;
int n, m;
int a[N], h[M];
int l, r, minx = 1e9;
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++) cin >> a[i];
int i = 1, j = 0, cnt = 0;
while(i <= n && j <= n)
{
if(j <= n && cnt < m)
{
j++;
if(++h[a[j]] == 1) cnt++;
}
if(cnt == m)
{
if(j - i < minx) l = i, r = j, minx = r - l;
if(--h[a[i]] == 0) cnt--;
i++;
}
}
cout << l << " " << r;
return 0;
}