CSES - Datatähti 2025 alku - Results
Submission details
Task:Niitty
Sender:Pikaksi
Submission time:2024-11-06 20:39:20 +0200
Language:C++ (C++20)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
#60
Test results
testverdicttimegroup
#1--1, 2, 3, 4, 5, 6details
#2--1, 2, 3, 4, 5, 6details
#3--1, 2, 3, 4, 5, 6details
#4--1, 2, 3, 4, 5, 6details
#5--1, 2, 3, 4, 5, 6details
#6--2, 3, 4, 5, 6details
#7--2, 3, 4, 5, 6details
#8--2, 3, 4, 5, 6details
#9--2, 3, 4, 5, 6details
#10--3, 4, 5, 6details
#11--3, 4, 5, 6details
#12--3, 4, 5, 6details
#13--3, 4, 5, 6details
#14--4, 5, 6details
#15--4, 5, 6details
#16--4, 5, 6details
#17--4, 5, 6details
#18--5, 6details
#19--5, 6details
#20--5, 6details
#21--5, 6details
#22--6details
#23--6details
#24--6details
#25--6details

Compiler report

input/code.cpp: In function 'll bruteForce(std::vector<std::__cxx11::basic_string<char> >&)':
input/code.cpp:50:36: warning: narrowing conversion of 'x' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   50 |                     Fence fence = {x, y, z, w};
      |                                    ^
input/code.cpp:50:39: warning: narrowing conversion of 'y' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   50 |                     Fence fence = {x, y, z, w};
      |                                       ^
input/code.cpp:50:42: warning: narrowing conversion of 'z' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   50 |                     Fence fence = {x, y, z, w};
      |                                          ^
