#include <stdio.h>
#include <stdlib.h>

int* light_change(int* A,int N,int K){
    int* B=(int*)malloc(sizeof(int));
    
    int i,j,k;
    for(k=0;k<K;k++){
        for(i=0;i<N;i++){
            B[i]=0;
        }
       for(i=0;i<N;i++){
           for(j=(i-A[i]>=0? i-A[i]:0); j<(i+A[i]+1<=N ? i+A[i]+1: N); j++){
            B[j]++;
        }
    }
    for(i=0;i<N;i++){
        A[i]=B[i];
    }
    }
    free(B);
    return A;
}


int main(void){
    // Your code here!
    //int N,K;
    //scanf("%d %d",&N,&K);
    //printf("%d %d",N,K);
    int N=5;
    int K=2;
    int A[]={1,0,0,1,0};
    light_change(A,N,K);
    
    int i=0;
    for(i=0;i<N;i++){
        printf("%d ",A[i]);
    }
    return 0;
}
