python解题-数字游戏

解题思路:
找出规律，栋栋每次读的数字为(1+(i*n)*(i*n+1)//2) % k

注意事项:

注意超时

参考代码:

def function(n,k,t):  
    A = []  
    temp = 1        #每轮要读出的输入  
  
    for i in range(1,t+1):  
        A.append(temp)  
        temp = (1+(i*n)*(i*n+1)//2) % k  
    
    print(int(sum(A)))  
  
  
if __name__ == '__main__':  
    n,k,t = map(int,input().strip().split())  
    function(n,k,t)