Kortit II

input/code.cpp:22:780: warning: integer constant is so large that it is unsigned
   22 | constexpr uint64_t bits[64] = { 1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576, 2097152, 4194304, 8388608, 16777216, 33554432, 67108864, 134217728, 268435456, 536870912, 1073741824, 2147483648, 4294967296, 8589934592, 17179869184, 34359738368, 68719476736, 137438953472, 274877906944, 549755813888, 1099511627776, 2199023255552, 4398046511104, 8796093022208, 17592186044416, 35184372088832, 70368744177664, 140737488355328, 281474976710656, 562949953421312, 1125899906842624, 2251799813685248, 4503599627370496, 9007199254740992, 18014398509481984, 36028797018963968, 72057594037927936, 144115188075855872, 288230376151711744, 576460752303423488, 1152921504606846976, 2305843009213693952, 4611686018427387904, 9223372036854775808 };
      |...

#pragma GCC optimize("O3")

#include "bits/stdc++.h"

using namespace std;

#define byte unsigned char

// Yhteisi� yhdelle testille.
int cardCount = 0;
int aTarget = 0;
int bTarget = 0;
int maxZeros = 0;

unordered_map<size_t, int> seenStates;
size_t correctHash = 0;
bool correctHashSet = false;

int CALLS = 0;

// Bit masks
constexpr uint64_t bits[64] = { 1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576, 2097152, 4194304, 8388608, 16777216, 33554432, 67108864, 134217728, 268435456, 536870912, 1073741824, 2147483648, 4294967296, 8589934592, 17179869184, 34359738368, 68719476736, 137438953472, 274877906944, 549755813888, 1099511627776, 2199023255552, 4398046511104, 8796093022208, 17592186044416, 35184372088832, 70368744177664, 140737488355328, 281474976710656, 562949953421312, 1125899906842624, 2251799813685248, 4503599627370496, 9007199254740992, 18014398509481984, 36028797018963968, 72057594037927936, 144115188075855872, 288230376151711744, 576460752303423488, 1152921504606846976, 2305843009213693952, 4611686018427387904, 9223372036854775808 };
constexpr uint64_t rbits[64] = { 18446744073709551614, 18446744073709551613, 18446744073709551611, 18446744073709551607, 18446744073709551599, 18446744073709551583, 18446744073709551551, 18446744073709551487, 18446744073709551359, 18446744073709551103, 18446744073709550591, 18446744073709549567, 18446744073709547519, 18446744073709543423, 18446744073709535231, 18446744073709518847, 18446744073709486079, 18446744073709420543, 18446744073709289471, 18446744073709027327, 18446744073708503039, 18446744073707454463, 18446744073705357311, 18446744073701163007, 18446744073692774399, 18446744073675997183, 18446744073642442751, 18446744073575333887, 18446744073441116159, 18446744073172680703, 18446744072635809791, 18446744071562067967, 18446744069414584319, 18446744065119617023, 18446744056529682431, 18446744039349813247, 18446744004990074879, 18446743936270598143, 18446743798831644671, 18446743523953737727, 18446742974197923839, 18446741874686296063, 18446739675663040511, 18446735277616529407, 18446726481523507199, 18446708889337462783, 18446673704965373951, 18446603336221196287, 18446462598732840959, 18446181123756130303, 18445618173802708991, 18444492273895866367, 18442240474082181119, 18437736874454810623, 18428729675200069631, 18410715276690587647, 18374686479671623679, 18302628885633695743, 18158513697557839871, 17870283321406128127, 17293822569102704639, 16140901064495857663, 13835058055282163711, 9223372036854775807 };
//const uint64_t smasks[64] = { 18446744073709551615, 18446744073709551614, 18446744073709551612, 18446744073709551608, 18446744073709551600, 18446744073709551584, 18446744073709551552, 18446744073709551488, 18446744073709551360, 18446744073709551104, 18446744073709550592, 18446744073709549568, 18446744073709547520, 18446744073709543424, 18446744073709535232, 18446744073709518848, 18446744073709486080, 18446744073709420544, 18446744073709289472, 18446744073709027328, 18446744073708503040, 18446744073707454464, 18446744073705357312, 18446744073701163008, 18446744073692774400, 18446744073675997184, 18446744073642442752, 18446744073575333888, 18446744073441116160, 18446744073172680704, 18446744072635809792, 18446744071562067968, 18446744069414584320, 18446744065119617024, 18446744056529682432, 18446744039349813248, 18446744004990074880, 18446743936270598144, 18446743798831644672, 18446743523953737728, 18446742974197923840, 18446741874686296064, 18446739675663040512, 18446735277616529408, 18446726481523507200, 18446708889337462784, 18446673704965373952, 18446603336221196288, 18446462598732840960, 18446181123756130304, 18445618173802708992, 18444492273895866368, 18442240474082181120, 18437736874454810624, 18428729675200069632, 18410715276690587648, 18374686479671623680, 18302628885633695744, 18158513697557839872, 17870283321406128128, 17293822569102704640, 16140901064495857664, 13835058055282163712, 9223372036854775808 };
//const uint64_t emasks[64] = { 1, 3, 7, 15, 31, 63, 127, 255, 511, 1023, 2047, 4095, 8191, 16383, 32767, 65535, 131071, 262143, 524287, 1048575, 2097151, 4194303, 8388607, 16777215, 33554431, 67108863, 134217727, 268435455, 536870911, 1073741823, 2147483647, 4294967295, 8589934591, 17179869183, 34359738367, 68719476735, 137438953471, 274877906943, 549755813887, 1099511627775, 2199023255551, 4398046511103, 8796093022207, 17592186044415, 35184372088831, 70368744177663, 140737488355327, 281474976710655, 562949953421311, 1125899906842623, 2251799813685247, 4503599627370495, 9007199254740991, 18014398509481983, 36028797018963967, 72057594037927935, 144115188075855871, 288230376151711743, 576460752303423487, 1152921504606846975, 2305843009213693951, 4611686018427387903, 9223372036854775807, 18446744073709551615 };

