Thieves and Prisons

import java.util.*;

public class ThievesAndPrisons {
  
  // First method: Brute force - Just try to put the thieves in prison and see if the assignments are valid.
  // Should we simulate one at a time - or all at the same time? Lets try one at a time
  // Alternative methods can be to assign events to the prisoners, so we know if they have to be realeased sometime soon and such.

  public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    int thiefCount = scanner.nextInt();
    int prisonCount = scanner.nextInt();
    int eventCount = scanner.nextInt();

    List<Event> events = new ArrayList<>(eventCount);
    for (int i = 0; i < eventCount; i++) {
      Event.Type eType = scanner.next().equals("C") ? Event.Type.CAUGHT : Event.Type.OPEN; 
      events.add(new Event(eType, scanner.nextInt() - 1));
    }

    int runCount = 0;
    outerloop:
    while (true) {
      if (runCount > prisonCount * eventCount) {
        System.out.println("IMPOSSIBLE");
        break;
      }

      List<Thief> thieves = new ArrayList<>(thiefCount);
      for (int i = 0; i < thiefCount; i++) {
        thieves.add(new Thief(i));
      }

      List<Prison> prisons = new ArrayList<>(prisonCount);
      for (int i = 0; i < prisonCount; i++) {
        prisons.add(new Prison(i, thiefCount));
      }

      // Loop through the events and run a simulation
      String solution = "";
      int tryNum = runCount;
      innerloop:
      for (int i = 0; i < eventCount; i++) {
        Event e = events.get(i);
        int prisonNum = tryNum >= prisonCount ? (tryNum % prisonCount) : Math.max(tryNum, 0);
        tryNum -= prisonNum + 1;
        if (e.getType() == Event.Type.CAUGHT) {
          if (thieves.get(e.getThiefId()).getState() == Thief.State.PRISON) {
            runCount++;
            continue outerloop;
          } else {
            thieves.get(e.getThiefId()).Catch(prisons.get(prisonNum));
          }
        } else {
          if (thieves.get(e.getThiefId()).getState() == Thief.State.PRISON || prisons.get(prisonNum).getPrisoners().isEmpty()) {
            runCount++;
            continue outerloop;
          } else {
            prisons.get(prisonNum).OpenGate();
          }
        }

        solution += (prisonNum + 1) + " ";
      }

      // For loop suceeded, the solution must be possible
      System.out.println(solution.trim());
      break;
    }

    scanner.close();
  }

}

class Event {
  enum Type {
    CAUGHT,
    OPEN
  }

  Type type;
  int thiefId;

  public Event(Type type, int thiefId) {
    this.type = type;
    this.thiefId = thiefId;
  }

  /**
   * @return the thiefId
   */
  public int getThiefId() {
    return thiefId;
  }

  /**
   * @return the type
   */
  public Type getType() {
    return type;
  }
}

class Thief {
  enum State {
    FREE,
    PRISON
  }

  int id;
  State state;
  Prison currentPrison;

  public Thief(int id) {
    this.id = id;
    this.state = State.FREE;
    this.currentPrison = null;
  }

  /**
   * @return the state
   */
  public State getState() {
    return state;
  }

  /**
   * @return the id
   */
  public int getId() {
    return id;
  }

  /**
   * @return the currentPrison
   */
  public Prison getCurrentPrison() {
    return currentPrison;
  }

  public void Catch(Prison prison) {
    state = State.PRISON;
    currentPrison = prison;
    prison.getPrisoners().add(this);
  }

  public void Release() {
    state = State.FREE;
    currentPrison = null;
  }
}

class Prison {
  int id;
  List<Thief> prisoners;

  public Prison(int id, int thiefCount) {
    this.id = id;
    this.prisoners = new ArrayList<>(thiefCount);
  }

  public void OpenGate() {
    for (Thief prisoner : prisoners) {
      prisoner.Release();
    }

    prisoners.clear();
  }

  /**
   * @return the id
   */
  public int getId() {
    return id;
  }

  /**
   * @return the prisoners
   */
  public List<Thief> getPrisoners() {
    return prisoners;
  }
}

Test 1

Group: 2, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 4, 5

Verdict: ACCEPTED

Test 3

Group: 2, 4, 5

Verdict: ACCEPTED

Test 4

Group: 2, 4, 5

Verdict: ACCEPTED

Test 5

Group: 2, 4, 5

Verdict: ACCEPTED

Test 6

Group: 4, 5

Verdict: ACCEPTED

Test 7

Group: 4, 5

Verdict: ACCEPTED

Test 8

Group: 4, 5

Verdict: ACCEPTED

Test 9

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 10

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 11

Group: 1, 3, 4, 5

Verdict: WRONG ANSWER

Test 12

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 13

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 14

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 15

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 16

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 17

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

Test 18

Group: 1, 3, 4, 5

Verdict: ACCEPTED

Test 19

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 3, 4, 5

Verdict: WRONG ANSWER

Test 25

Group: 3, 4, 5

Verdict: WRONG ANSWER

Test 26

Group: 3, 4, 5

Verdict: WRONG ANSWER

Test 27

Group: 3, 4, 5

Verdict: WRONG ANSWER

Test 28

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 29

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 30

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 31

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 32

Group: 2, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 33

Group: 2, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 34

Group: 2, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 35

Group: 2, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 36

Group: 3, 5

Verdict: TIME LIMIT EXCEEDED

Test 37

Group: 3, 5

Verdict: TIME LIMIT EXCEEDED

Test 38

Group: 3, 5

Verdict: TIME LIMIT EXCEEDED

Test 39

Group: 3, 5

Verdict: TIME LIMIT EXCEEDED

Test 40

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 41

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 42

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 43

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 44

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED

Test 45

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED

Test 46

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED

Test 47

Group: 2, 5

Verdict: TIME LIMIT EXCEEDED