Submission #22557207

---

Source Code Expand

def sigma(n):
    return n*(n+1)//2

def oracle1(l):
    if l in range(0, N):
        return sigma(l+1)
    if l in range(N, 2*N-3+1):
        return sigma(l+1)-3*sigma(l+1-N)
    if l in range(2*N-3+1, 3*N-3+1):
        return sigma(3*N-2-l)
    return 0

def oracle2(l, i):
    L = max(0, l-i-(N-1))
    R = min(N-1, l-i)
    return max(0, R-L+1)

def oracle3(l, i, j):
    k = l - i - j
    if k in range(N):
        return 1
    return 0

N, K = map(int, input().split())
for l in range(0, 3*N-2):
    d = oracle1(l)
    if K-d <= 0:
        break
    K -= d
for i in range(N):
    d = oracle2(l, i)
    if K-d <= 0:
        break
    K -= d
for j in range(N):
    d = oracle3(l, i, j)
    if K-d <= 0:
        break
    K -= d
k = l - i - j
print(i+1, j+1, k+1)

Submission Info

Judge Result