UVA 12174 Shuffle (滑动窗口)

题意:有一个播放器用于播放音乐,现在给出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;
    }
}


 

posted @ 2019-01-16 15:20  dslybyme7  阅读(120)  评论(0编辑  收藏  举报