CSES - Datatähti 2025 alku - Results
Submission details
Task:Niitty
Sender:DLPS
Submission time:2024-11-04 16:35:44 +0200
Language:C++ (C++20)
Status:READY
Result:33
Feedback
groupverdictscore
#1ACCEPTED4
#2ACCEPTED6
#3ACCEPTED10
#4ACCEPTED13
#50
#60
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 3, 4, 5, 6details
#2ACCEPTED0.01 s1, 2, 3, 4, 5, 6details
#3ACCEPTED0.01 s1, 2, 3, 4, 5, 6details
#4ACCEPTED0.01 s1, 2, 3, 4, 5, 6details
#5ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#6ACCEPTED0.01 s2, 3, 4, 5, 6details
#7ACCEPTED0.01 s2, 3, 4, 5, 6details
#8ACCEPTED0.01 s2, 3, 4, 5, 6details
#9ACCEPTED0.01 s2, 3, 4, 5, 6details
#10ACCEPTED0.02 s3, 4, 5, 6details
#11ACCEPTED0.02 s3, 4, 5, 6details
#12ACCEPTED0.02 s3, 4, 5, 6details
#13ACCEPTED0.01 s3, 4, 5, 6details
#14ACCEPTED0.12 s4, 5, 6details
#15ACCEPTED0.21 s4, 5, 6details
#16ACCEPTED0.13 s4, 5, 6details
#17ACCEPTED0.02 s4, 5, 6details
#18ACCEPTED0.99 s5, 6details
#19--5, 6details
#20--5, 6details
#21ACCEPTED0.14 s5, 6details
#22--6details
#23--6details
#24--6details
#25--6details

Compiler report

input/code.cpp:3:25: warning: extra tokens at end of #include directive
    3 | #include "bits/stdc++.h";
      |                         ^
input/code.cpp:4:25: warning: extra tokens at end of #include directive
    4 | #include "unordered_map";
      |                         ^
input/code.cpp: In function 'int main()':
input/code.cpp:189:35: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  189 |                 for (int x = 0; x < row.size(); ++x)
      |                                 ~~^~~~~~~~~~~~

Code

#pragma GCC optimize("O3")

#include "bits/stdc++.h";
#include "unordered_map";

using namespace std;

#define byte unsigned char

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

int preCalcMap[26][4004001];
int flowerCounts[26] = { 0 };

// Debug
int CALLS = 0;


int fieldSize = 0;

struct flowerData
{
	uint64_t rows[500][8] = { {0} };

	int flowerCount = 0;

	void Insert(uint16_t x, uint16_t y)
	{
		++flowerCount;

		uint64_t(*row)[8] = &rows[y];

		(*row)[int(ceil(x / 64))] |= bits[x % 64];
	}

	bool InBox(int left, int right, int top, int bottom)
	{
		for (int y = top; y <= bottom; ++y)
		{
			uint64_t(*row)[8] = &rows[y];

			int minByte = left / 64;
			int minByteStart = left % 64;
			int maxByte = right / 64;
			int maxByteEnd = right % 64;

			if (minByte == maxByte)
			{
				// Masked from left and right
				if (((*row)[minByte] & smasks[minByteStart] & emasks[maxByteEnd]) > 0)
				{
					return true;
				}
			}
			else
			{
				// Left
				if (((*row)[minByte] & smasks[minByteStart]) > 0)
				{
					return true;
				}
				// Center
				for (int i = minByte + 1; i < maxByte; ++i)
				{
					if ((*row)[i] > 0)
					{
						return true;
					}
				}
				// Right
				if (((*row)[maxByte] & emasks[maxByteEnd]) > 0)
				{
					return true;
				}
			}
		}

		return false;
	}
};

const unordered_map<char, int> flowerIndecies = { {'A',0},{'B',1},{'C',2},{'D',3},{'E',4},{'F',5},{'G',6},{'H',7},{'I',8},{'J',9},{'K',10},{'L',11},{'M',12},{'N',13},{'O',14},{'P',15},{'Q',16},{'R',17},{'S',18},{'T',19},{'U',20},{'V',21},{'W',22},{'X',23},{'Y',24},{'Z',25} };
flowerData differentFlowers[26] = {};

static bool ContainsAllFlowers(int x1, int x2, int y1, int y2)
{
	++CALLS;

	int i1 = (y2 + 1) * 2001 + x2 + 1;
	int i2 = y1 * 2001 + x2 + 1;
	int i3 = (y2 + 1) * 2001 + x1;
	int i4 = y1 * 2001 + x1;

	for (int i = 0; i < 26; ++i)
	{
		if (flowerCounts[i] == 0)
		{
			continue;
		}
		
		if (preCalcMap[i][i1] - preCalcMap[i][i2] - preCalcMap[i][i3] + preCalcMap[i][i4] == 0)
		{
			return false;
		}
	}

	return true;
}

int lastSolution = -1;

static int Y2Solver(int min, int max, int x1, int x2, int y1)
{
	// Esimerkiksi: min: 12 ja max: 25
	// T�ll�in current = 12 + (13) / 2 = 18

	if (lastSolution != -1 && ContainsAllFlowers(x1, x2, y1, lastSolution))
	{
		return (max - lastSolution + 1);
	}

	int localMin = min;
	int localMax = max;

	int current = (min + max) / 2;

	bool first = true;
	int change = 0;
	bool wasSuccess = false;

	while (true)
	{
		if (ContainsAllFlowers(x1, x2, y1, current))
		{
			// Nousi yhdell� ja ei sis�lt�nyt ennen.
			if (change == 1 && !wasSuccess)
			{
				break;
			}
			if (change == 0 && wasSuccess)
			{
				break;
			}

			localMax = current;

			change = (localMin + current) / 2 - current;
			current += change;
			wasSuccess = true;
		}
		else
		{
			// laski yhdell� ja nyt ei sis�ll�.
			if (change == -1 && wasSuccess)
			{
				// Siirret��n takaisin oikeaan.
				++current;
				break;
			}
			if (change == 0 && !first)
			{
				return 0;
			}


			localMin = current;
			change = ceil((current + localMax) / 2.0f) - current;
			current += change;
			wasSuccess = false;
		}
		first = false;
	}
	lastSolution = current;
	return (max - current + 1);
}

