这个思路和代码是参考紫书。保存数的长度为s + n + s + 1。这样能保证窗口可以从第一个数开始滑,可以求出所有情况(有例外,就是第三组数据类型)。以第一组数据为例,
数组为 -1,-1,-1,-1,3,4,4,1,3,2,1,2,3,4,-1,-1,-1,-1。窗口长度为s = 4。从第一个数开始滑,判断此窗口合法性,并进行有效数的数量调整,下一个窗口可据此判断。接下来,只需统计滑动的可能性就性了。
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxn = 3 * 100000 + 5;
int num[maxn];
int ok[maxn];
int cnt[maxn];
int main()
{
int T;
scanf("%d", &T);
while(T--)
{
int s, n;
scanf("%d%d", &s, &n);
fill(num, num + s * 2 + n + 1, -1);
for(int i = 0; i < n; i++)
scanf("%d", &num[s + i]);
memset(ok, 0, sizeof(cnt));
memset(cnt, 0, sizeof(cnt));
int tot = 0;
for(int i = 0; i < s + n + 1; i++)
{
if(tot == s) ok[i] = 1;
if(i < s && tot == i) ok[i] = 1;
if(i > n && tot == n + s - i) ok[i] = 1;//因为左右是补的虚拟窗户,所以要特殊处理。
if(i == s + n) break;
if((num[i] != -1) && --cnt[num[i]] == 0) tot--;
if(num[i + s] != -1 && cnt[num[i + s]]++ == 0) tot++;
}
int ans = 0;
for(int i = 0; i < s; i++)
{
int valid = 1;
for(int j = i; j < n + s + 1; j = j + s)
{
//printf("%d %d\n", j, ok[j]);
if(!ok[j]) valid = 0;
}
if(valid) ans++;
}
if(ans == n + 1) ans = s;//请思考最后一组数据
printf("%d\n", ans);
}
return 0;
}