CF1579F Array Stabilization (AND version) 题解

题目传送门

分析

根据 $\operatorname{and}$ 操作的性质，只要有一个 $0$ 存在，这个数记作 $a_i$，那么 $a_{i+d}$ 就可以变成 $0$，然后 $a_{i+d}$ 又可以继续更新下一个数为 $0$，所以直接 bfs 解决即可。

最终的答案即为 bfs 的次数。注意，由于无论如何都会入队一次，最终答案需要减一。

代码实现

#include <bits/stdc++.h>
using namespace std;
 
inline int read(){
    int x=0,f=0;char ch=getchar();
    while(!isdigit(ch))f^=!(ch^45),ch=getchar();
    while(isdigit(ch))x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    return f?-x:x;
}
inline void write(int x){
    if(x<0)x=-x,putchar('-');
    if(x>=10)write(x/10);
    putchar(x%10+'0');
}
inline void writeln(int x){write(x);puts("");}
 
int n,d;
int a[1000005];
 
int main(){
    int T=read();
    while(T--){
        n=read(),d=read();
        for(int i=0;i<n;i++)a[i]=read();
        queue <int> q;
        for(int i=0;i<n;i++)if(!a[i])q.push(i);
        int ans=0;
        while(!q.empty()){
            int len=q.size();
            for(int i=0;i<len;i++){
                int u=q.front();q.pop();
                int v=(u+d)%n;
                if(a[v]){
                    q.push(v);a[v]=0;
                }
            }
            ans++;
        }
        for(int i=0;i<n;i++)if(a[i])ans=-1;
        writeln(ans==-1?ans:ans-1);
    }
    #ifndef ONLINE_JUDGE
    system("pause");
    #endif
    return 0;
}