int main()
{
	cin >> fieldSize;

	for (int y = 0; y < fieldSize; ++y)
	{
		string row;
		cin >> row;

		for (int x = 0; x < row.size(); ++x)
		{
			int flowerType = flowerIndecies.at(row[x]);
			++flowerCounts[flowerType];

			int curI = (y + 1) * 2001 + x + 1;
			int lI = (y + 1) * 2001 + x;
			int tI = y * 2001 + x + 1;
			int tlI = y * 2001 + x;

			for (int i = 0; i < 26; ++i)
			{
				preCalcMap[i][curI] = preCalcMap[i][lI] + preCalcMap[i][tI] - preCalcMap[i][tlI] + (flowerType == i);
			}
		}
	}

	int count = 0;


	for (int x1 = 0; x1 < fieldSize; ++x1)
	{
		bool hadSuccessX2 = false;
		for (int x2 = fieldSize - 1; x2 >= x1; x2--)
		{
			lastSolution = -1;
			bool hadSuccessY1 = false;
			for (int y1 = 0; y1 < fieldSize; ++y1)
			{
				int ans = Y2Solver(y1, fieldSize - 1, x1, x2, y1);
				if (ans == 0)
				{
					break;
				}
				else
				{
					count += ans;
					hadSuccessY1 = true;
				}
			}
			if (!hadSuccessY1)
			{
				break;
			}
			else
			{
				hadSuccessX2 = true;
			}
		}
		if (!hadSuccessX2)
		{
			break;
		}
	}


	cout << count;
}

Test details

Test 1

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
TNCTNPNTPC
NPPNTNTPTP
NTNTTCNTCT
NPCPNPPNTT
...

correct output
2035

user output
2035

Test 2

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
NFWQLWNWYS
DZOQJVXFPJ
CNHXPXMCQD
QRTBVNLTQC
...

correct output
9

user output
9

Test 3

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
...

correct output
3025

user output
3025

Test 4

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
FFFFFFFFFF
FFFFFCFFFF
FFFFFFJFFF
FFFFFFFFFF
...

correct output
12

user output
12

Test 5

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
1
X

correct output
1

user output
1

Test 6

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
BBCBUBOUOBBCUUBBCOUO
BOUCOOCUBCOOOCOBOCUO
UCCUUUOBCOCBCBUBUCOO
BUOBUCUCUOOBCOOUBUOO
...

correct output
38724

user output
38724

Test 7

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
CBGLSHGZHYZDWBNDBJUG
SMUXOJQYPXZDTMJUIWOJ
XIDSTNBGHKRKOVUVMINB
MTQGCFRUHQKALXRNCQGS
...

correct output
8334

user output
8334

Test 8

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
...

correct output
44100

user output
44100

Test 9

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
AAAAAAAAXAAAAAAAAAAA
AAAWAAAAAAAAAAAAAOAA
AAAAAAAAAAAAAAAAAPAA
AAAAAAAAKAAAAAAAAAAZ
...

correct output
18

user output
18

Test 10

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
GRGREEEGREGXRXXEGXXREXGRRRGRRR...

correct output
1584665

user output
1584665

Test 11

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
AITIISJUHCCRZNKSDCNQKYSQRINFWJ...

correct output
1077746

user output
1077746

Test 12

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
OOOOOOOOOOOOOOOOOOOOOOOOOOOOOO...

correct output
1625625

user output
1625625

Test 13

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF...

correct output
1680

user output
1680

Test 14

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
NNCMDCDDCCNNNDNCMMNCDCDCCDCDNM...

correct output
25325366

user output
25325366

Test 15

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
LIMQQIHASECROEVILNVULGWZJPPKOG...

correct output
22342463

user output
22342463

Test 16

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
TTTTTTTTTTTTTTTTTTTTTTTTTTTTTT...

correct output
25502500

user output
25502500

Test 17

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
QXQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
25650

user output
25650

Test 18

Group: 5, 6

Verdict: ACCEPTED

input
200
NAANANMMKNKKAKMKMAKNKMNKMMNNAA...

correct output
403292767

user output
403292767

Test 19

Group: 5, 6

Verdict:

input
200
OMYWATTLURKQPTKEFMGGYAOONXWVSC...

correct output
388111321

user output
(empty)

Test 20

Group: 5, 6

Verdict:

input
200
CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC...

correct output
404010000

user output
(empty)

Test 21

Group: 5, 6

Verdict: ACCEPTED

input
200
LLLLLLLLLLLLLLLLLHLLLLLLLLLLLL...

correct output
14159445

user output
14159445

Test 22

Group: 6

Verdict:

input
500
VVHWVUHVHUWWWVUUUWVUUHUUWHWUVW...

correct output
15683003812

user output
(empty)

Test 23

Group: 6

Verdict:

input
500
OIMZGEQSBMBDSDXSWRFNKSGFEBBTJE...

correct output
15575906951

user output
(empty)

Test 24

Group: 6

Verdict:

input
500
IIIIIIIIIIIIIIIIIIIIIIIIIIIIII...

correct output
15687562500

user output
(empty)

Test 25

Group: 6

Verdict:

input
500
WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW...

correct output
3058970930

user output
(empty)