CSES - Datatähti 2019 alku - Results
Submission details
Task:Ruudukko
Sender:jarvenp
Submission time:2018-10-07 14:55:01 +0300
Language:C++
Status:READY
Result:14
Feedback
groupverdictscore
#10
#2ACCEPTED14
#30
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1details
#2ACCEPTED0.02 s1details
#3ACCEPTED0.01 s1details
#4ACCEPTED0.03 s1details
#5ACCEPTED0.01 s1details
#6ACCEPTED0.01 s1details
#7ACCEPTED0.03 s1details
#80.01 s1details
#90.02 s1details
#100.01 s1details
#11ACCEPTED0.02 s2details
#12ACCEPTED0.02 s2details
#13ACCEPTED0.03 s2details
#14ACCEPTED0.06 s2details
#15ACCEPTED0.17 s2details
#16ACCEPTED0.34 s2details
#17ACCEPTED0.47 s2details
#18ACCEPTED0.47 s2details
#19ACCEPTED0.48 s2details
#20ACCEPTED0.49 s2details
#210.03 s3details
#220.03 s3details
#230.03 s3details
#240.02 s3details
#250.03 s3details
#260.04 s3details
#270.02 s3details
#280.03 s3details
#290.03 s3details
#300.02 s3details

Code

#include <bits/stdc++.h>
#include <stdint.h>
using namespace std;
uint32_t n;
uint64_t counter;
uint32_t taikaluvut[501];
const uint64_t M = 1e9+7;
/*
void taulukot(uint32_t layer, char taul[][500], char goal[][500], uint16_t aUsed, uint16_t bUsed){
if(layer == n){
bool match = true;
for(uint32_t i=0; i<n; i++){
// cout << taul[i]<<"\n";
for(uint32_t i2=0; i2<n; i2++){
if((taul[i][i2] != goal[i][i2]) && (goal[i][i2] != '.')){
match = false;
break;
}
}
if(!match){
break;
}
}
if(match){
counter++;
}
// cout << "\n";
}
else{
char temp[500][500];
for(uint32_t i=0; i<n; i++){
for(uint32_t i2=0; i2<n; i2++){
temp[i][i2] = taul[i][i2];
}
temp[i][n] = 0;
}
for(uint32_t i=0; i<n; i++){
//cout <<"b\n";
if(((aUsed>>i)&1) == 0){
//cout << "a\n";
aUsed = aUsed | (1<<i);
temp[layer][i] = 'A';
for(uint32_t i2=0; i2<n; i2++){
if((((bUsed>>i2)&1) == 0) && (i2 != i)){
// cout << i<<" "<<temp[layer][i]<<"\n";
bUsed = bUsed | (1<<i2);
temp[layer][i2] = 'B';
taulukot(layer+1, temp, goal, aUsed, bUsed);
bUsed = bUsed & ~(1<<i2);
temp[layer][i2] = '.';
}
}
aUsed = aUsed & ~(1<<i);
temp[layer][i] = '.';
}
}
}
}
*/
uint64_t fact(uint32_t val){
uint64_t ans=1;
for(uint32_t i=1; i<=val; i++){
ans *= i;
ans %=M;
}
return ans;
}
uint64_t perm(uint32_t howMany, uint32_t size){
if(howMany > size){
return 0;
}
return((fact(size)/(fact(howMany)*fact(size-howMany)))%M);
}
uint64_t solutions(uint32_t size, uint32_t blocks){
uint64_t ans = (uint64_t)taikaluvut[size];
for(uint32_t i=1; i <= (size-blocks); i++){
ans += (perm(i,(size-blocks))*taikaluvut[size-i])%M;
ans %= M;
}
return ans;
}
int main(){
taikaluvut[0] = 1;
taikaluvut[1] = 0;
taikaluvut[2] = 1;
for(uint32_t i=3; i<501; i++){
taikaluvut[i] = (((taikaluvut[i-1]+taikaluvut[i-2])%M)*(i-1))%M;
}
cin >> n;
char goal[500][500];
for(uint32_t i=0; i<n; i++){
cin >> goal[i];
}
/*
if(!empty){
char taul[500][500];
for(uint32_t i=0; i<n; i++){
for(uint32_t i2=0; i2<n; i2++){
taul[i][i2] = '.';
}
taul[i][n] = 0;
}
counter = 0;
taulukot(0,taul,goal,0,0);
cout << counter<<"\n";
}
else{
cout << (fact(n)*(uint64_t)taikaluvut[n])%M<<"\n";
}
*/
uint16_t aAmount = 0;
for(uint16_t i=0; i<n; i++){
for(uint16_t i2=0; i2<n; i2++){
if(goal[i][i2] == 'A'){
aAmount++;
}
}
}
uint16_t xSize = n-aAmount;
uint16_t ySize = n-aAmount;
uint16_t aSize = n-aAmount;
uint16_t extraB = 0;
uint16_t taulSize = n;
for(uint16_t i=0; i<n; i++){
for(uint16_t i2=0; i2<n; i2++){
if(goal[i][i2] == 'B'){
bool aX = false;
bool aY = false;
for(uint16_t i3=0; i3<n; i3++){
if(goal[i][i3] == 'A'){
aX = true;
break;
}
}
for(uint16_t i3=0; i3<n; i3++){
if(goal[i3][i2] == 'A'){
aY = true;
break;
}
}
taulSize--;
if(aX && aY){
aAmount -= 2;
}
else if(aX){
aAmount--;
ySize--;
}
else if(aY){
aAmount--;
xSize--;
}
else if(!aY && !aX){
xSize--;
ySize--;
extraB++;
}
}
}
}
//cout << aAmount<<" "<< xSize<<" " << ySize<<" "<<aSize<<" "<<extraB<<" "<<taulSize<<"\n";
if(xSize < ySize){
uint16_t temp = xSize;
xSize = ySize;
ySize = temp;
}
uint64_t ans=0;
for(int16_t i=ySize; i>= 0; i--){
uint64_t b = ((fact(ySize)/fact(ySize-i))*perm(i,xSize))%M;
// cout <<b <<"\n";
uint16_t tempY = ySize-i;
uint16_t tempX = xSize-i;
uint64_t temp = 0;
for(uint16_t i2=0; i2 <= tempX; i2++){
if((taulSize-ySize-(aAmount-i2)) >= tempX){
uint64_t temp1 = (perm(i2, aAmount)*(perm(tempX-i2,taulSize-ySize-aAmount))*fact(tempX))%M;
for(uint16_t i3=0; (i3 <= tempY)&& (i3 <= (aAmount-i2)); i3++){
if((taulSize-xSize-(aAmount-i2-i3)) >= tempY){
uint64_t temp2 = (perm(i3,aAmount-i2)*(perm(tempY-i3,taulSize-xSize-aAmount+i2))*fact(tempY))%M;
temp += (temp1*temp2*solutions(taulSize-i-tempX-tempY, aAmount-i2-i3))%M;
}
}
}
}
// cout << temp<<"\n";
ans += (b*temp*solutions(aSize, i+extraB))%M;
}
cout <<ans<<"\n";
}

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
2
..
..

