Kortit II

input/code.cpp: In function 'int main()':
input/code.cpp:20:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |                         if (j >= s.size()) {
      |                             ~~^~~~~~~~~~~

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main(){
    int n, a, b; cin >> n >> a >> b;

    int m = max(a, b);
    int w = min(a, b);

    vector<int> s;
    for (int i = 0; i < n; i++) s.push_back(i + 1);

	int c = 0;

    do {
        int t_1 = 0, t_2 = 0;
		for (int j = 0; j < n; j++) {
			if (j >= s.size()) {
				t_1 = -1;
				break;
			}
			if (s[j] < j + 1)
				t_1++;
			else if (s[j] > j + 1)
				t_2++;
		}

		if (t_2 == m && t_1 == w) {
			c++;
		}
    } while (next_permutation(s.begin(), s.end()));

	long long mod = 1000000007;
	long long r = c;
	for (int i = 1; i < n + 1; i++) {
		r = (r * i) % mod;
	}
    cout << r << '\n';
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 2

Group: 2, 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 5

Verdict: TIME LIMIT EXCEEDED