Time Exceeded
Code
def binSearch(k, l, r):
if r - l <= 1:
return l**2 == k
mid = (r + l) // 2
if mid ** 2 == k:
return True
if mid ** 2 < k:
return binSearch(k, mid, r)
return binSearch(k, l, mid)
def isSquare(k):
return binSearch(k, 1, k)
def isTriangleNumber(k):
return isSquare(8*k + 9)
def kthChar(k):
return 'b' if isTriangleNumber(k) else 'a'
t = int(input())
for _ in range(t):
k = int(input())
print(kthChar(k))
Information
- Submit By
- Type
- Submission
- Problem
- P1223 C. Infinity Sequence
- Contest
- Happy New Year 2026
- Language
- Python 3 (Python 3.12.3)
- Submit At
- 2026-01-06 16:06:08
- Judged At
- 2026-01-06 16:06:08
- Judged By
- Score
- 20
- Total Time
- ≥1097ms
- Peak Memory
- ≥3.305 MiB