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) // 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) 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 += 1 n -= e nums.append(e) else: k = binsearch(n,1,k) completed.append([x for x in nums]) mc = 10**12 mp = None # print(len(completed), completed) for c in completed: if len(c) < mc: mp = c mc = 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 |