3639 T2 点亮灯笼

#include<iostream>
using namespace std;
const int N=1005;
int n,m,a[N],f;
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        a[i]=1;
    }
    for(int i=1;i<=m;i++){
        f=0;
        cin>>f;
        if(f!=m&&f!=m-1&&f!=1){
            a[f-1]*=-1;
            a[f]*=-1;
            a[f+1]*=-1;
        }
        else if(f==m){
            a[1]*=-1;
            a[m]*=-1;
            a[m-1]*=-1;
        }
        else if(f==m+1){
            a[m]*=-1;
            a[1]*=-1;
            a[2]*=-1;
        }
    }
    for(int i=1;i<=n;i++){
        if(a[i]=1){
            a[i]=0;
        }
        else{
            a[i]=1;
        }
    }
    for(int i=1;i<=n;i++){
        cout<<a[i];
    }
    return 0;
}

求调！回帖必关uwu