P1638 逛画展 - 洛谷 | 计算机科学教育新生态
这道题我们只要用一个kind和一个mp[N]的数组就能解决了
我们的解法1就是暴力枚举,先固定2,从2开始找连续的满足所有种类的最短的子数组,然后固定5,3,1,3,2,分别找出满足所有种类的最短子数组
mp[i]如果是从0到1,kind++,如果是从1到0,kind--
如图,暴力枚举的话j指向的一定是第一次出现的最新的元素种类,如果我们是暴力枚举的话,我们枚举5的时候,j也会回到5
我们对5枚举的时候,j一定会再走到4那个位置,我们何必让j回退呢?
那我们的算法流程就是用left和right指针指向第一个元素,然后让right指针向后走把每个元素进窗口,如果kind种类够了的话,left不断++,再不断更新每个合法的子数组的大小,直到kind不够了再退出去继续right向后走
比如这是一种结果,2出窗口后又是一个结果
5出窗口后种类不够了,j向后走
不满足要求就不更新结果直到再次符合要求,我们再次让left出窗口
到这里把3出窗口之后再次成为不合法数组
再次合法,继续出left,出了一个就不合法了,right向后走一格,结束
我们来展示一下代码
#include <iostream>
using namespace std;const int N = 1e6+10;
int n,m;
int mp[N];
int a[N];
int main()
{cin >> n >> m;for(int i = 1;i<=n;i++) cin >> a[i];int kind = 0;int left = 1 , right = 1;int ret = n,begin = 1;while(right<=n){if(mp[a[right]]++ == 0) kind++;while(kind == m){int len = right-left+1;if(len < ret){ret = len;begin = left;}if(mp[a[left]]-- == 1) kind--;left++;} right++;}cout << begin <<" " << begin+ret-1<< endl;return 0;
}