CSES - HIIT Open 2024 - Results
Submission details
Task:Equilateral numbers
Sender:Tunkkari
Submission time:2024-11-16 15:51:57 +0200
Language:Python3 (PyPy3)
Status:READY
Result:
Test results
testverdicttime
#1ACCEPTED0.04 sdetails
#2ACCEPTED0.04 sdetails
#3ACCEPTED0.04 sdetails
#4ACCEPTED0.04 sdetails
#50.04 sdetails
#60.05 sdetails
#70.04 sdetails
#80.06 sdetails
#90.05 sdetails
#10ACCEPTED0.05 sdetails
#110.05 sdetails
#120.05 sdetails

Code

import math

k = int(input())

def eq(n):
    return int((n*(n+1))/2)

def inner(target):
    if target == 0:
        return 0

    i = 0
    e = eq(i)

    while e < target:
        i += 1
        e = eq(i)
    
    
    if e == target:
        # print(e,i)
        return 1
    elif e > target:
        one_smaller = eq(i-1)
        # print(one_smaller,i-1)
        diff = target - one_smaller
        return 1 + inner(diff)

    elif e < target:
        return 1 + inner(target-e)

print(inner(k))

Test details

Test 1

Verdict: ACCEPTED

input
1

correct output
1

user output
1

Test 2

Verdict: ACCEPTED

input
2

correct output
2

user output
2

Test 3

Verdict: ACCEPTED

input
3

correct output
1

user output
1

Test 4

Verdict: ACCEPTED

input
5

correct output
3

user output
3

Test 5

Verdict:

input
33

correct output
3

user output
4

Test 6

Verdict:

input
12385719843

correct output
3

user output
5

Test 7

Verdict:

input
10935032

correct output
3

user output
4

Test 8

Verdict:

input
659023495928

correct output
3

user output
5

Test 9

Verdict:

input
913591235689

correct output
2

user output
6

Test 10

Verdict: ACCEPTED

input
999999911791

correct output
1

user output
1

Test 11

Verdict:

input
1000000000000

correct output
2

user output
5

Test 12

Verdict:

input
999999999998

correct output
3

user output
5