input/code.cpp:50:45: warning: narrowing conversion of 'w' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   50 |                     Fence fence = {x, y, z, w};
      |...

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int FIELD_SIZE = 500;
const int FIELD_ROW_OFFSET = FIELD_SIZE * (FIELD_SIZE + 1) / 2;
union Fence
{
struct {
uint16_t x;
uint16_t y;
uint16_t sizeX;
uint16_t sizeY;
};
ll bytes;
};
void getFlowerCount(Fence fence, array<int, 26>& flowerCount, vector<vector<char>>& field)
{
//printFence(fence);
for (int y = 0; y < fence.sizeY; y++) {
for (int x = 0; x < fence.sizeX; x++) {
flowerCount[field[fence.y + y][fence.x + x]] += 1;
}
}
}
ll bruteForce(vector<string>& fieldText)
{
int n = fieldText.size();
vector<vector<char>> field = vector<vector<char>>(n, vector<char>(n));
array<bool, 26> containedNumbers = {0};
for (int i = 0; i < n; i++) {
for (int k = 0; k < n; k++) {
int number = fieldText[i][k] - 'A';
field[i][k] = number;
containedNumbers[number] = true;
}
}
ll ans = 0;
for (int x = 0; x < n; x++) {
for (int y = 0; y < n; y++) {
for (int z = 1; z < n - x + 1; z++) {
for (int w = 1; w < n - y + 1; w++) {
Fence fence = {x, y, z, w};
array<int, 26> flowerCount = {0};
getFlowerCount(fence, flowerCount, field);
bool hasAll = true;
for (int i = 0; i < 26; i++) {
if (containedNumbers[i] && flowerCount[i] <= 0) {
hasAll = false;
break;
}
}
if (hasAll) {
ans += 1;
}
}
}
}
}
return ans;
}
void printRows(int n, int* field)
{
for (int y = 0; y < n; y++) {
int rowSpace = y * FIELD_ROW_OFFSET;
int prewOrTowerSpace = rowSpace;
int offsetTriangle = 0;
cout << "row:\n";
for (int x = 0; x < n; x++) {
int orTowerSpace = rowSpace + offsetTriangle;
for (int len = 0; len < n - x; len++) {
cout << field[orTowerSpace + len] << " at " << (orTowerSpace + len) << " ";
}
prewOrTowerSpace = orTowerSpace;
offsetTriangle += n - x;
cout << "\n";
}
}
}
ll calculateAnsBitMasks(int n, int* field, int containedFlowers)
{
ll ans = 0;
if (__builtin_popcount(containedFlowers) == 1) {
ans += n * n;
}
for (int y = 0; y < n; y++) {
int rowSpace = y * FIELD_ROW_OFFSET;
int prewOrTowerSpace = rowSpace;
int offsetTriangle = n;
for (int x = 1; x < n; x++) { // loop or tower spaces
int orTowerSpace = rowSpace + offsetTriangle;
for (int len = 0; len < n - x; len++) { // loop over or tower
int bitMaskCombination = field[prewOrTowerSpace + len] | field[prewOrTowerSpace + len + 1];
field[orTowerSpace + len] = bitMaskCombination;
ans += bitMaskCombination == containedFlowers;
}
prewOrTowerSpace = orTowerSpace;
offsetTriangle += n - x;
}
}
//printRows(n, field);
for (int y = 0; y < n - 1; y++) { // loop rows
int rowSpace = y * FIELD_ROW_OFFSET;
for (int orRowIndex = y + 1; orRowIndex < n; orRowIndex++) { // loop other rows to add to main row
int addRowSpace = orRowIndex * FIELD_ROW_OFFSET;
int offsetTriangle = 0;
for (int x = 0; x < n; x++) { // loop orTowers
int mainOrTowerSpace = rowSpace + offsetTriangle;
int addOrTowerSpace = addRowSpace + offsetTriangle;
for (int len = 0; len < n - x; len++) { // loop orTower bitmasks
field[mainOrTowerSpace + len] |= field[addOrTowerSpace + len];
ans += field[mainOrTowerSpace + len] == containedFlowers;
}
offsetTriangle += n - x;
}
}
}
return ans;
}
void shuffle(int n, vector<string>& testField)
{
testField = vector<string>(n);
for (int i = 0; i < n; i++) {
string row = "";
for (int k = 0; k < n; k++) {
row += rand() % 26 + 'A';
}
//cout << "row = " << row << "\n";
testField[i] = row;
}
}
int main()
{
//vector<string> testField = {"AAAAA", "AAAAA", "AABAA", "AAACA", "AAADA"};
//vector<string> testField = {"BAA", "AAB", "AAA"};
//vector<string> testField = {"Z"};
//vector<string> testField = {"AA", "AA"};
//vector<string> testField = {"AAAC", "ABAA", "ABAA", "AAAA"};
//vector<string> testField = {"AAAAA", "AAAAA", "AAAAA", "AAAAA", "AAAAA"};
//vector<string> testField = {"ZAAAA", "AAAAA", "AAAAA", "AAAAA", "AAAAK"};
//vector<string> testField = {"AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA","AAAAAAAAAA",};
//vector<string> testField = {"AUFCHGK","GXAPVWB","NUHYMEA","HSCSNSI","IMBMTJS","TPVRJUF","SIKDVWY"};
vector<string> testField;
shuffle(300, testField);
auto begin = std::chrono::high_resolution_clock::now();
int n;
n = testField.size();
//cin >> n;
// n * n(n+1) / 2
int* field = new int[FIELD_SIZE * FIELD_ROW_OFFSET]{0};
int containedFlowers = 0;
for (int i = 0; i < n; i++) {
string row;
row = testField[i];
//cout << row << "\n";
//cin >> row;
int rowSpace = i * FIELD_ROW_OFFSET;
for (int k = 0; k < row.size(); k++) {
int bitMask = 1 << (row[k] - 'A');
field[rowSpace + k] = bitMask;
containedFlowers |= bitMask;
}
}
ll ans = calculateAnsBitMasks(n, field, containedFlowers);
cout << ans;
/*vector<string> test;
for (int i = 0; i < n; i++) {
string text;
cin >> text;
test.push_back(text);
}*/
//cout << "\n";
//ll bruteForceAns = bruteForce(testField);
//cout << bruteForceAns;
auto end = std::chrono::high_resolution_clock::now();
auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(end-begin).count();
std::cout << " ms: " << duration << "\n";
}

