AtCoder Beginner Contest 106 C. To Infinity【Python】

https://atcoder.jp/contests/abc106/tasks/abc106_c

AtCoder ProblemsのRecommendationで Difficulty: 441、Solve Probability: 43%でした。

1文字目から1が何文字続くかとその次の数字のみ考えれば良いと考えた結果解くことができました。

S = input()
K = int(input())

one_count = 0

for i in S:
    if i == '1':
        one_count += 1
    else:
        inf_num = i
        break

if K <= one_count:
    print(1)
else:
    print(int(i))