Get Task Index at Cycle

main.py

def get_task_index_at_cycle(jobs, cycle):
  execution_order = sorted(range(len(jobs)), key=lambda k: (jobs[k], k))
  total_time = sum(jobs)
  cycle = cycle % total_time + 1 if cycle >= total_time else cycle
  current_time = 0

  for job_index in execution_order:
    current_time += jobs[job_index]
    if cycle <= current_time:
      return job_index

  return execution_order[-1]