Kortit II

input/code.cpp: In function 'int calculate(int, int, int)':
input/code.cpp:9:23: warning: suggest parentheses around comparison in operand of '^' [-Wparentheses]
    9 |         if (n==0 || (a==0 xor b==0)) {
      |                      ~^~~

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

#define out_mod 1000000007

int calculate(int n, int a, int b) {
	int draws = n-a-b;
	//cout << draws;
	if (n==0 || (a==0 xor b==0)) {
		cout << "0\n";
		return 0;
	}
	if (draws < 0) {
		cout << "0\n";
		return 0;
	}
	if (draws == n-1) {
		cout << "0\n";
		return 0;
	}
	
	uint64_t out = 1;
	
	int k = draws;
	int s = n-k;
	
	//n!n!/k!s!
	
	for (int i = min(k,s) + 1; i <= max(k,s); i++) {
		out = (out * i) % out_mod;
	}
	for (int i = max(k,s) + 1; i <= n; i++) {
		out = (out * i * i) % out_mod;
	}
	
	if (min(a,b) != 1) {
		uint64_t temp = 1;
		for (int i = min(a,b) + 1; i <= n - draws; i++) {
			temp = (temp * i) % out_mod;
		}
		temp = temp / 2 + 1;
		
		out = (out * temp) % out_mod;
	}
	cout << out << "\n";
	
	return 0;
}

int main() {
	int t;
	cin >> t;

	for (int i = 0; i < t; i++) {
		int n,a,b;
		cin >> n >> a >> b;
		calculate(n,a,b);
	}
	return 0;
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 3

Group: 3, 4, 5

Verdict: WRONG ANSWER

Test 4

Group: 4, 5

Verdict: WRONG ANSWER

Test 5

Group: 5

Verdict: WRONG ANSWER

Test 6

Group: 5

Verdict: WRONG ANSWER