Thieves and Prisons

input/code.cpp: In function 'int main()':
input/code.cpp:54:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int j = 0; j < input.size(); j++)
                   ~~^~~~~~~~~~~~~~

#include <bits/stdc++.h>
using namespace std;

#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define trav(a, x) for(auto& a : x)
#define all(x) x.begin(), x.end()
#define allP(x) x->begin(), x->end()
#define sz(x) (int)(x).size()

// define int ll

typedef long long ll;

const int inf = 0x3f3f3f3f;
const long long mod = 1000000007;

int main()
{
	cin.sync_with_stdio(false);
	cin.tie(NULL);
	// cin.exceptions(cin.failbit);
	// cout << setprecision(10);
	// cout << fixed;

	int n, k, m; cin >> n >> k >> m;

	vector<pair<char, int>> input(m);

	for (int i = 0; i < m; i++)
	{
		char c; int temp; cin >> c >> temp;

		input[i] = make_pair(c, temp);
	}

	vector<int> prison1;
	vector<int> prison2;

	string end = "111111111111";

	double limit = pow(2, k);
	for (int i = 0; i < limit; i++)
	{
		prison1.clear();
		prison2.clear();
		vector<bool> isImprisoned(n, false);

		string permutation = bitset<10>(i).to_string();

		string subPermutation = permutation.substr(10 - m, 10);

		bool success = true;

		for (int j = 0; j < input.size(); j++)
		{
			char character = input[j].first;
			int thief = input[j].second;
			thief--;

			if (isImprisoned[thief])
			{
				success = false;
				break;
			}

			if (character == 'C')
			{
				if (subPermutation[j] == '0')
				{
					prison1.push_back(thief);
					isImprisoned[thief] = true;
				}
				else
				{
					prison2.push_back(thief);
					isImprisoned[thief] = true;
				}
			}
			else
			{
				if (subPermutation[j] == '0')
				{
					if (prison1.size() == 0)
					{
						success = false;
						break;
					}

					for (auto index : prison1)
					{
						isImprisoned[index] = false;
					}
					prison1.clear();
				}
				else
				{
					if (prison2.size() == 0)
					{
						success = false;
						break;
					}

					for (auto index : prison2)
					{
						isImprisoned[index] = false;
					}
					prison2.clear();
				}
			}
		}

		if (success)
		{
			string answer;
			for (char thing : subPermutation)
			{
				if (thing == '0')
				{
					answer.append("1 ");
				}
				else
				{
					answer.append("2 ");
				}
			}
			cout << answer;

			return 0;
		}

		if (subPermutation == end.substr(10 - m, 10))
		{
			break;
		}
	}

	cout << "IMPOSSIBLE";

	return 0;
}

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

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: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 20

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 21

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 22

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 23

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 24

Group: 3, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 25

Group: 3, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 26

Group: 3, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 27

Group: 3, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 28

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 29

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 30

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 31

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 32

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 33

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 34

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 35

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709551126) > this->size() (which is 10)

Test 36

Group: 3, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 37

Group: 3, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 38

Group: 3, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 39

Group: 3, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 40

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 41

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 42

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 43

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 44

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 45

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 46

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)

Test 47

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::out_of_range'
  what():  basic_string::substr: __pos (which is 18446744073709451626) > this->size() (which is 10)