1. 程式人生 > >牛客網線上程式設計(六):求數列的和-python

牛客網線上程式設計(六):求數列的和-python

題目描述

數列的第一項為n,以後各項為前一項的平方根,求數列的前m項的和。

輸入描述:

輸入資料有多組,每組佔一行,由兩個整數n(n < 10000)和m(m < 1000)組成,n和m的含義如前所述。

輸出描述:

對於每組輸入資料,輸出該數列的和,每個測試例項佔一行,要求精度保留2位小數。

示例1

輸入

81 4
2 2

輸出

94.73
3.41

程式碼實現

import sys
import math
lines = sys.stdin.readlines()

for line in lines:
    n,m=map(int,line.split())
    lis=[n]
    for i in range(m-1):
            lis.append(math.sqrt(lis[i]))
    print ('%.2f' %sum(lis))