Codeforces1579 F. Array Stabilization (AND version)（思维+bfs）

Posted 2021-10-26 live4m

题意：

解法：

注意:题目的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>
// #define int long long
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()){    //bfs
        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;
}

---