static void InsertHand(int card, uint64_t* hand)
{
	int index = card / 64;
	int bit = card % 64;

	hand[index] |= bits[bit];
}
void HandToCards(uint64_t* hand, int count, int ret[2000])
{
	int added = 0;

	for (int index = 0; index < ceil(cardCount / 64.f); index++)
	{
		for (int bit = 0; bit < 64; bit++)
		{
			if ((hand[index] & bits[bit]) > 0)
			{
				ret[added] = index * 64 + bit;
				++added;
				if (added == count) 
				{ 
					return;
				}
			}
		}
	}

	return;
}
static void RemoveCard(int card, uint64_t* hand)
{
	hand[(card) / 64] &= rbits[(card) % 64];
}
static void HandToString(uint64_t* hand, string& target)
{
	for (int i = 0; i < 32; ++i)
	{
		target += hand[i];
	}
}
static void HashHand(uint64_t* hand, size_t& target)
{
	std::hash<int> hasher;
	for (int i = 0; i < ceil(cardCount / 64.f); ++i)
	{
		target ^= hasher(hand[i]) + 0x9e3779b9 + (target << 6) + (target >> 2);
	}
}

size_t HashState(uint64_t* aHand, uint64_t* bHand, int& aScore, int& bScore, int& ties)
{
	/*
	string preHash = "";
	preHash.reserve((cardCount / 64 + 1) * 20 + 12);

	HandToString(aHand, preHash);
	HandToString(bHand, preHash);

	preHash += aScore * 10000 + bScore;
	preHash += ties;

	std::hash<string> hasher;

	return hasher(preHash);*/

	std::hash<int> hasher;

	size_t hash = hasher(ties);

	hash ^= hasher(aScore) + 0x9e3779b9 + (hash << 6) + (hash >> 2);
	hash ^= hasher(bScore) + 0x9e3779b9 + (hash << 6) + (hash >> 2);
	HashHand(aHand, hash);
	HashHand(bHand, hash);

	return hash;
}
size_t HashTargets()
{
	uint64_t preHash = cardCount * 100000000 + aTarget * 10000 + bTarget;

	std::hash<uint64_t> hasher;

	return hasher(preHash);
}
void AddHash(size_t& hash, int& value)
{
	if (correctHashSet && hash == correctHash)
	{
		return;
	}
	seenStates.insert(pair<size_t, int>(hash, value));
}

