CSES - Datatähti 2025 alku - Results
Submission details
Task:Niitty
Sender:Pikaksi
Submission time:2024-11-06 18:54:32 +0200
Language:C++ (C++20)
Status:READY
Result:58
Feedback
groupverdictscore
#1ACCEPTED4
#2ACCEPTED6
#3ACCEPTED10
#4ACCEPTED13
#5ACCEPTED25
#60
Test results
testverdicttimegroup
#1ACCEPTED0.19 s1, 2, 3, 4, 5, 6details
#2ACCEPTED0.18 s1, 2, 3, 4, 5, 6details
#3ACCEPTED0.18 s1, 2, 3, 4, 5, 6details
#4ACCEPTED0.18 s1, 2, 3, 4, 5, 6details
#5ACCEPTED0.18 s1, 2, 3, 4, 5, 6details
#6ACCEPTED0.19 s2, 3, 4, 5, 6details
#7ACCEPTED0.18 s2, 3, 4, 5, 6details
#8ACCEPTED0.18 s2, 3, 4, 5, 6details
#9ACCEPTED0.19 s2, 3, 4, 5, 6details
#10ACCEPTED0.18 s3, 4, 5, 6details
#11ACCEPTED0.19 s3, 4, 5, 6details
#12ACCEPTED0.19 s3, 4, 5, 6details
#13ACCEPTED0.19 s3, 4, 5, 6details
#14ACCEPTED0.22 s4, 5, 6details
#15ACCEPTED0.22 s4, 5, 6details
#16ACCEPTED0.22 s4, 5, 6details
#17ACCEPTED0.22 s4, 5, 6details
#18ACCEPTED0.69 s5, 6details
#19ACCEPTED0.69 s5, 6details
#20ACCEPTED0.69 s5, 6details
#21ACCEPTED0.69 s5, 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:56:36: warning: narrowing conversion of 'x' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   56 |                     Fence fence = {x, y, z, w};
      |                                    ^
input/code.cpp:56:39: warning: narrowing conversion of 'y' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   56 |                     Fence fence = {x, y, z, w};
      |                                       ^
input/code.cpp:56:42: warning: narrowing conversion of 'z' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   56 |                     Fence fence = {x, y, z, w};
      |                                          ^
input/code.cpp:56:45: warning: narrowing conversion of 'w' from 'int' to 'uint16_t' {aka 'short unsigned int'} [-Wnarrowing]
   56 |                     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;

int containedFlowers = 0;





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)
{
    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;

                if (bitMaskCombination == containedFlowers) {
                    ans += 1;
                }
            }

            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];

                    if (field[mainOrTowerSpace + len] == containedFlowers) {
                        ans += 1;
                    }
                }

                offsetTriangle += n - x;
            }
        }
    }

    return ans;
}

 
void shuffle(int n, vector<string> testField)
{
    for (int i = 0; i < n; i++) {
        for (int k = 0; k < n; k++) {
            testField[i][k] = rand() % 26 + 'A';
        }
    }
}


 
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"};

    int n;
    //n = testField.size();
    //shuffle(n, testField);
    cin >> n;
    //auto begin = std::chrono::high_resolution_clock::now();
    
    // n * n(n+1) / 2
    int* field = new int[FIELD_SIZE * FIELD_ROW_OFFSET]{0};

    for (int i = 0; i < n; i++) {
        string row;
        //row = testField[i];
        //cout << row << "\n";
        cin >> row;

        for (int k = 0; k < row.size(); k++) {

            int bitMask = 1 << (row[k] - 'A');
            field[i * FIELD_ROW_OFFSET + k] = bitMask;
            containedFlowers |= bitMask;
        }
    }
    ll ans = calculateAnsBitMasks(n, field);
    cout << ans;

    /*vector<string> test;
    for (int i = 0; i < n; i++) {
        string text;
        cin >> text;
        test.push_back(text);
    }*/
    //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: 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: ACCEPTED

input
200
OMYWATTLURKQPTKEFMGGYAOONXWVSC...

correct output
388111321

user output
388111321

Test 20

Group: 5, 6

Verdict: ACCEPTED

input
200
CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC...

correct output
404010000

user output
404010000

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)