CSES - Datatähti 2026 alku

Submission details
Task:	Tulkki
Sender:	jokeri2222
Submission time:	2025-11-04 14:22:41 +0200
Language:	Python3 (PyPy3)
Status:	READY
Result:	44

Feedback
group	verdict	score
#1	ACCEPTED	12
#2	ACCEPTED	32
#3	TIME LIMIT EXCEEDED	0

Test results
test	verdict	time	group
#1	ACCEPTED	0.04 s	1, 2, 3	details
#2	ACCEPTED	0.04 s	1, 2, 3	details
#3	ACCEPTED	0.04 s	1, 2, 3	details
#4	ACCEPTED	0.04 s	1, 2, 3	details
#5	ACCEPTED	0.10 s	1, 2, 3	details
#6	ACCEPTED	0.10 s	1, 2, 3	details
#7	ACCEPTED	0.04 s	2, 3	details
#8	ACCEPTED	0.04 s	2, 3	details
#9	ACCEPTED	0.04 s	2, 3	details
#10	ACCEPTED	0.04 s	2, 3	details
#11	ACCEPTED	0.05 s	2, 3	details
#12	ACCEPTED	0.04 s	2, 3	details
#13	ACCEPTED	0.04 s	3	details
#14	ACCEPTED	0.04 s	3	details
#15	TIME LIMIT EXCEEDED	--	3	details
#16	ACCEPTED	0.10 s	3	details
#17	TIME LIMIT EXCEEDED	--	3	details
#18	TIME LIMIT EXCEEDED	--	3	details

Code

nth_index = lambda l, item, n: [idx for idx, elem in enumerate(l) if elem == item][n-1]

program = []
commands = [
    "PRINT",
    "CLEAR",
    "INCREASE",
    "REPEAT"
]

for i in range(1000):
    try:
        _input = input().replace("TIMES", "")
    except:
        break
    if _input == "\\r":
        break
    sub_prog = list(filter(lambda s: s, _input.split("#")[0].split(" ")))
    if sub_prog:
        program.extend(sub_prog)


stack = []
counter = 0
for i, sub in enumerate(program):
    if sub == "(":
        stack.append(f"({counter}")
        counter+=1
    if sub == ")":
        loop_start_idx = nth_index(program, "(", len(stack))
        loop_start = stack.pop(-1)
        program[loop_start_idx]=loop_start
        program[i]=f"){loop_start[1:]}"


output = []
VARS = {}
def CLEAR(_, var, _2):
    VARS[var]=0
    return _
def INCREASE(_, var, _2):
    VARS[var]+=1
    return _
def PRINT(_, var, _2):
    output.append(VARS[var])
    return _
def REPEAT(index_of_code, var_times, _loop_idx):
    _command_used = ""
    _end_idx = program.index(f"){_loop_idx}")
    for _ in range(VARS[var_times]):
        _next_idx = index_of_code
        for _i, _sub in enumerate(program[index_of_code:_end_idx+1]):
            _i+=index_of_code
            if _i < _next_idx:
                continue
            if _command_used:
                if _sub not in VARS.keys():
                    VARS[_sub] = 0
                next_loop_idx=_loop_idx
                if _command_used == "REPEAT":
                    next_loop_idx = int(program[_i + 1][1:])
                _next_idx = eval(f"{_command_used}({_i+1}, '{_sub}', {next_loop_idx})")
                _command_used = ""
            if _sub in commands:
                _command_used = _sub

    return _end_idx+1

command_used = ""
next_idx = -1
loop_idx = 0
for i, sub in enumerate(program):
    if i<next_idx:
        continue
    if command_used:
        if sub not in VARS.keys():
            VARS[sub] = 0
        if command_used=="REPEAT":
            loop_idx = int(program[i+1][1:])
        next_idx = eval(f"{command_used}({i+1}, '{sub}', {loop_idx})")
        command_used=""
    if sub in commands:
        command_used = sub

print(" ".join(map(str, output)))

Test details

Test 1 (public)

