| Task: | Equilateral numbers |
| Sender: | ht |
| Submission time: | 2024-11-16 16:31:13 +0200 |
| Language: | Python3 (PyPy3) |
| Status: | READY |
| Result: | WRONG ANSWER |
| test | verdict | time | |
|---|---|---|---|
| #1 | ACCEPTED | 0.04 s | details |
| #2 | ACCEPTED | 0.04 s | details |
| #3 | WRONG ANSWER | 0.04 s | details |
| #4 | ACCEPTED | 0.04 s | details |
| #5 | WRONG ANSWER | 0.04 s | details |
| #6 | WRONG ANSWER | 0.04 s | details |
| #7 | WRONG ANSWER | 0.04 s | details |
| #8 | WRONG ANSWER | 0.04 s | details |
| #9 | WRONG ANSWER | 0.04 s | details |
| #10 | ACCEPTED | 0.04 s | details |
| #11 | WRONG ANSWER | 0.04 s | details |
| #12 | WRONG ANSWER | 0.04 s | details |
Code
def eqnum(k):
if k % 2 == 1:
return k*((k+1)//2)
else:
return (k//2)*(k+1)
# print(eqnum(120000), 10**12)
# exit()
def binsearch(n, k0, k1):
mid = (k0 + k1) // 2
e = eqnum(mid)
# print(n, k0, k1, e)
if e == n:
return mid
elif k0 >= k1:
if e > n:
return k1-1
else:
return k1
elif e > n:
return binsearch(n, k0, mid-1)
else:
return binsearch(n, mid+1, k1)
n0 = int(input())
others = [([], 10**13, n0, 0)]
completed = []
while len(others):
nums, k, n, c = others[0]
others = others[1:]
while(n):
e = eqnum(k)
e2 = eqnum(k-1)
if e2 >= 1 and 2*e2 <= n:
nums2 = [x for x in nums]
nums2.append(e2)
nums2.append(e2)
others.append([nums2, k-1, n-e2*2, c+2])
if e <= n:
c += 1
n -= e
nums.append(e)
else:
k = binsearch(n,1,k)
completed.append(nums)
mc = 10**12
mp = []
for c in completed:
if len(c) < mc:
mp = c
mc = len(c)
print(len(c))
# print(c, sum(nums), nums)
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: WRONG ANSWER
| input |
|---|
| 3 |
| correct output |
|---|
| 1 |
| user output |
|---|
| 3 |
Test 4
Verdict: ACCEPTED
| input |
|---|
| 5 |
| correct output |
|---|
| 3 |
| user output |
|---|
| 3 |
Test 5
Verdict: WRONG ANSWER
| input |
|---|
| 33 |
| correct output |
|---|
| 3 |
| user output |
|---|
| 4 |
Test 6
Verdict: WRONG ANSWER
| input |
|---|
| 12385719843 |
| correct output |
|---|
| 3 |
| user output |
|---|
| 7 |
Test 7
Verdict: WRONG ANSWER
| input |
|---|
| 10935032 |
| correct output |
|---|
| 3 |
| user output |
|---|
| 4 |
Test 8
Verdict: WRONG ANSWER
| input |
|---|
| 659023495928 |
| correct output |
|---|
| 3 |
| user output |
|---|
| 7 |
Test 9
Verdict: WRONG ANSWER
| input |
|---|
| 913591235689 |
| correct output |
|---|
| 2 |
| user output |
|---|
| 6 |
Test 10
Verdict: ACCEPTED
| input |
|---|
| 999999911791 |
| correct output |
|---|
| 1 |
| user output |
|---|
| 1 |
Test 11
Verdict: WRONG ANSWER
| input |
|---|
| 1000000000000 |
| correct output |
|---|
| 2 |
| user output |
|---|
| 8 |
Test 12
Verdict: WRONG ANSWER
| input |
|---|
| 999999999998 |
| correct output |
|---|
| 3 |
| user output |
|---|
| 6 |
