题意:有一个播放器用于播放音乐,现在给出s(已有曲目的数量),n给出记录的长度。播放器有随机播放的功能,每次生成一个1~s的随机系列进行播放,当s首歌全部播放完后,重新生成一个播放序列。现在有一段长度为n的播放记录片段(即不完全,前后可能还有歌曲),问下一次随机排序的位置有几种可能。(这题意读了好久。。)
分析:可以开一个n+s的数组,从后向前处理,判断每一个窗口是否能成为一个播放列表,之后枚举可能的情况。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define FRER() freopen("in.txt","r",stdin)
#define FREW() freopen("out.txt","w",stdout)
#define go int T;cin>>T;while(T--)
using namespace std;
const int maxn = 2e5 + 7;
int a[maxn],vis[maxn];
bool ok[maxn];
int n,s,num;
void Init(){
memset(vis, 0, sizeof(vis));
memset(a, 0, sizeof(a));
memset(ok, 0, sizeof(ok));
num = 0;
scanf("%d%d",&s,&n);
for(int i=s;i<n+s;i++)
scanf("%d",&a[i]);
int cnt = min(n,s);
int ii = n+s-1;
while(cnt--){
if(vis[a[ii]]) num++;
vis[a[ii]]++;
ii--;
}
for(int i=n+s-1;i>=0;i--){
if(num==0) ok[i] = true;
if(vis[a[i]]==2) num--;
vis[a[i]]--;
int k = i-s;
if(k<s) continue;
if(vis[a[k]])num++;
vis[a[k]]++;
}
}
bool judge(int p){
for(int i=n+s-p-1;i>=0;i-=s){
if(!ok[i]) return false;
}
return true;
}
int main(){
//FRER();
//FREW();
go{
Init();
// for(int i = n+s-1;i>=0;i--) if(ok[i]) cout<<i<<" ";
// cout<<endl;
int ans = 0;
memset(vis, 0, sizeof(vis));
for(int i=0;i<s;i++){
if(judge(i)) ans++;
if(vis[a[n+s-i-1]])break;
if(n+s-i-1<s)continue;
vis[a[n+s-i-1]]++;
}
cout<<ans<<endl;
}
}