prison_cells_after_N_days.java

prison_cells_after_N_days.java

class Solution {
    public int[] prisonAfterNDays(int[] cells, int N) {
                
        int cycle =1;
        int first[] = new int[8];

        for(int i=1; i<7; i++)
        {
            first[i] = (cells[i-1] == cells[i+1] ? 1 :0);
        }
        N-=1;
        
        for(int i=0; i<8; i++)
        cells[i] = first[i];
        
    
        while(N-- >0)
        {
        
            int temp [] = new int [8];    
     
        for(int i=1; i<7; i++)
        {
            temp[i] = (cells[i-1] == cells[i+1] ? 1 :0);
        }
      
            if(Arrays.equals(temp, first)) N%=cycle;
     
               for(int i=0; i<8; i++)
               cells[i] = temp[i];   
       
            cycle++;
        }
        
        return cells;
        
        
    }
    
}