题意:![在这里插入图片描述](https://img-blog.csdnimg.cn/53f5917cccc84efcbce78c85dd26a970.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBAbGl2ZTRt,size_20,color_FFFFFF,t_70,g_se,x_16)
解法:
注意:题目的d不一定是n的因子,因此不能直接下标对d取模分组.
什么时候a[i]会从1变成0?
一定是a[i]=1且a[i+d]=0的时候,
我们反过来考虑,如果a[i]=0且a[i-d]=1,那么a[i-d]就会变成0.
从0开始bfs即可.
对于a[x]=0的x,如果a[x-d]=1,
那么将a[x-d]由1变成0,并将x-d推入队列.
如果bfs之后仍然有数=1,那么无解.
否则答案就是bfs的轮数.
算法复杂度O(n).
code:
#include<bits/stdc++.h>
#define MULTI_CASE
#define PI pair<int,int>
using namespace std;
const int maxm=2e6+5;
int a[maxm];
int n,d;
void solve(){
cin>>n>>d;
for(int i=0;i<n;i++){
cin>>a[i];
}
int ans=-1;
queue<int>q;
for(int i=0;i<n;i++){
if(!a[i])q.push(i);
}
while(q.size()){
int len=q.size();
ans++;
for(int i=0;i<len;i++){
int x=q.front();q.pop();
int pre=(x-d+n)%n;
if(a[pre]){
a[pre]=0;
q.push(pre);
}
}
}
for(int i=0;i<n;i++){
if(a[i]){
cout<<-1<<endl;return ;
}
}
cout<<ans<<endl;
}
void Main(){
#ifdef MULTI_CASE
int T;cin>>T;while(T--)
#endif
solve();
}
void Init(){
ios::sync_with_stdio(0);cin.tie(0);
#ifndef ONLINE_JUDGE
freopen("../in.txt","r",stdin);
freopen("../out.txt","w",stdout);
#endif
}
signed main(){
Init();
Main();
return 0;
}