UVa 12174 Shuffle（滑动窗口）

https://vjudge.net/problem/UVA-12174

题意：

你在听音乐播放器，它采用随机播放形式。随机播放的原理时先随机产生一个1~n的排列，然后就按这个排列顺序播放歌曲。播放完这序列的所有歌曲以后，再次随机生成一个1～n的排列，再继续播放。

现在给你一个播放历史记录，这个历史记录是不完整的，以为它是开始记录时，已经有些歌曲播放过了而没有记录到。

现在给你一段从某个时刻开始的播放音乐的历史记录，以及播放器里一共有多少首歌。

问历史记录中第一首歌可能是某个随机列表中的第几首，总共有多少种可能？

思路：

滑动窗口，依次遍历，检查每个数能否作为循环的开始。

#include<iostream> 
#include<cstring>
#include<set>
using namespace std;

const int N = 1e5 + 5;

int s, n, a[N], vis[N];
bool flag[N];
int ans;

void init() {
    cin >> s >> n;
    int num = 0;
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        if (i < s) {   //对前面的s个进行分析
            if (vis[a[i]]) num++;   //统计前s个中重复的数字
            vis[a[i]]++;
        }
    }

    for (int i = 0; i < n; i++) {
        //如果num=0，说明前s个中没有重复的数字，那么第一个数字可以作为循环的开始
        if (num == 0) flag[i] = true;

        //窗口开始滑动
        if (vis[a[i]] == 2) num--;    //如果此时最左边的数为重复了的数，num需要减1
        vis[a[i]]--;

        int k = i + s;     //新数字进入滑动窗口
        if (k >= n) continue;
        if (vis[a[k]]) num++;   //如果已经出现过
        vis[a[k]]++;
    }
}

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;
        //从左往右依次遍历，如果当前a[i]前面已经出现过，那么前面必须会有开头，此时必须结束循环
        if (vis[a[i]]) break;
        vis[a[i]]++;
    }
}

int main() {
    //freopen("D:\txt.txt", "r", stdin);
    int t;
    cin >> t;
    while (t--) {
        memset(flag, 0, sizeof(flag));
        memset(vis, 0, sizeof(vis));
        init();
        solve();
        cout << ans << endl;
    }
    return 0;
}

本来我是这样写的，结果不行，超时了。

#include<iostream> 
#include<cstring>
#include<set>
using namespace std;

const int maxn = 100000 + 5;
int a[maxn];
int vis[maxn],flag[maxn];

int s, n, ans;

bool judge(int x)
{
    for (int i = x; i < n; i += s)
    if (!flag[i]) return false;
    return true;
}

int main() {
    //freopen("D:\txt.txt", "r", stdin);
    int t;
    cin >> t;
    while (t--) 
    {
        memset(flag, 0, sizeof(flag));
        cin >> s >> n;
        for (int i = 0; i < n; i++)
        {
            cin >> a[i];
        }
        for (int i = 0; i < n; i++)
        {
            int ok = 0;
            memset(vis, 0, sizeof(vis));
            for (int j = i; j < i + s && j < n; j++)
            { 
                if (vis[a[j]])    break;
                vis[a[j]] = 1;
                if (j == i + s - 1 || j==n-1)    ok = 1;
            }
            if (ok)     flag[i] = 1;
        }
        ans = 0;
        memset(vis, 0, sizeof(vis));
        for (int i = 0; i < s; i++)
        {
            if (judge(i))    ans++;
            if (i >= n)    continue;
            if (vis[a[i]])    break;
            vis[a[i]] = 1;
        }
        cout << ans << endl;
    }
    return 0;
}