Test details

Test 1

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

Verdict:

input
10
TNCTNPNTPC
NPPNTNTPTP
NTNTTCNTCT
NPCPNPPNTT
...

correct output
2035

user output
(empty)

Test 2

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

Verdict:

input
10
NFWQLWNWYS
DZOQJVXFPJ
CNHXPXMCQD
QRTBVNLTQC
...

correct output
9

user output
(empty)

Test 3

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

Verdict:

input
10
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
...

correct output
3025

user output
(empty)

Test 4

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

Verdict:

input
10
FFFFFFFFFF
FFFFFCFFFF
FFFFFFJFFF
FFFFFFFFFF
...

correct output
12

user output
(empty)

Test 5

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

Verdict:

input
1
X

correct output
1

user output
(empty)

Test 6

Group: 2, 3, 4, 5, 6

Verdict:

input
20
BBCBUBOUOBBCUUBBCOUO
BOUCOOCUBCOOOCOBOCUO
UCCUUUOBCOCBCBUBUCOO
BUOBUCUCUOOBCOOUBUOO
...

correct output
38724

user output
(empty)

Test 7

Group: 2, 3, 4, 5, 6

Verdict:

input
20
CBGLSHGZHYZDWBNDBJUG
SMUXOJQYPXZDTMJUIWOJ
XIDSTNBGHKRKOVUVMINB
MTQGCFRUHQKALXRNCQGS
...

correct output
8334

user output
(empty)

Test 8

Group: 2, 3, 4, 5, 6

Verdict:

input
20
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
...

correct output
44100

user output
(empty)

Test 9

Group: 2, 3, 4, 5, 6

Verdict:

input
20
AAAAAAAAXAAAAAAAAAAA
AAAWAAAAAAAAAAAAAOAA
AAAAAAAAAAAAAAAAAPAA
AAAAAAAAKAAAAAAAAAAZ
...

correct output
18

user output
(empty)

Test 10

Group: 3, 4, 5, 6

Verdict:

input
50
GRGREEEGREGXRXXEGXXREXGRRRGRRR...

correct output
1584665

user output
(empty)

Test 11

Group: 3, 4, 5, 6

Verdict:

input
50
AITIISJUHCCRZNKSDCNQKYSQRINFWJ...

correct output
1077746

user output
(empty)

Test 12

Group: 3, 4, 5, 6

Verdict:

input
50
OOOOOOOOOOOOOOOOOOOOOOOOOOOOOO...

correct output
1625625

user output
(empty)

Test 13

Group: 3, 4, 5, 6

Verdict:

input
50
FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF...

correct output
1680

user output
(empty)

Test 14

Group: 4, 5, 6

Verdict:

input
100
NNCMDCDDCCNNNDNCMMNCDCDCCDCDNM...

correct output
25325366

user output
(empty)

Test 15

Group: 4, 5, 6

Verdict:

input
100
LIMQQIHASECROEVILNVULGWZJPPKOG...

correct output
22342463

user output
(empty)

Test 16

Group: 4, 5, 6

Verdict:

input
100
TTTTTTTTTTTTTTTTTTTTTTTTTTTTTT...

correct output
25502500

user output
(empty)

Test 17

Group: 4, 5, 6

Verdict:

input
100
QXQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
25650

user output
(empty)

Test 18

Group: 5, 6

Verdict:

input
200
NAANANMMKNKKAKMKMAKNKMNKMMNNAA...

correct output
403292767

user output
(empty)

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:

input
200
LLLLLLLLLLLLLLLLLHLLLLLLLLLLLL...

correct output
14159445

user output
(empty)

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)