This function is to generate a grid of treatment status
treatment_status(N, K, stime, treatment_init, treatment_shift_time, max_time)