Go to Play Maimai DX 2023牛客暑期多校训练营5 G

登录—专业IT笔试面试备考平台_牛客网

题目大意：给出一长度为n的仅由1,2,3,4组成的数组和一整数k，求一个最短的区间使得1,2,3,4至少各有一个，且4的数量>=k

1<=k<=n<=1e5

思路：用双指针l，r维护合法区间，先向右扩展r，同时记录访问过几个不同的数以及4的数量直到满足要求，然后缩短左端点l，维护当前区间内每个数的个数，直到不满足条件就记录答案

//#include<__msvc_all_public_headers.hpp>
#include
using namespace std;
typedef long long ll;
const int N = 1e5 + 5;
const int INF = 0x7fffffff;
int a[N];
int vis[5];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n, k;
	cin >> n >> k;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	int l = 1, r = 1;
	int cnt = 0, cnt4 = 0;
	int ans = INF;
	while (1)
	{
		bool temp = 0;
		while (r <= n && (cnt != 4 || cnt4 < k))
		{
			if (!vis[a[r]])
				cnt++;
			vis[a[r]]++;
			if (a[r] == 4)
				cnt4++;
			r++;
		}
		while(cnt==4&&cnt4>=k)
		{		
			temp = 1;
			vis[a[l]]--;
			if (!vis[a[l]])
				cnt--;
			if (a[l] == 4)
				cnt4--;
			l++;
		}
		if (temp)
		{
			ans = min(ans, r - l + 1);
		}
		else
			break;
	}
	cout << ans << endl;
	return 0;
}