滑动窗口思想。
用一个数组flag记录每个点是否可以作为一个窗口的起点(窗口内的元素不能重复),O(n)。
然后,枚举第一个完整窗口的起点(0-s-1)。如果某个元素出现第二次,那么在两个相同元素之间一定需要一个窗口的边界,跳出循环。
s 可以> n,所以不能等输入完成了再去统计前s个,这样是不对的。
for(int i = 0; i < n; ++i) cin>>a[i]
int num = 0;//重复在当前窗口重复出现的个数
for(int i = 0; i < s; ++i){
if(vis[a[i]]) ++num;
++vis[a[i]];
}
正确的是:
int num = 0;//重复在当前窗口重复出现的个数
for(int i = 0; i < n; ++i) {
scanf("%d",&A[i]);
if(i < s){
if(vis[A[i]]) ++num;
++vis[A[i]];
}
}
完整代码:
#include <cstdio>
#include <cstring>
const int N = 100000+5;
int s, n, ans, A[N], vis[N];
bool flag[N];// flag[i]: i 是否可以作为一段的起点
void init(){
memset(vis,0,sizeof(vis));
memset(flag,0,sizeof(flag));
scanf("%d %d",&s,&n);
//统计前s个
int num = 0;//重复在当前窗口重复出现的个数
for(int i = 0; i < n; ++i) {
scanf("%d",&A[i]);
if(i < s){
if(vis[A[i]]) ++num;
++vis[A[i]];
}
}
for(int i = 0; i < n; ++i){
if(0 == num) flag[i] = true; // 以i开始的这一段没有重复,i可以作为一段的起点
// 滑动窗口
if(vis[A[i]] > 1) --num; // 如果这个数是重复的,因为它马上要离开窗口,所以num--。
--vis[A[i]];
int k = i + s; // 右移
if(k >= n) continue;
if(vis[A[k]]) ++num;
++vis[A[k]];
}
}
// 判断以x作为起点,是否可行
bool judge(int x){
for(int i = x; i < n; i+= s)
if(!flag[i]) return false;
return true;
}
void solve(){
memset(vis,0,sizeof(vis));
ans = 0;
for(int i = 0; i < s; ++i){
if(judge(i)) ++ans;
if(i >= n) continue;
if(vis[A[i]]) break;
++vis[A[i]];
}
}
int main()
{
//freopen("in.txt","r",stdin);
int T; scanf("%d",&T);
while( T-- ){
init();
solve();
printf("%d\n",ans);
}
return 0;
}