Group: 1, 2, 3

Verdict: ACCEPTED

input
`PRINT X INCREASE X PRINT X INCREASE X PRINT X ...` view save

correct output
`0 1 2 0` view save

user output
`0 1 2 0` view save

Test 2 (public)

Group: 1, 2, 3

Verdict: ACCEPTED

input
`INCREASE X # aybabtu PRINT X INCREASE # test ...` view save

correct output
`1 3` view save

user output
`1 3` view save

Test 3 (public)

Group: 1, 2, 3

Verdict: ACCEPTED

input
`# Create number 3 INCREASE X INCREASE X INCREASE X ...` view save

correct output
`3` view save

user output
`3` view save

Test 4 (public)

Group: 1, 2, 3

Verdict: ACCEPTED

input
`INCREASE A PRINT A INCREASE B PRINT B INCREASE C ...` view save

correct output
`1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

user output
`1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

Test 5 (public)

Group: 1, 2, 3

Verdict: ACCEPTED

input
`INCREASE X INCREASE X INCREASE X INCREASE X INCREASE X ...` view save

correct output
`999` view save

user output
`999` view save

Test 6 (public)

Group: 1, 2, 3

Verdict: ACCEPTED

input
`PRINT X PRINT X PRINT X PRINT X PRINT X ...` view save

correct output
`0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...` view save

user output
`0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...` view save

Test 7 (public)

Group: 2, 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`5 5 5 5 5` view save

user output
`5 5 5 5 5` view save

Test 8 (public)

Group: 2, 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`0 0 0 0 0` view save

user output
`0 0 0 0 0` view save

Test 9 (public)

Group: 2, 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`6 7 8 9 10` view save

user output
`6 7 8 9 10` view save

Test 10 (public)

Group: 2, 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`5 5` view save

user output
`5 5` view save

Test 11 (public)

Group: 2, 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`20` view save

user output
`20` view save

Test 12 (public)

Group: 2, 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE B INCREASE B ...` view save

correct output
`42` view save

user output
`42` view save

Test 13 (public)

Group: 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`1 2 2 3 3 3 4 4 4 4 5 5 5 5 5` view save

user output
`1 2 2 3 3 3 4 4 4 4 5 5 5 5 5` view save

Test 14 (public)

Group: 3

Verdict: ACCEPTED

input
`# Create number 3 INCREASE A INCREASE A INCREASE...` view save

correct output
`12` view save

user output
`12` view save

Test 15 (public)

Group: 3

Verdict: TIME LIMIT EXCEEDED

input
`INCREASE X INCREASE X INCREASE X INCREASE X INCREASE X ...` view save

correct output
`531441` view save

user output
(empty)

Test 16 (public)

Group: 3

Verdict: ACCEPTED

input
`INCREASE A INCREASE A INCREASE A INCREASE A INCREASE A ...` view save

correct output
`1337` view save

user output
`1337` view save

Test 17 (public)

Group: 3

Verdict: TIME LIMIT EXCEEDED

input
`INCREASE A INCREASE A REPEAT A TIMES ( REPEAT A TIMES ( ...` view save

correct output
`1 2 1 2 1 1 3 4 3 4 3 4 3 4 3 ...` view save

user output
(empty)

Test 18 (public)

Group: 3

Verdict: TIME LIMIT EXCEEDED

input
`# Efficient algorithm for find...` view save

correct output
`2 3 5 7 11 13 17 19 23 29 31 3...` view save

user output
(empty)

Datatähti 2026 alku

Tulkki

Code

Test details

Test 1 (public)

Test 2 (public)

Test 3 (public)

Test 4 (public)

Test 5 (public)

Test 6 (public)

Test 7 (public)

Test 8 (public)

Test 9 (public)

Test 10 (public)

Test 11 (public)

Test 12 (public)

Test 13 (public)

Test 14 (public)

Test 15 (public)

Test 16 (public)

Test 17 (public)

Test 18 (public)

Tasks

Submissions (jokeri2222)