| Task: | Ruudukko |
| Sender: | okkokko |
| Submission time: | 2022-11-09 23:56:51 +0200 |
| Language: | Python3 (PyPy3) |
| Status: | READY |
| Result: | 61 |
| group | verdict | score |
|---|---|---|
| #1 | ACCEPTED | 28 |
| #2 | ACCEPTED | 33 |
| #3 | TIME LIMIT EXCEEDED | 0 |
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.07 s | 1, 2, 3 | details |
| #2 | ACCEPTED | 0.07 s | 1, 2, 3 | details |
| #3 | ACCEPTED | 0.07 s | 1, 2, 3 | details |
| #4 | ACCEPTED | 0.09 s | 2, 3 | details |
| #5 | ACCEPTED | 0.09 s | 2, 3 | details |
| #6 | ACCEPTED | 0.11 s | 2, 3 | details |
| #7 | TIME LIMIT EXCEEDED | -- | 3 | details |
| #8 | TIME LIMIT EXCEEDED | -- | 3 | details |
| #9 | TIME LIMIT EXCEEDED | -- | 3 | details |
Code
# completes tests 1 and 2, but not 3
import random
import time
import sys
from typing import Generator, TypeVar
import itertools
def I():
return map(int, input().split())
# DIVISOR = 10**9 + 7
_V = TypeVar("_V")
test_N = 1000
test_seed = 1001
# seed is 1001
test_expected_results = {
100: 773224951,
200: 452330452,
300: 37161558,
400: 250332178,
1000: 298053617
}
def test():
n = test_N
def text_gen():
yield str(n)
random.seed(test_seed)
for i in range(n):
yield " ".join((str(random.randint(1, n * n)) for _ in range(n)))
return iter(list(text_gen())).__next__
# input = test()
timer_init = time.perf_counter()
class Node:
# ended up unused
"Binary Search Tree node"
def __init__(self, parent: "Node|None", left: "Node|None", right: "Node|None", key: "int|float"):
self.key = key
self.parent = parent
self.left = left
self.right = right
self.values = []
self.marker = 0
pass
def Get(self) -> "list":
return self.values
...
def Add(self, value):
self.values.append(value)
def next(self):
if self.marker == 0:
self.marker += 1
if self.left:
return self.left
if self.marker == 1:
self.marker += 1
if self.right:
return self.right
self.marker = 0
return self.parent
class BST:
# ended up unused
def __init__(self, middle: float):
self.root = Node(None, None, None, middle)
def Insert(self, key: "int | float", value: "_V"):
node, pos = self.search(key)
if pos == 1:
node.left = Node(node, None, None, key)
node.left.Add(value)
elif pos == 2:
node.right = Node(node, None, None, key)
node.right.Add(value)
else:
node.Add(value)
# def Get(self, key: "int | float") -> "Node":
# ...
def Iter(self) -> "Generator[Node,None,None]":
node = self.root
while 1:
new = node.next()
if new is None:
break
if node.marker == 0:
# new is parent
if new.marker == 1:
# node is left
yield new
elif new.left is None:
yield new
# elif node.marker == 1:
# # new is right child
node = new
def Iter_Values(self):
# for node in self.Iter():
# for x, y in node.Get():
# yield node.key, x, y
return ((node.key, v) for node in self.Iter() for v in node.Get())
def search(self, key: "int|float") -> "tuple[Node, int]":
"""returns a node and a number.
0 means the node has the wanted value,
1 means the value doesn't exist but would be the node's left branch if inserted,
2 means the same as 1 but for the right branch."""
# no recursion
current = self.root
while 1:
if key < current.key:
if current.left:
current = current.left
else:
return current, 1
elif key > current.key:
if current.right:
current = current.right
else:
return current, 2
else:
return current, 0
def timer(desc: "str", prev: "float"):
t = time.perf_counter()
print(desc, t - prev, file=sys.stderr)
return t
def main():
DIVISOR = 10**9 + 7
timer_0 = time.perf_counter()
n, = I()
grid = [list(I()) for _ in range(n)]
sett = list((grid[y][x], x, y) for y in range(n) for x in range(n))
# sett = set((a, x, y) for (y, x), a in zip(itertools.product(range(n), repeat=2), itertools.chain(*(I() for _ in range(n)))))
timer_1 = timer("timer 1:", timer_0)
line_row = [0] * n
line_row_last = [1] * n
line_col = [0] * n
line_col_last = [1] * n
line_row_added = [0] * n
line_col_added = [0] * n
timer_2 = timer("timer 2:", timer_1)
sor = sorted(sett, key=lambda x: x[0])
timer_3 = timer("sorting:", timer_2)
total = 0
for number, x, y in sor:
if number != line_row_last[y]:
line_row_last[y] = number
line_row[y] += line_row_added[y]
line_row_added[y] = 0
if number != line_col_last[x]:
line_col_last[x] = number
line_col[x] += line_col_added[x]
line_col_added[x] = 0
value = (line_col[x] + line_row[y] + 1) % DIVISOR
line_row_added[y] += value
line_col_added[x] += value
total += value
# total = sum(calculate_routes(y, x, k)for k, x, y in sor)
# total = sum(calculate_routes_full(*s)for s in sor)
timer_4 = timer("timer 4:", timer_3)
return total % DIVISOR
def old_iteration(n, grid, exists_row, exists_all, calculate_routes):
for a in range(1, n * n + 1):
if a not in exists_all:
continue
for y in range(n):
if a not in exists_row[y]:
continue
x = -1
try:
while True:
x = grid[y].index(a, x + 1)
calculate_routes(y, x, a)
except ValueError:
pass
timer_start = time.perf_counter()
w = main()
print(w)
# 100: 773224951
# 200: 452330452
timer_end = time.perf_counter()
print("time:", timer_end - timer_start,
("\ncorrect" if w == test_expected_results[test_N] else "\ndifferent") if test_N in test_expected_results.keys() else "no data",
# "\ninit:", timer_start - timer_init,
"\nseed:", test_seed,
file=sys.stderr)
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 |
Error:
timer 1: 0.00020664592739194632 timer 2: 2.742407377809286e-05 sorting: 2.7357949875295162...
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 |
Error:
timer 1: 0.00019310100469738245 timer 2: 2.7289963327348232e-05 sorting: 2.771697472780943...
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 |
Error:
timer 1: 0.00018205400556325912 timer 2: 2.65931012108922e-05 sorting: 2.6968889869749546e...
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 |
Error:
timer 1: 0.0109414680628106 timer 2: 6.188603583723307e-05 sorting: 0.0036987699568271637...
Test 5
Group: 2, 3
Verdict: ACCEPTED
| input |
|---|
| 100 1 2 3 4 5 6 7 8 9 10 11 12 13 ... |
| correct output |
|---|
| 187458477 |
| user output |
|---|
| 187458477 |
Error:
timer 1: 0.010855545056983829 timer 2: 6.181898061186075e-05 sorting: 0.00383433501701802...
Test 6
Group: 2, 3
Verdict: ACCEPTED
| input |
|---|
| 100 2995 8734 1018 2513 7971 5063 ... |
| correct output |
|---|
| 964692694 |
| user output |
|---|
| 964692694 |
Error:
timer 1: 0.012620743014849722 timer 2: 6.153108552098274e-05 sorting: 0.01132322195917368...
Test 7
Group: 3
Verdict: TIME LIMIT EXCEEDED
| 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: TIME LIMIT EXCEEDED
| 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: TIME LIMIT EXCEEDED
| input |
|---|
| 1000 520283 805991 492643 75254 527... |
| correct output |
|---|
| 951147313 |
| user output |
|---|
| (empty) |
