Kortit II

input/code.cpp: In function 'long int solve(int, int, int)':
input/code.cpp:42:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   42 |   for (int i = 0; i < p.size(); i += n)
      |                   ~~^~~~~~~~~~

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> findPermutations(int n)
{
  int v[n];

  for (int i = 0; i < n; i++)
  {
    v[i] = i + 1;
  }

  vector<int> p;

  do
  {
    for (int i = 0; i < n; i++)
    {
      p.push_back(v[i]);
    }
  } while (next_permutation(v, v + n));

  return p;
}

long factorial(long x)
{
  long res = 1;
  for (long i = 2; i < x + 1; i++) res = (res * i) % 1000000007;
  return res;
}

long solve(int n, int a, int b)
{
  vector<int> p = findPermutations(n);

  long sum = 0;

  for (int i = 0; i < p.size(); i += n)
  {
    int asum = 0;
    int bsum = 0;
    for (int j = 0; j < n; j++)
    {
      if (p[i + j] > j + 1) asum++;
      else if (p[i + j] < j + 1) bsum++;
    }
    if (asum == a && bsum == b) sum = (sum + 1) % 1000000007;
  }

  return sum;
}

int main()
{
  int t;

  cin >> t;

  for (int i = 0; i < t; i++)
  {
    int n, a, b;

    cin >> n >> a >> b;

    cout << solve(n, a, b) * factorial(n) % 1000000007 << endl;
  }
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 3, 4, 5

Verdict: ACCEPTED

Test 3

Group: 3, 4, 5

Verdict: RUNTIME ERROR

Test 4

Group: 4, 5

Verdict: RUNTIME ERROR

Test 5

Group: 5

Verdict: RUNTIME ERROR

Test 6

Group: 5

Verdict: RUNTIME ERROR