Task: | Equilateral numbers |
Sender: | ht |
Submission time: | 2024-11-16 16:43:26 +0200 |
Language: | Python3 (PyPy3) |
Status: | READY |
Result: | TIME LIMIT EXCEEDED |
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 | ACCEPTED | 0.04 s | details |
#6 | TIME LIMIT EXCEEDED | -- | details |
#7 | WRONG ANSWER | 0.08 s | details |
#8 | WRONG ANSWER | 0.08 s | details |
#9 | WRONG ANSWER | 0.04 s | details |
#10 | ACCEPTED | 0.04 s | details |
#11 | WRONG ANSWER | 0.08 s | details |
#12 | WRONG ANSWER | 0.07 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])for l in range(k-2,k-32,-1):if l < 1:breakeee = eqnum(l)if eee >= 1 and (k - l)*eee <= n:nums3 = [x for x in nums]for _ in range(k - l):nums3.append(eee)others.append([nums3, l, n-(k - l) * eee, 0])if e <= n:c += 1n -= enums.append(e)else:k = binsearch(n,1,k)# print(len(others))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: ACCEPTED
input |
---|
33 |
correct output |
---|
3 |
user output |
---|
3 |
Test 6
Verdict: TIME LIMIT EXCEEDED
input |
---|
12385719843 |
correct output |
---|
3 |
user output |
---|
(empty) |
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 |
---|
5 |
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 |