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

来源：互联网发布：美工设计电脑配置编辑：程序博客网时间：2024/05/16 06:54

紫书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;  }

0 0