核电站python直观解法

解题思路:

后续添加

注意事项:

后续添加

参考代码:

def f(m1, n1):
   if m1 == 0:
       return 1

   sum_m = 1
   for n_f in range(n1-m1+1, n1+1):
       sum_m *= n_f
   for m_f in range(1, m1+1):
       sum_m /= m_f
   return int(sum_m)


results = []

while True:
   try:
       n = int(input())
   except:
       break
   else:
       over_number = 0
       min_pit = n // 3
       for pit_number in range(min_pit, n):
           min_double_nm = max(0, n - 2 * pit_number - 1)
           max_double_nm = (n - pit_number) // 2

           for double_nm_number in range(min_double_nm, max_double_nm+1):
               over_number += f(double_nm_number, pit_number+1) * f(n-pit_number-2*double_nm_number, pit_number+1-double_nm_number)

       results.append(over_number+1)

for result in results:
   print(result)