[算法竞赛入门经典] UVA 12174 - Shuffle

紫书P247 例题8-15

先预处理一遍每个窗口内的每个数字是否只出现一次，结果存到flag数组里。

还有个特殊构造的地方，由于首尾窗口可能存在不完整的情况，那么我们在n个数后面在添加s个元素，尾部窗口就不会出现不完整的情况了。

然后直接枚举求出结果。

#include <stdio.h>  
#include <string.h>  
#include <math.h>  
#include <stdlib.h>  
#include <algorithm>  
#include <iostream>  
#include <set>  
#include <map>  
#include <queue>  
#include <stack>
#include <assert.h>
typedef long long LL;
using namespace std;
int a[200001], b[300001], flag[200001];
int main()  
{  
    //freopen("test0.in", "r", stdin);  
    //freopen("test0.out", "w", stdout);  
    int T, s, n, ans;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d %d", &s, &n);
        for(int i = 0; i < n; ++i)
        {
            scanf("%d", &a[i]);
        }
        for(int i = n; i < n + s; ++i)
        {
            a[i] = 100001+i-n;
        }
        memset(flag, 0, sizeof(flag));
        memset(b, 0, sizeof(b));
        int cnt = 0;
        for(int i = 0; i < n+s; ++i)
        {
            if(i >= s)
            {
                --b[a[i-s]];
                if(b[a[i-s]] >= 1)
                {
                    --cnt;
                }
            }
            ++b[a[i]];
            if(b[a[i]] == 1 && cnt == 0)
            {
                flag[i] = 1;
            }
            else if(b[a[i]] > 1)
            {
                ++cnt;
            }
        }
        ans = 0;
        for(int i = 0; i < s; ++i)
        {
            int flg = 1;
            for(int j = i; j < n+s; j += s)
            {
                if(!flag[j])
                {
                    flg = 0;
                    break;
                }
            }
            if(flg)
            {
                ++ans;
            }
        }
        printf("%d\n", ans);
    }
    return 0;  
}