编程题--疯狂序列----京东大数据笔试

1. 题目描述

东东从京京那里了解到有一个无限长的数字序列: 1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 5, …(数字k在该序列中正好出现k次)。东东想知道这个数字序列的第n项是多少,你能帮帮他么？

输入描述:
输入包括一个整数n(1 ≤ n ≤ 10^18)

输出描述:
输出一个整数,即数字序列的第n项

输入例子1:
169

输出例子1:
18

2. 解题思路

第一项1个数，第二项2个数，第三项3个数…
所以求第n个序列代表的数就是求前K项的和
1+2+3+…

3. 代码实现

import math
class Solution:
    def crazySequence(self,n):
        return math.ceil((2 * n + 0.25) ** 0.5 - 0.5)

s = Solution()
result = s.crazySequence(169)
print(result)