# 求组合数
def C(a, b):  # a为上限， b为下限res = 1for i in range(a):res *= b / a# 当结果大于目标值时无需继续运算，提高效率if res > target:return resb -= 1a -= 1return res# 二分查找目标元素
def search(k):# 起始下限，也就是对称轴位置的元素low = 2 * k# 终点下限high = target# 可能出现high 小于 low 的情况，比如目标值很小，但行数还在十多行的时候。# 这时候直接判断该斜行第一个元素也就是对称轴位置的元素的值是否是目标值即可。if high <= low and C(k, low) != target:return Falsewhile low <= high:mid = low + (high - low) // 2val = C(k, mid)if val > target:high = mid - 1elif val < target:low = mid + 1else:# 根据等差数列前N项和公式求出前面有多少个元素，然后再加上他所在的列数print(int(mid * (mid + 1) / 2) + k + 1)return Truereturn Falsetarget = int(input())
# range第二个参数必须是-1，因为第0斜行才有1。
for i in range(16, -1, -1):if search(i):break

转载学习