Task: | Equilateral numbers |
Sender: | ht |
Submission time: | 2024-11-16 16:37:51 +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 | ACCEPTED | 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) // 2e = eqnum(mid)# print(n, k0, k1, e)if e == n:return midelif k0 >= k1:if e > n:return k1-1else:return k1elif 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)e3 = eqnum(k-2)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 e3 >= 1 and 3*e3 <= n:nums3 = [x for x in nums]nums3.append(e3)nums3.append(e3)nums3.append(e3)others.append([nums3, k-2, n-e3*3, c+3])if e <= n:c += 1n -= enums.append(e)else:k = binsearch(n,1,k)completed.append([x for x in nums])mc = 10**12mp = None# print(len(completed), completed)for c in completed:if len(c) < mc:mp = cmc = len(c)# print("val", c)print(mc)# print(len(mc), sum(nums), nums)# 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: ACCEPTED
input |
---|
3 |
correct output |
---|
1 |
user output |
---|
1 |
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 |
---|
5 |
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 |
---|
5 |
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 |
---|
5 |
Test 12
Verdict: WRONG ANSWER
input |
---|
999999999998 |
correct output |
---|
3 |
user output |
---|
5 |