Kortit II

#include <algorithm>
#include <bits/stdc++.h>

using namespace std;

long mod = 1000000000 + 7;

int fc(int factor) { //factorial
  int answer = 1;
  for (long i = 1; i <= factor; i++) {
    answer *= i;
  }
  return answer;
}

long fcm(int factor) { //factorial
  long answer = 1;
  for (long i = 1; i <= factor; i++) {
    answer = (answer % mod * i % mod) % mod;
  }
  return answer;
}

stack<long> createNumberStack(int n) {
  stack<long> answer = {};
  for (long i = 1; i <= n; i++) {
    answer.push(i);
  }
  return answer;
}
vector<long> createNumberLine(int n) {
  vector<long> answer = {};
  for (long i = 1; i <= n; i++) {
    answer.push_back(i);
  }
  return answer;
}

vector<vector<long>> testInput(int amt) {
  vector<vector<long>> answer = {};
  int a = 1;
  int b = 1;
  int n = 1;
  for (int i = 1; i <= amt*amt*amt; i++) {
    if (a % amt == 1 && a != 1) {
      a = 1;
      b++;
      if (b % amt == 1 && b != 1) {
      b = 1;
      n++;
      }
    }
    answer.push_back({n, a, b});
    cout << n << " " << a << " " << b << "\n";
    a++;
  }
  return answer;
}

string printFactor(int n) {
  string answer = "1";
  for (int i = 2 ; i <= n; i++) {
    answer += "*" + to_string(i);
  }
  return answer;
}  

void printArray(vector<long> a) {
  for (int i = 0; i < (int)a.size(); i++) {
    cout << a[i] << " ";
  }
  cout << "\n";
}


long sumOfSums(long d, long n) {
  return (fcm(n)%mod)/((fcm(d)%mod)*(fcm(n-d)%mod));
}

long bruteForce(long n, long a, long b) {
  long d = n-(a+b); //Tasapelit

  long n2 = n-d;
  long p = 0;

  if (a+b <= n && (abs(a-b) != a+b || a+b == 0)) {
    vector<long> l1 = createNumberLine(n2);
    for (long j = 1; j <= fc(n2); j++) {
      long a1 = 0;
      long b1 = 0;
      for (long k = 0; k < (long)l1.size(); k++) {
        if (l1[k] > k+1) {
          b1++;
        } else if (l1[k] < k+1) {
          a1++;
        }
      }
      if (a1 == a && b1 == b) {
        p++;
      }
      next_permutation(l1.begin(),l1.end());
    }
    return (( p % mod ) * ( fcm(n2) % mod ) * ( sumOfSums(d, n) % mod ) * ( sumOfSums(d, n) % mod ) * ( fcm(d) % mod ) ) % mod;
  }
  return 0;
}

string isCorrect(long answer, int n, int a, int b) {
  long bf = bruteForce(n, a, b);
  if (answer == bf) {
    return "true " + to_string(bf);
  } else {
    return "false A: " + to_string(answer) + " | BF: " + to_string(bf) + " " + to_string(n) + " " + to_string(a) + " " + to_string(b);
  }
}

long sumOfSums2(long d, long n) {
  return (fc(n))/((fc(d))*(fc(n-d)));
}

//Current iteration = a+b, koska tasapelejä ei lasketa.
long iterateGame(vector<long> nl, long a, long b, long a1, long b1) {
  //printArray(nl);
  long i = a1+b1+1;
  long p = 0;
  int size = (int)nl.size();
  for (int j = 0; j < size; j++) {
    long a2 = 0;
    long b2 = 0;
    long num = nl[j];
    bool work = true;
    if (i > num) {
      a2++;
    } else if (i < num) {
      b2++;
    } else { //Tasapeli, ei mahdollista.
      work = false;
    }
    if (a1+a2 > a || b1+b2 > b) {
      work = false;
      //cout << "False permutation, no more solutions." << " i: " << i << " num: " << num << endl;
    } else if (size == 1 && (a == a1+a2 && b == b1+b2)) {
      //cout << "Permutation found" << " i: " << i << " num: " << num << endl;
      return 1;
      //p += 1;
    }
    //cout << "i: " << i << " a: " << a1 << " b: " << b1 << " num: " << num << "\n";
    if (work) {
      //cout << "Continue." << " i: " << i << " num: " << num << endl;
      vector<long> nnl = nl;
      nnl.erase(nnl.begin() + j);
      p += iterateGame(nnl, a, b, a1+a2, b1+b2);
    }
  }
  return p;
}

int main() {
  int t;
  vector<vector<long>> g = {};//testInput(8);
  cin >> t;
  for (int i = 0; i < t; i++) {
    long n, a, b;
    cin >> n >> a >> b;
    g.push_back({n, a, b});
  }

  for (int i = 0; i < (int)g.size(); i++) {
    long n = g[i][0];
    long a = g[i][1];
    long b = g[i][2];
    long d = n-(a+b); //Tasapelit

    long n2 = n-d; //Uusi n
    long p = 0;

    if (a+b <= n && (abs(a-b) != a+b || a+b == 0)) {
      vector<long> l1 = createNumberLine(n2);
      p = iterateGame(l1, a, b, 0, 0);
      /*cout << "- - -" << endl;
      cout << p << endl;
      cout << bruteForce(n, a, b) << endl;
      cout << "- - -" << endl;*/
      long vastaus = ((p%mod) * (fcm(n2)%mod) * (sumOfSums(d, n)%mod) * (sumOfSums(d, n)%mod) * (fcm(d)%mod))%mod;
      //long vastaus = ((p) * (fc(n2)) * (sumOfSums2(d, n)) * (sumOfSums2(d, n)) * (fc(d)));
      //cout << isCorrect(vastaus, n, a, b) << "\n";//<< " | " << n << " " << a <<  " " << b << "\n";
      //cout << isCorrect(vastaus, n, a, b) << "\n";
      cout << vastaus << "\n";
      //v.push_back(vastaus);

    } else {
      //cout << isCorrect(0, n, a, b) << "\n";
      cout << 0 << "\n";
      //v.push_back(0);
    }
  }
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 2

Group: 2, 3, 4, 5

Verdict: WRONG ANSWER

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