CSES - Datatähti 2023 alku - Results
Submission details
Task:Ruudukko
Sender:eetumaenpaa
Submission time:2022-11-08 22:33:06 +0200
Language:CPython3
Status:READY
Result:28
Feedback
groupverdictscore
#1ACCEPTED28
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1, 2, 3details
#2ACCEPTED0.02 s1, 2, 3details
#3ACCEPTED0.02 s1, 2, 3details
#4ACCEPTED0.03 s2, 3details
#5--2, 3details
#6--2, 3details
#7--3details
#8--3details
#9--3details

Code

# n = 3
# rows = [
#     [2,1,3], 
#     [1,1,1], 
#     [9,2,7]
# ]

# for row in rows:
#     for element in row:
#         print(element, end=" ")
#     print()

n = int(input())
rows = []
for i in range(n):
    rows.append([int(x) for x in input().split(" ")])

unique_numbers = []
for i in range(n):
    for j in range(n):
        if(rows[i][j] not in unique_numbers):
            unique_numbers.append(rows[i][j])
unique_numbers = sorted(unique_numbers)

def count_global(lambda_function):
    c = 0
    for i in range(n):
        for j in range(n):
            if lambda_function(rows[i][j]):
                c += 1
    return c

def get_numbers_on_row_and_column(x, y, include_self):
    numbers = []
    for i in range(n):
        if include_self:
            numbers.append(rows[y][i])
        else:
            if i != x:
                numbers.append(rows[y][i])
    for i in range(n):
        if i != y:
            numbers.append(rows[i][x])
    return numbers

def get_coordinates_of(lambda_function):
    coords = []
    for i in range(n):
        for j in range(n):
            if lambda_function(rows[i][j]):
                coords.append((i, j))
    return coords

def count_elements_on_same_row_and_column(x, y, lambda_function):
    c = 0
    for i in range(n):
        if lambda_function(rows[y][i]):
            c += 1
        if lambda_function(rows[i][x]):
            c += 1
    return c

cache = {}

for num in unique_numbers:

    coords = get_coordinates_of(lambda x: x == num)
    # print(cache)
    if num == unique_numbers[0]:
        for c in coords:
            cache[c] = 1
    else:
        for c in coords:
            # lines = get_numbers_on_row_and_column(c[1], c[0], False)
            # lines = [x for x in lines if x < num]
            # print(cache)
            # print(lines)
            lines = get_coordinates_of(lambda x: x < num)
            lines = [x for x in lines if x[0] == c[0] or x[1] == c[1]]
            # print(lines)
            cache[c] = sum([cache[x] for x in lines]) + 1


    # c = 0
    # coords = get_coordinates_of(lambda x: x == num)
    # print(num, coords)
    # for coord in coords:
    #     c += count_elements_on_same_row_and_column(coord[0], coord[1], lambda x: x < num)
    
    # global_count = count_global(lambda x: x == num)
    # print(num, "adding", c + global_count)

    # cache[num] = c + global_count

print(sum(cache.values()))

# answer = 0
# for i in range(len(unique_numbers)):
#     for j in range(i):
#         answer += cache[unique_numbers[j]]

# print(answer % (10**9 + 7))

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
1 1 1
1 1 1
1 1 1

correct output
9

user output
9

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
1 2 3
6 5 4
7 8 9

correct output
135

user output
135

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
7 8 1
4 5 4
3 9 6

correct output
57

user output
57

Test 4

Group: 2, 3

Verdict: ACCEPTED

input
100
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
10000

user output
10000

Test 5

Group: 2, 3

Verdict:

input
100
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
187458477

user output
(empty)

Test 6

Group: 2, 3

Verdict:

input
100
2995 8734 1018 2513 7971 5063 ...

correct output
964692694

user output
(empty)

Test 7

Group: 3

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
1000000

user output
(empty)

Test 8

Group: 3

Verdict:

input
1000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
229147081

user output
(empty)

Test 9

Group: 3

Verdict:

input
1000
520283 805991 492643 75254 527...

correct output
951147313

user output
(empty)