Codeforces 1237D. Balanced Playlist

传送门

首先显然的,如果一个位置开始播放了两圈还没结束,那么就永远不会结束

先考虑位置 $1$ 开始播放,用一个 $multisetset$ 维护一下当前听的所有歌,直到某一首歌 $r$ 不合法了就停止,此时播放的区间即为位置 $1$ 开始的答案

然后考虑从位置 $2$ 开始播放时和从位置 $1$ 开始播放有什么变化,显然播放的歌曲一定可以到 $r$ (反证法容易证明),并且 $multiset$ 里少了一首位置 $1$ 的歌

那么直接把 $multiset$ 更新一下,然后继续模拟过程直到下一个不合法,然后又可以算出答案

这样一直算下去即可求出所有答案

均摊复杂度 $O(n \log n)$,注意要把听歌的序列延长 $3$ 倍而不是两倍(看样例 $2$ 就知道了)

数组也要开 $3$ 倍!(如果你像我一样不小心开了两倍,那么结果就是通过 $\text{pretest}$ 然后 $WA$ 在 $\text{system test}$)

#include
#include
#include
#include
#include
#include<set>
using namespace std;
typedef long long ll;
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while(ch>='0'&&ch<='9') { x=(x<<1)+(x<<3)+(ch^48); ch=getchar(); }
    return x*f;
}
const int N=3e5+7;
int n,n3,ans[N];
int A[N];
multiset <int> S;
int main()
{
    n=read(),n3=n*3;
    for(int i=1;i<=n;i++) A[n*2+i]=A[n+i]=A[i]=read();
    int r=1;
    for(int i=1;i<=n;i++)
    {
        while(r<=n3)
        {
            if(S.size())
            {
                int p=*S.rbegin();
                if(A[r]

2 || (A[r]==p/2 && p&1)) break; } S.insert(A[r]); r++; } ans[i]= r<=n3 ? r-i : -1; S.erase(S.find(A[i])); } for(int i=1;i<=n;i++) printf("%d ",ans[i]); puts(""); return 0; }

 

你可能感兴趣的:(Codeforces 1237D. Balanced Playlist)