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

Code

#include <iostream>
#include <vector>
#include <set>

using namespace std;

typedef vector<int> v1i;
typedef vector<v1i> v2i;
typedef vector<v2i> v3i;

set<int> s;

v3i t(501, v2i(501, v1i(26)));

void initPrefixSum(int n)
{
  for (int l = 0; l < 26; l++)
  {
    for (int j = 1; j <= n; j++)
    {
      for (int i = 1; i <= n; i++)
      {
        t[i][j][l] += t[i-1][j][l];
      }
      for (int i = 1; i <= n; i++)
      {
        t[i][j][l] += t[i][j-1][l];
      }
    }
  }
}

int sum(int l, int sx, int sy, int ex, int ey)
{
  return t[ex][ey][l] - t[sx][ey][l] - t[ex][sy][l] + t[sx][sy][l];
}

bool containsAllSpecies(int sx, int sy, int ex, int ey)
{
  for (int species : s)
  {
    if (sum(species, sx, sy, ex, ey) == 0) return false;
  }
  return true;
}

int solve(int n)
{
  int solutions = 0;
  for (int sx = 0; sx < n; sx++)
  for (int sy = 0; sy < n; sy++)
  {
    for (int ex = sx + 1; ex <= n; ex++)
    {
      int a = sy + 1;
      int b = n;
      while (a <= b)
      {
        int k = (a + b) / 2;

        bool v1 = containsAllSpecies(sx, sy, ex, k);
        bool v2 = containsAllSpecies(sx, sy, ex, k - 1);
        if (v1 == true && v2 == false)
        {
          solutions += n - k + 1;
          break;
        }
        if (v1) b = k - 1;
        else a = k + 1;
      }
    }
  }
  return solutions;
}

int main()
{
  int n;

  cin >> n;

  for (int j = 1; j <= n; j++)
  for (int i = 1; i <= n; i++)
  {
    char c;

    cin >> c;

    s.insert(c - 65);

    t[i][j][c - 65] = 1;
  }

  initPrefixSum(n);

  cout << solve(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:

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: ACCEPTED

input
200
CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC...

correct output
404010000

user output
404010000

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)