correct output
2

user output
2

Test 2

Group: 1

Verdict: ACCEPTED

input
2
..
A.

correct output
1

user output
1

Test 3

Group: 1

Verdict: ACCEPTED

input
2
B.
.A

correct output
0

user output
0

Test 4

Group: 1

Verdict: ACCEPTED

input
3
...
...
...

correct output
12

user output
12

Test 5

Group: 1

Verdict: ACCEPTED

input
4
....
....
....
....

correct output
216

user output
216

Test 6

Group: 1

Verdict: ACCEPTED

input
5
.....
.....
.....
.....
...

correct output
5280

user output
5280

Test 7

Group: 1

Verdict: ACCEPTED

input
5
....A
.....
.....
.....
...

correct output
264

user output
264

Test 8

Group: 1

Verdict:

input
5
B....
.....
.....
.A.B.
...

correct output
22

user output
0

Test 9

Group: 1

Verdict:

input
5
B.A..
....A
.....
A.B..
...

correct output
2

user output
0

Test 10

Group: 1

Verdict:

input
5
A.B..
BA...
.B.A.
...BA
...

correct output
1

user output
0

Test 11

Group: 2

Verdict: ACCEPTED

input
10
..........
..........
..........
..........
...

correct output
306442892

user output
306442892

Test 12

Group: 2

Verdict: ACCEPTED

input
50
.................................

correct output
694861480

user output
694861480

Test 13

Group: 2

Verdict: ACCEPTED

input
111
.................................

correct output
555319110

user output
555319110

Test 14

Group: 2

Verdict: ACCEPTED

input
222
.................................

correct output
108372237

user output
108372237

Test 15

Group: 2

Verdict: ACCEPTED

input
333
.................................

correct output
259107857

user output
259107857

Test 16

Group: 2

Verdict: ACCEPTED

input
444
.................................

correct output
19906314

user output
19906314

Test 17

Group: 2

Verdict: ACCEPTED

input
497
.................................

correct output
224313667

user output
224313667

Test 18

Group: 2

Verdict: ACCEPTED

input
498
.................................

correct output
929574601

user output
929574601

Test 19

Group: 2

Verdict: ACCEPTED

input
499
.................................

correct output
600226043

user output
600226043

Test 20

Group: 2

Verdict: ACCEPTED

input
500
.................................

correct output
198353194

user output
198353194

Test 21

Group: 3

Verdict:

input
499
.................................

correct output
840243733

user output
0

Test 22

Group: 3

Verdict:

input
499
........................A........

correct output
4146290

user output
0

Test 23

Group: 3

Verdict:

input
499
B.........A......................

correct output
173518884

user output
0

Test 24

Group: 3

Verdict:

input
499
...A....B........................

correct output
20044800

user output
0

Test 25

Group: 3

Verdict:

input
499
AB...............................

correct output
2

user output
0

Test 26

Group: 3

Verdict:

input
500
.................................

correct output
121064146

user output
0

Test 27

Group: 3

Verdict:

input
500
.................................

correct output
848435259

user output
0

Test 28

Group: 3

Verdict:

input
500
.....B........A..................

correct output
296240911

user output
0

Test 29

Group: 3

Verdict:

input
500
.A......B........................

correct output
2196

user output
0

Test 30

Group: 3

Verdict:

input
500
...AB............................

correct output
1

user output
0