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;
}