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 + ;
 
 int s, n, a[N], vis[N];
 bool flag[N];
 int ans;
 
 void init() {
     cin >> s >> n;
     int num = ;
     for (int i = ; i < n; i++) {
         cin >> a[i];
         if (i < s) {   //对前面的s个进行分析
             if (vis[a[i]]) num++;   //统计前s个中重复的数字
             vis[a[i]]++;
         }
     }
 
     for (int i = ; i < n; i++) {
         //如果num=0，说明前s个中没有重复的数字，那么第一个数字可以作为循环的开始
         if (num == ) flag[i] = true;
 
         //窗口开始滑动
         if (vis[a[i]] == ) 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, , sizeof(vis));
 
     ans = ;
     for (int i = ; 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, , sizeof(flag));
         memset(vis, , sizeof(vis));
         init();
         solve();
         cout << ans << endl;
     }
     return ;
 }

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

 #include<iostream>
 #include<cstring>
 #include<set>
 using namespace std;
 
 const int maxn =  + ;
 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, , sizeof(flag));
         cin >> s >> n;
         for (int i = ; i < n; i++)
         {
             cin >> a[i];
         }
         for (int i = ; i < n; i++)
         {
             int ok = ;
             memset(vis, , sizeof(vis));
             for (int j = i; j < i + s && j < n; j++)
             {
                 if (vis[a[j]])    break;
                 vis[a[j]] = ;
                 if (j == i + s -  || j==n-)    ok = ;
             }
             if (ok)     flag[i] = ;
         }
         ans = ;
         memset(vis, , sizeof(vis));
         for (int i = ; i < s; i++)
         {
             if (judge(i))    ans++;
             if (i >= n)    continue;
             if (vis[a[i]])    break;
             vis[a[i]] = ;
         }
         cout << ans << endl;
     }
     return ;
 }