static int TestBranch(uint64_t* aHand, uint64_t* bHand, int aScore, int bScore, int ties)
{
	CALLS++;

	int remCards = cardCount - aScore - bScore - ties;

	if (!(remCards > 0))
	{
		if (aScore == aTarget && bScore == bTarget && ties == maxZeros)
		{
			if (!correctHashSet)
			{
				correctHash = HashState(aHand, bHand, aScore, bScore, ties);
				correctHashSet = true;
				seenStates.erase(correctHash);
			}
			return 1;
		}
		else
		{
			return 0;
		}
	}
	if (aScore > aTarget || bScore > bTarget || ties > maxZeros)
	{
		return 0;
	}
	if (aScore + remCards < aTarget || bScore + remCards < bTarget)
	{
		return 0;
	}

	int aCards[2000];
	HandToCards(aHand, remCards, aCards);
	int bCards[2000];
	HandToCards(bHand, remCards, bCards);


	int childTotal = 0;

	for (int i = 0; i < remCards; i++)
	{
		//uint64_t newAHand[32];
		//copy(aHand, aHand + 32, newAHand);
		//RemoveCard(aCards[i], newAHand);
		int removedACard = aCards[i];
		RemoveCard(removedACard, aHand);
		for (int j = 0; j < remCards; j++)
		{
			//uint64_t newBHand[32];
			//copy(bHand, bHand + 32, newBHand);
			//RemoveCard(bCards[j], newBHand);

			int removedBCard = bCards[j];
			RemoveCard(removedBCard, bHand);

			int newScoreA = aScore;
			int newScoreB = bScore;
			int newTies = ties;

			if (removedACard == removedBCard)
			{
				newTies++;
			}
			else if (removedACard > removedBCard)
			{
				newScoreA++;
			}
			else
			{
				newScoreB++;
			}

			size_t hash = HashState(aHand, bHand, newScoreA, newScoreB, newTies);

			if (seenStates.contains(hash))
			{
				childTotal += seenStates[hash];
			}
			else
			{
				int ret = TestBranch(aHand, bHand, newScoreA, newScoreB, newTies);
				AddHash(hash, ret);
				childTotal += ret;
			}
			childTotal %= 1000000007;
			InsertHand(removedBCard, bHand);
		}
		InsertHand(removedACard, aHand);
	}
	return childTotal % 1000000007;
}

static void Test(int n, int a, int b)
{
	if (a + b > n)
	{
		cout << "0";
		return;
	}

	cardCount = n;
	aTarget = a;
	bTarget = b;
	maxZeros = n - a - b;

	seenStates.clear();

	uint64_t baseHandA[32] = { 0 };
	uint64_t baseHandB[32] = { 0 };

	for (int i = 0; i < n; i++)
	{
		InsertHand(i + 1, baseHandA);
		InsertHand(i + 1, baseHandB);
	}

	cout << TestBranch(baseHandA, baseHandB, 0, 0, 0);
}

int main()
{
	int testCount = 0;
	cin >> testCount;

	cout << "";

	for (int i = 0; i < testCount; i++)
	{
		int n, a, b;
		cin >> n >> a >> b;

		Test(n, a, b);

		if (i < testCount - 1)
		{
			cout << "\n";
		}
	}
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

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