Thieves and Prisons

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Thieves_and_Prisons {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        int N = Integer.valueOf(input[0]);
        int K = Integer.valueOf(input[1]);
        int M = Integer.valueOf(input[2]);
        int pointer = 0;
        int inPrisoned = 0;
        String[] result = new String[M];
        boolean impossible = false;
        for(int i = 0; i < M; i++){
            input = br.readLine().split(" ");
            if(input[0].equals("C")){
                if(inPrisoned == N){
                    impossible = true;
                    break;
                }
                inPrisoned++;
                result[i] = ""+(pointer+1);
                if(pointer != K-1) pointer++;
            }else{ //check if thieve x is in prison
                if(pointer == 0){
                    impossible = true;
                    break;
                }
                result[i] = ""+(inPrisoned >= K ? (pointer+1) : pointer); // I think this is right
                pointer -= (inPrisoned >= K ? 0 : 1);
                inPrisoned -= (inPrisoned >= K ? inPrisoned - K + 1 : 1);
            }
        }
        if(!impossible){
            for(int i = 0; i < result.length; i++){
                System.out.print(result[i] + (i == result.length-1 ? "" : " "));
            }
        }else{
            System.out.print("IMPOSSIBLE");
        }
    }
}

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: WRONG ANSWER

Test 9

Group: 1, 3, 4, 5

Verdict: WRONG ANSWER

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: WRONG ANSWER

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: ACCEPTED

Test 23

Group: 5

Verdict: ACCEPTED

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: ACCEPTED

Test 29

Group: 4, 5

Verdict: WRONG ANSWER

Test 30

Group: 4, 5

Verdict: WRONG ANSWER

Test 31

Group: 4, 5

Verdict: WRONG ANSWER

Test 32

Group: 2, 4, 5

Verdict: ACCEPTED

Test 33

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 34

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 35

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 36

Group: 3, 5

Verdict: TIME LIMIT EXCEEDED

Test 37

Group: 3, 5

Verdict: TIME LIMIT EXCEEDED

Test 38

Group: 3, 5

Verdict: WRONG ANSWER

Test 39

Group: 3, 5

Verdict: WRONG ANSWER

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