Python80分求助

原题链接：https://www.luogu.com.cn/problem/B2145

代码：

def digit(n, k):
    p = list(str(n))
    return p[-k]

num = input().split()
n = int(num[0])
k = int(num[1])

print(digit(n, k))